summaryrefslogtreecommitdiff
path: root/vendor/github.com/uptrace/bun/query_insert.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_insert.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_insert.go')
-rw-r--r--vendor/github.com/uptrace/bun/query_insert.go51
1 files changed, 39 insertions, 12 deletions
diff --git a/vendor/github.com/uptrace/bun/query_insert.go b/vendor/github.com/uptrace/bun/query_insert.go
index 7fb358de8..7cf053756 100644
--- a/vendor/github.com/uptrace/bun/query_insert.go
+++ b/vendor/github.com/uptrace/bun/query_insert.go
@@ -48,9 +48,17 @@ func (q *InsertQuery) Model(model interface{}) *InsertQuery {
return q
}
+func (q *InsertQuery) Err(err error) *InsertQuery {
+ q.setErr(err)
+ return q
+}
+
// Apply calls the fn passing the SelectQuery as an argument.
func (q *InsertQuery) Apply(fn func(*InsertQuery) *InsertQuery) *InsertQuery {
- return fn(q)
+ if fn != nil {
+ return fn(q)
+ }
+ return q
}
func (q *InsertQuery) With(name string, query schema.QueryAppender) *InsertQuery {
@@ -189,7 +197,7 @@ func (q *InsertQuery) AppendQuery(fmter schema.Formatter, b []byte) (_ []byte, e
return nil, err
}
- b, err = q.appendColumnsValues(fmter, b)
+ b, err = q.appendColumnsValues(fmter, b, false)
if err != nil {
return nil, err
}
@@ -211,7 +219,7 @@ func (q *InsertQuery) AppendQuery(fmter schema.Formatter, b []byte) (_ []byte, e
}
func (q *InsertQuery) appendColumnsValues(
- fmter schema.Formatter, b []byte,
+ fmter schema.Formatter, b []byte, skipOutput bool,
) (_ []byte, err error) {
if q.hasMultiTables() {
if q.columns != nil {
@@ -272,7 +280,7 @@ func (q *InsertQuery) appendColumnsValues(
b = q.appendFields(fmter, b, fields)
b = append(b, ")"...)
- if q.hasFeature(feature.Output) && q.hasReturning() {
+ if q.hasFeature(feature.Output) && q.hasReturning() && !skipOutput {
b = append(b, " OUTPUT "...)
b, err = q.appendOutput(fmter, b)
if err != nil {
@@ -534,35 +542,54 @@ func (q *InsertQuery) appendSetValues(b []byte, fields []*schema.Field) []byte {
//------------------------------------------------------------------------------
+func (q *InsertQuery) Scan(ctx context.Context, dest ...interface{}) error {
+ _, err := q.scanOrExec(ctx, dest, true)
+ return err
+}
+
func (q *InsertQuery) Exec(ctx context.Context, dest ...interface{}) (sql.Result, error) {
+ return q.scanOrExec(ctx, dest, len(dest) > 0)
+}
+
+func (q *InsertQuery) scanOrExec(
+ ctx context.Context, dest []interface{}, hasDest bool,
+) (sql.Result, error) {
+ if q.err != nil {
+ return nil, q.err
+ }
+
if q.table != nil {
if err := q.beforeInsertHook(ctx); err != nil {
return nil, err
}
}
- if q.err != nil {
- return nil, q.err
- }
+ // 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.InsertReturning|feature.Output))
+ var model Model
- if hasDest := len(dest) > 0; hasDest ||
- (q.hasReturning() && q.hasFeature(feature.InsertReturning|feature.Output)) {
- 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