diff options
author | 2023-01-22 12:26:47 +0100 | |
---|---|---|
committer | 2023-01-22 12:26:47 +0100 | |
commit | 0ceacd7b1d0b03e3da1d552aa0bbbe7037e57e90 (patch) | |
tree | 106bd6f80fe5466522b37b5778ebc7c4fd6ab728 /vendor/github.com/uptrace/bun/query_insert.go | |
parent | [chore] Add name to instance field for autosuggestion (#1359) (diff) | |
download | gotosocial-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.go | 51 |
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 |