67 lines
1.2 KiB
Go
67 lines
1.2 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
func (c *Client) Delete(ctx context.Context, table, keyField string, data map[string]any) error {
|
|
_, err := validateData(table, keyField, data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
where := make([]string, 0, len(data))
|
|
args := make([]any, 0, len(data))
|
|
i := 1
|
|
|
|
for k, v := range data {
|
|
where = append(where, fmt.Sprintf("%s = $%d", quoteCol(k), i))
|
|
args = append(args, v)
|
|
i++
|
|
}
|
|
|
|
sql := fmt.Sprintf(
|
|
"DELETE FROM %s WHERE %s",
|
|
quoteTable(table),
|
|
strings.Join(where, " AND "),
|
|
)
|
|
|
|
_, err = c.exec().Exec(ctx, sql, args...)
|
|
return err
|
|
}
|
|
|
|
func (c *Client) BatchDelete(ctx context.Context, table, keyField string, list []map[string]any) error {
|
|
if len(list) == 0 {
|
|
return fmt.Errorf("empty data")
|
|
}
|
|
|
|
var (
|
|
args []any
|
|
in []string
|
|
i = 1
|
|
)
|
|
|
|
for _, row := range list {
|
|
v, err := validateData(table, keyField, row)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
in = append(in, fmt.Sprintf("$%d", i))
|
|
args = append(args, v)
|
|
i++
|
|
}
|
|
|
|
sql := fmt.Sprintf(
|
|
"DELETE FROM %s WHERE %s IN (%s)",
|
|
quoteTable(table),
|
|
quoteCol(keyField),
|
|
strings.Join(in, ", "),
|
|
)
|
|
|
|
_, err := c.exec().Exec(ctx, sql, args...)
|
|
return err
|
|
}
|