75 lines
1.2 KiB
Go
75 lines
1.2 KiB
Go
|
package pgm
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"github.com/jackc/pgx/v5"
|
||
|
|
)
|
||
|
|
|
||
|
|
type (
|
||
|
|
deleteQry struct {
|
||
|
|
table string
|
||
|
|
condition []Conditioner
|
||
|
|
args []any
|
||
|
|
debug bool
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
func (t *Table) Delete() WhereOrExec {
|
||
|
|
qb := &deleteQry{
|
||
|
|
table: t.Name,
|
||
|
|
debug: t.debug,
|
||
|
|
}
|
||
|
|
|
||
|
|
return qb
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q *deleteQry) Where(cond ...Conditioner) WhereOrExec {
|
||
|
|
q.condition = append(q.condition, cond...)
|
||
|
|
return q
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q *deleteQry) Exec(ctx context.Context) error {
|
||
|
|
_, err := poolPGX.Load().Exec(ctx, q.String(), q.args...)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q *deleteQry) ExecTx(ctx context.Context, tx pgx.Tx) error {
|
||
|
|
_, err := tx.Exec(ctx, q.String(), q.args...)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q *deleteQry) String() string {
|
||
|
|
sb := getSB()
|
||
|
|
defer putSB(sb)
|
||
|
|
|
||
|
|
n := len("DELETE FROM ") + len(q.table) + 20
|
||
|
|
sb.Grow(n)
|
||
|
|
|
||
|
|
// DELETE
|
||
|
|
sb.WriteString("DELETE FROM ")
|
||
|
|
sb.WriteString(q.table)
|
||
|
|
|
||
|
|
// WHERE
|
||
|
|
if len(q.condition) > 0 {
|
||
|
|
sb.WriteString(" WHERE ")
|
||
|
|
var argIdx int
|
||
|
|
for i, c := range q.condition {
|
||
|
|
argIdx = len(q.args)
|
||
|
|
if i > 0 {
|
||
|
|
sb.WriteString(" AND ")
|
||
|
|
}
|
||
|
|
sb.WriteString(c.Condition(&q.args, argIdx))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return sb.String()
|
||
|
|
}
|