summaryrefslogtreecommitdiff
path: root/vendor/github.com/uptrace/bun/query_delete.go
diff options
context:
space:
mode:
authorLibravatar tobi <31960611+tsmethurst@users.noreply.github.com>2023-01-22 12:26:47 +0100
committerLibravatar GitHub <noreply@github.com>2023-01-22 12:26:47 +0100
commit0ceacd7b1d0b03e3da1d552aa0bbbe7037e57e90 (patch)
tree106bd6f80fe5466522b37b5778ebc7c4fd6ab728 /vendor/github.com/uptrace/bun/query_delete.go
parent[chore] Add name to instance field for autosuggestion (#1359) (diff)
downloadgotosocial-0ceacd7b1d0b03e3da1d552aa0bbbe7037e57e90.tar.xz
[chore] bump db dependencies (#1366)
Diffstat (limited to 'vendor/github.com/uptrace/bun/query_delete.go')
-rw-r--r--vendor/github.com/uptrace/bun/query_delete.go54
1 files changed, 41 insertions, 13 deletions
diff --git a/vendor/github.com/uptrace/bun/query_delete.go b/vendor/github.com/uptrace/bun/query_delete.go
index 1572ac328..49a750cc8 100644
--- a/vendor/github.com/uptrace/bun/query_delete.go
+++ b/vendor/github.com/uptrace/bun/query_delete.go
@@ -39,9 +39,17 @@ func (q *DeleteQuery) Model(model interface{}) *DeleteQuery {
return q
}
+func (q *DeleteQuery) Err(err error) *DeleteQuery {
+ q.setErr(err)
+ return q
+}
+
// Apply calls the fn passing the DeleteQuery as an argument.
func (q *DeleteQuery) Apply(fn func(*DeleteQuery) *DeleteQuery) *DeleteQuery {
- return fn(q)
+ if fn != nil {
+ return fn(q)
+ }
+ return q
}
func (q *DeleteQuery) With(name string, query schema.QueryAppender) *DeleteQuery {
@@ -127,13 +135,6 @@ func (q *DeleteQuery) Returning(query string, args ...interface{}) *DeleteQuery
return q
}
-func (q *DeleteQuery) hasReturning() bool {
- if !q.db.features.Has(feature.Returning) {
- return false
- }
- return q.returningQuery.hasReturning()
-}
-
//------------------------------------------------------------------------------
func (q *DeleteQuery) Operation() string {
@@ -189,6 +190,14 @@ func (q *DeleteQuery) AppendQuery(fmter schema.Formatter, b []byte) (_ []byte, e
}
}
+ if q.hasFeature(feature.Output) && q.hasReturning() {
+ b = append(b, " OUTPUT "...)
+ b, err = q.appendOutput(fmter, b)
+ if err != nil {
+ return nil, err
+ }
+ }
+
b, err = q.mustAppendWhere(fmter, b, withAlias)
if err != nil {
return nil, err
@@ -223,7 +232,18 @@ func (q *DeleteQuery) softDeleteSet(fmter schema.Formatter, tm time.Time) string
//------------------------------------------------------------------------------
+func (q *DeleteQuery) Scan(ctx context.Context, dest ...interface{}) error {
+ _, err := q.scanOrExec(ctx, dest, true)
+ return err
+}
+
func (q *DeleteQuery) Exec(ctx context.Context, dest ...interface{}) (sql.Result, error) {
+ return q.scanOrExec(ctx, dest, len(dest) > 0)
+}
+
+func (q *DeleteQuery) scanOrExec(
+ ctx context.Context, dest []interface{}, hasDest bool,
+) (sql.Result, error) {
if q.err != nil {
return nil, q.err
}
@@ -234,25 +254,33 @@ func (q *DeleteQuery) Exec(ctx context.Context, dest ...interface{}) (sql.Result
}
}
+ // Run append model hooks before generating the query.
if err := q.beforeAppendModel(ctx, q); err != nil {
return nil, err
}
+ // Generate the query before checking hasReturning.
queryBytes, err := q.AppendQuery(q.db.fmter, q.db.makeQueryBytes())
if err != nil {
return nil, err
}
- query := internal.String(queryBytes)
-
- var res sql.Result
+ useScan := hasDest || (q.hasReturning() && q.hasFeature(feature.Returning|feature.Output))
+ var model Model
- if hasDest := len(dest) > 0; hasDest || q.hasReturning() {
- model, err := q.getModel(dest)
+ if useScan {
+ var err error
+ model, err = q.getModel(dest)
if err != nil {
return nil, err
}
+ }
+
+ query := internal.String(queryBytes)
+
+ var res sql.Result
+ if useScan {
res, err = q.scan(ctx, q, query, model, hasDest)
if err != nil {
return nil, err