mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2024-12-14 19:26:43 +00:00
ec325fee14
* [chore] Update a bunch of database dependencies * fix lil thing
98 lines
1.7 KiB
Go
98 lines
1.7 KiB
Go
package bun
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
|
|
"github.com/uptrace/bun/schema"
|
|
)
|
|
|
|
type RawQuery struct {
|
|
baseQuery
|
|
|
|
query string
|
|
args []interface{}
|
|
}
|
|
|
|
// Deprecated: Use NewRaw instead. When add it to IDB, it conflicts with the sql.Conn#Raw
|
|
func (db *DB) Raw(query string, args ...interface{}) *RawQuery {
|
|
return &RawQuery{
|
|
baseQuery: baseQuery{
|
|
db: db,
|
|
conn: db.DB,
|
|
},
|
|
query: query,
|
|
args: args,
|
|
}
|
|
}
|
|
|
|
func NewRawQuery(db *DB, query string, args ...interface{}) *RawQuery {
|
|
return &RawQuery{
|
|
baseQuery: baseQuery{
|
|
db: db,
|
|
conn: db.DB,
|
|
},
|
|
query: query,
|
|
args: args,
|
|
}
|
|
}
|
|
|
|
func (q *RawQuery) Conn(db IConn) *RawQuery {
|
|
q.setConn(db)
|
|
return q
|
|
}
|
|
|
|
func (q *RawQuery) Err(err error) *RawQuery {
|
|
q.setErr(err)
|
|
return q
|
|
}
|
|
|
|
func (q *RawQuery) Exec(ctx context.Context, dest ...interface{}) (sql.Result, error) {
|
|
return q.scanOrExec(ctx, dest, len(dest) > 0)
|
|
}
|
|
|
|
func (q *RawQuery) Scan(ctx context.Context, dest ...interface{}) error {
|
|
_, err := q.scanOrExec(ctx, dest, true)
|
|
return err
|
|
}
|
|
|
|
func (q *RawQuery) scanOrExec(
|
|
ctx context.Context, dest []interface{}, hasDest bool,
|
|
) (sql.Result, error) {
|
|
if q.err != nil {
|
|
return nil, q.err
|
|
}
|
|
|
|
var model Model
|
|
var err error
|
|
|
|
if hasDest {
|
|
model, err = q.getModel(dest)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
query := q.db.format(q.query, q.args)
|
|
var res sql.Result
|
|
|
|
if hasDest {
|
|
res, err = q.scan(ctx, q, query, model, hasDest)
|
|
} else {
|
|
res, err = q.exec(ctx, q, query)
|
|
}
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return res, nil
|
|
}
|
|
|
|
func (q *RawQuery) AppendQuery(fmter schema.Formatter, b []byte) ([]byte, error) {
|
|
return fmter.AppendQuery(b, q.query, q.args...), nil
|
|
}
|
|
|
|
func (q *RawQuery) Operation() string {
|
|
return "SELECT"
|
|
}
|