1042 lines
20 KiB
Go
1042 lines
20 KiB
Go
|
//lint:file-ignore * generated file
|
||
|
|
// AUTOGENERATED BY storj.io/dbx
|
||
|
|
// DO NOT EDIT.
|
||
|
|
|
||
|
|
package dbx
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"context"
|
||
|
|
"database/sql"
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
"reflect"
|
||
|
|
"strconv"
|
||
|
|
"strings"
|
||
|
|
"sync"
|
||
|
|
"time"
|
||
|
|
"unicode"
|
||
|
|
|
||
|
|
"github.com/jackc/pgconn"
|
||
|
|
_ "github.com/jackc/pgx/v4/stdlib"
|
||
|
|
|
||
|
|
"storj.io/storj/private/tagsql"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Prevent conditional imports from causing build failures
|
||
|
|
var _ = strconv.Itoa
|
||
|
|
var _ = strings.LastIndex
|
||
|
|
var _ = fmt.Sprint
|
||
|
|
var _ sync.Mutex
|
||
|
|
|
||
|
|
var (
|
||
|
|
WrapErr = func(err *Error) error { return err }
|
||
|
|
Logger func(format string, args ...interface{})
|
||
|
|
|
||
|
|
errTooManyRows = errors.New("too many rows")
|
||
|
|
errUnsupportedDriver = errors.New("unsupported driver")
|
||
|
|
errEmptyUpdate = errors.New("empty update")
|
||
|
|
)
|
||
|
|
|
||
|
|
func logError(format string, args ...interface{}) {
|
||
|
|
if Logger != nil {
|
||
|
|
Logger(format, args...)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type ErrorCode int
|
||
|
|
|
||
|
|
const (
|
||
|
|
ErrorCode_Unknown ErrorCode = iota
|
||
|
|
ErrorCode_UnsupportedDriver
|
||
|
|
ErrorCode_NoRows
|
||
|
|
ErrorCode_TxDone
|
||
|
|
ErrorCode_TooManyRows
|
||
|
|
ErrorCode_ConstraintViolation
|
||
|
|
ErrorCode_EmptyUpdate
|
||
|
|
)
|
||
|
|
|
||
|
|
type Error struct {
|
||
|
|
Err error
|
||
|
|
Code ErrorCode
|
||
|
|
Driver string
|
||
|
|
Constraint string
|
||
|
|
QuerySuffix string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e *Error) Error() string {
|
||
|
|
return e.Err.Error()
|
||
|
|
}
|
||
|
|
|
||
|
|
func wrapErr(e *Error) error {
|
||
|
|
if WrapErr == nil {
|
||
|
|
return e
|
||
|
|
}
|
||
|
|
return WrapErr(e)
|
||
|
|
}
|
||
|
|
|
||
|
|
func makeErr(err error) error {
|
||
|
|
if err == nil {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
e := &Error{Err: err}
|
||
|
|
switch err {
|
||
|
|
case sql.ErrNoRows:
|
||
|
|
e.Code = ErrorCode_NoRows
|
||
|
|
case sql.ErrTxDone:
|
||
|
|
e.Code = ErrorCode_TxDone
|
||
|
|
}
|
||
|
|
return wrapErr(e)
|
||
|
|
}
|
||
|
|
|
||
|
|
func unsupportedDriver(driver string) error {
|
||
|
|
return wrapErr(&Error{
|
||
|
|
Err: errUnsupportedDriver,
|
||
|
|
Code: ErrorCode_UnsupportedDriver,
|
||
|
|
Driver: driver,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func emptyUpdate() error {
|
||
|
|
return wrapErr(&Error{
|
||
|
|
Err: errEmptyUpdate,
|
||
|
|
Code: ErrorCode_EmptyUpdate,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func tooManyRows(query_suffix string) error {
|
||
|
|
return wrapErr(&Error{
|
||
|
|
Err: errTooManyRows,
|
||
|
|
Code: ErrorCode_TooManyRows,
|
||
|
|
QuerySuffix: query_suffix,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func constraintViolation(err error, constraint string) error {
|
||
|
|
return wrapErr(&Error{
|
||
|
|
Err: err,
|
||
|
|
Code: ErrorCode_ConstraintViolation,
|
||
|
|
Constraint: constraint,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
type driver interface {
|
||
|
|
ExecContext(ctx context.Context, query string, args ...interface{}) (sql.Result, error)
|
||
|
|
QueryContext(ctx context.Context, query string, args ...interface{}) (tagsql.Rows, error)
|
||
|
|
QueryRowContext(ctx context.Context, query string, args ...interface{}) *sql.Row
|
||
|
|
}
|
||
|
|
|
||
|
|
var (
|
||
|
|
notAPointer = errors.New("destination not a pointer")
|
||
|
|
lossyConversion = errors.New("lossy conversion")
|
||
|
|
)
|
||
|
|
|
||
|
|
type DB struct {
|
||
|
|
tagsql.DB
|
||
|
|
dbMethods
|
||
|
|
|
||
|
|
Hooks struct {
|
||
|
|
Now func() time.Time
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func Open(driver, source string) (db *DB, err error) {
|
||
|
|
var sql_db *sql.DB
|
||
|
|
switch driver {
|
||
|
|
case "pgx":
|
||
|
|
sql_db, err = openpgx(source)
|
||
|
|
default:
|
||
|
|
return nil, unsupportedDriver(driver)
|
||
|
|
}
|
||
|
|
if err != nil {
|
||
|
|
return nil, makeErr(err)
|
||
|
|
}
|
||
|
|
defer func(sql_db *sql.DB) {
|
||
|
|
if err != nil {
|
||
|
|
sql_db.Close()
|
||
|
|
}
|
||
|
|
}(sql_db)
|
||
|
|
|
||
|
|
if err := sql_db.Ping(); err != nil {
|
||
|
|
return nil, makeErr(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
db = &DB{
|
||
|
|
DB: tagsql.Wrap(sql_db),
|
||
|
|
}
|
||
|
|
db.Hooks.Now = time.Now
|
||
|
|
|
||
|
|
switch driver {
|
||
|
|
case "pgx":
|
||
|
|
db.dbMethods = newpgx(db)
|
||
|
|
default:
|
||
|
|
return nil, unsupportedDriver(driver)
|
||
|
|
}
|
||
|
|
|
||
|
|
return db, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (obj *DB) Close() (err error) {
|
||
|
|
return obj.makeErr(obj.DB.Close())
|
||
|
|
}
|
||
|
|
|
||
|
|
func (obj *DB) Open(ctx context.Context) (*Tx, error) {
|
||
|
|
tx, err := obj.DB.BeginTx(ctx, nil)
|
||
|
|
if err != nil {
|
||
|
|
return nil, obj.makeErr(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return &Tx{
|
||
|
|
Tx: tx,
|
||
|
|
txMethods: obj.wrapTx(tx),
|
||
|
|
}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (obj *DB) NewRx() *Rx {
|
||
|
|
return &Rx{db: obj}
|
||
|
|
}
|
||
|
|
|
||
|
|
func DeleteAll(ctx context.Context, db *DB) (int64, error) {
|
||
|
|
tx, err := db.Open(ctx)
|
||
|
|
if err != nil {
|
||
|
|
return 0, err
|
||
|
|
}
|
||
|
|
defer func() {
|
||
|
|
if err == nil {
|
||
|
|
err = db.makeErr(tx.Commit())
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
if err_rollback := tx.Rollback(); err_rollback != nil {
|
||
|
|
logError("delete-all: rollback failed: %v", db.makeErr(err_rollback))
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
return tx.deleteAll(ctx)
|
||
|
|
}
|
||
|
|
|
||
|
|
type Tx struct {
|
||
|
|
Tx tagsql.Tx
|
||
|
|
txMethods
|
||
|
|
}
|
||
|
|
|
||
|
|
type dialectTx struct {
|
||
|
|
tx tagsql.Tx
|
||
|
|
}
|
||
|
|
|
||
|
|
func (tx *dialectTx) Commit() (err error) {
|
||
|
|
return makeErr(tx.tx.Commit())
|
||
|
|
}
|
||
|
|
|
||
|
|
func (tx *dialectTx) Rollback() (err error) {
|
||
|
|
return makeErr(tx.tx.Rollback())
|
||
|
|
}
|
||
|
|
|
||
|
|
type pgxImpl struct {
|
||
|
|
db *DB
|
||
|
|
dialect __sqlbundle_pgx
|
||
|
|
driver driver
|
||
|
|
}
|
||
|
|
|
||
|
|
func (obj *pgxImpl) Rebind(s string) string {
|
||
|
|
return obj.dialect.Rebind(s)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (obj *pgxImpl) logStmt(stmt string, args ...interface{}) {
|
||
|
|
pgxLogStmt(stmt, args...)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (obj *pgxImpl) makeErr(err error) error {
|
||
|
|
constraint, ok := obj.isConstraintError(err)
|
||
|
|
if ok {
|
||
|
|
return constraintViolation(err, constraint)
|
||
|
|
}
|
||
|
|
return makeErr(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
type pgxDB struct {
|
||
|
|
db *DB
|
||
|
|
*pgxImpl
|
||
|
|
}
|
||
|
|
|
||
|
|
func newpgx(db *DB) *pgxDB {
|
||
|
|
return &pgxDB{
|
||
|
|
db: db,
|
||
|
|
pgxImpl: &pgxImpl{
|
||
|
|
db: db,
|
||
|
|
driver: db.DB,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (obj *pgxDB) Schema() string {
|
||
|
|
return `CREATE TABLE nodes (
|
||
|
|
id bytea NOT NULL,
|
||
|
|
name text NOT NULL,
|
||
|
|
api_secret bytea NOT NULL,
|
||
|
|
logo bytea NOT NULL,
|
||
|
|
PRIMARY KEY ( id )
|
||
|
|
);`
|
||
|
|
}
|
||
|
|
|
||
|
|
func (obj *pgxDB) wrapTx(tx tagsql.Tx) txMethods {
|
||
|
|
return &pgxTx{
|
||
|
|
dialectTx: dialectTx{tx: tx},
|
||
|
|
pgxImpl: &pgxImpl{
|
||
|
|
db: obj.db,
|
||
|
|
driver: tx,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type pgxTx struct {
|
||
|
|
dialectTx
|
||
|
|
*pgxImpl
|
||
|
|
}
|
||
|
|
|
||
|
|
func pgxLogStmt(stmt string, args ...interface{}) {
|
||
|
|
// TODO: render placeholders
|
||
|
|
if Logger != nil {
|
||
|
|
out := fmt.Sprintf("stmt: %s\nargs: %v\n", stmt, pretty(args))
|
||
|
|
Logger(out)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type pretty []interface{}
|
||
|
|
|
||
|
|
func (p pretty) Format(f fmt.State, c rune) {
|
||
|
|
fmt.Fprint(f, "[")
|
||
|
|
nextval:
|
||
|
|
for i, val := range p {
|
||
|
|
if i > 0 {
|
||
|
|
fmt.Fprint(f, ", ")
|
||
|
|
}
|
||
|
|
rv := reflect.ValueOf(val)
|
||
|
|
if rv.Kind() == reflect.Ptr {
|
||
|
|
if rv.IsNil() {
|
||
|
|
fmt.Fprint(f, "NULL")
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
val = rv.Elem().Interface()
|
||
|
|
}
|
||
|
|
switch v := val.(type) {
|
||
|
|
case string:
|
||
|
|
fmt.Fprintf(f, "%q", v)
|
||
|
|
case time.Time:
|
||
|
|
fmt.Fprintf(f, "%s", v.Format(time.RFC3339Nano))
|
||
|
|
case []byte:
|
||
|
|
for _, b := range v {
|
||
|
|
if !unicode.IsPrint(rune(b)) {
|
||
|
|
fmt.Fprintf(f, "%#x", v)
|
||
|
|
continue nextval
|
||
|
|
}
|
||
|
|
}
|
||
|
|
fmt.Fprintf(f, "%q", v)
|
||
|
|
default:
|
||
|
|
fmt.Fprintf(f, "%v", v)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
fmt.Fprint(f, "]")
|
||
|
|
}
|
||
|
|
|
||
|
|
type Node struct {
|
||
|
|
Id []byte
|
||
|
|
Name string
|
||
|
|
ApiSecret []byte
|
||
|
|
Logo []byte
|
||
|
|
}
|
||
|
|
|
||
|
|
func (Node) _Table() string { return "nodes" }
|
||
|
|
|
||
|
|
type Node_Update_Fields struct {
|
||
|
|
Name Node_Name_Field
|
||
|
|
Logo Node_Logo_Field
|
||
|
|
}
|
||
|
|
|
||
|
|
type Node_Id_Field struct {
|
||
|
|
_set bool
|
||
|
|
_null bool
|
||
|
|
_value []byte
|
||
|
|
}
|
||
|
|
|
||
|
|
func Node_Id(v []byte) Node_Id_Field {
|
||
|
|
return Node_Id_Field{_set: true, _value: v}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f Node_Id_Field) value() interface{} {
|
||
|
|
if !f._set || f._null {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
return f._value
|
||
|
|
}
|
||
|
|
|
||
|
|
func (Node_Id_Field) _Column() string { return "id" }
|
||
|
|
|
||
|
|
type Node_Name_Field struct {
|
||
|
|
_set bool
|
||
|
|
_null bool
|
||
|
|
_value string
|
||
|
|
}
|
||
|
|
|
||
|
|
func Node_Name(v string) Node_Name_Field {
|
||
|
|
return Node_Name_Field{_set: true, _value: v}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f Node_Name_Field) value() interface{} {
|
||
|
|
if !f._set || f._null {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
return f._value
|
||
|
|
}
|
||
|
|
|
||
|
|
func (Node_Name_Field) _Column() string { return "name" }
|
||
|
|
|
||
|
|
type Node_ApiSecret_Field struct {
|
||
|
|
_set bool
|
||
|
|
_null bool
|
||
|
|
_value []byte
|
||
|
|
}
|
||
|
|
|
||
|
|
func Node_ApiSecret(v []byte) Node_ApiSecret_Field {
|
||
|
|
return Node_ApiSecret_Field{_set: true, _value: v}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f Node_ApiSecret_Field) value() interface{} {
|
||
|
|
if !f._set || f._null {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
return f._value
|
||
|
|
}
|
||
|
|
|
||
|
|
func (Node_ApiSecret_Field) _Column() string { return "api_secret" }
|
||
|
|
|
||
|
|
type Node_Logo_Field struct {
|
||
|
|
_set bool
|
||
|
|
_null bool
|
||
|
|
_value []byte
|
||
|
|
}
|
||
|
|
|
||
|
|
func Node_Logo(v []byte) Node_Logo_Field {
|
||
|
|
return Node_Logo_Field{_set: true, _value: v}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f Node_Logo_Field) value() interface{} {
|
||
|
|
if !f._set || f._null {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
return f._value
|
||
|
|
}
|
||
|
|
|
||
|
|
func (Node_Logo_Field) _Column() string { return "logo" }
|
||
|
|
|
||
|
|
func toUTC(t time.Time) time.Time {
|
||
|
|
return t.UTC()
|
||
|
|
}
|
||
|
|
|
||
|
|
func toDate(t time.Time) time.Time {
|
||
|
|
// keep up the minute portion so that translations between timezones will
|
||
|
|
// continue to reflect properly.
|
||
|
|
return t.Truncate(time.Minute)
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// runtime support for building sql statements
|
||
|
|
//
|
||
|
|
|
||
|
|
type __sqlbundle_SQL interface {
|
||
|
|
Render() string
|
||
|
|
|
||
|
|
private()
|
||
|
|
}
|
||
|
|
|
||
|
|
type __sqlbundle_Dialect interface {
|
||
|
|
Rebind(sql string) string
|
||
|
|
}
|
||
|
|
|
||
|
|
type __sqlbundle_RenderOp int
|
||
|
|
|
||
|
|
const (
|
||
|
|
__sqlbundle_NoFlatten __sqlbundle_RenderOp = iota
|
||
|
|
__sqlbundle_NoTerminate
|
||
|
|
)
|
||
|
|
|
||
|
|
func __sqlbundle_Render(dialect __sqlbundle_Dialect, sql __sqlbundle_SQL, ops ...__sqlbundle_RenderOp) string {
|
||
|
|
out := sql.Render()
|
||
|
|
|
||
|
|
flatten := true
|
||
|
|
terminate := true
|
||
|
|
for _, op := range ops {
|
||
|
|
switch op {
|
||
|
|
case __sqlbundle_NoFlatten:
|
||
|
|
flatten = false
|
||
|
|
case __sqlbundle_NoTerminate:
|
||
|
|
terminate = false
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if flatten {
|
||
|
|
out = __sqlbundle_flattenSQL(out)
|
||
|
|
}
|
||
|
|
if terminate {
|
||
|
|
out += ";"
|
||
|
|
}
|
||
|
|
|
||
|
|
return dialect.Rebind(out)
|
||
|
|
}
|
||
|
|
|
||
|
|
func __sqlbundle_flattenSQL(x string) string {
|
||
|
|
// trim whitespace from beginning and end
|
||
|
|
s, e := 0, len(x)-1
|
||
|
|
for s < len(x) && (x[s] == ' ' || x[s] == '\t' || x[s] == '\n') {
|
||
|
|
s++
|
||
|
|
}
|
||
|
|
for s <= e && (x[e] == ' ' || x[e] == '\t' || x[e] == '\n') {
|
||
|
|
e--
|
||
|
|
}
|
||
|
|
if s > e {
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
x = x[s : e+1]
|
||
|
|
|
||
|
|
// check for whitespace that needs fixing
|
||
|
|
wasSpace := false
|
||
|
|
for i := 0; i < len(x); i++ {
|
||
|
|
r := x[i]
|
||
|
|
justSpace := r == ' '
|
||
|
|
if (wasSpace && justSpace) || r == '\t' || r == '\n' {
|
||
|
|
// whitespace detected, start writing a new string
|
||
|
|
var result strings.Builder
|
||
|
|
result.Grow(len(x))
|
||
|
|
if wasSpace {
|
||
|
|
result.WriteString(x[:i-1])
|
||
|
|
} else {
|
||
|
|
result.WriteString(x[:i])
|
||
|
|
}
|
||
|
|
for p := i; p < len(x); p++ {
|
||
|
|
for p < len(x) && (x[p] == ' ' || x[p] == '\t' || x[p] == '\n') {
|
||
|
|
p++
|
||
|
|
}
|
||
|
|
result.WriteByte(' ')
|
||
|
|
|
||
|
|
start := p
|
||
|
|
for p < len(x) && !(x[p] == ' ' || x[p] == '\t' || x[p] == '\n') {
|
||
|
|
p++
|
||
|
|
}
|
||
|
|
result.WriteString(x[start:p])
|
||
|
|
}
|
||
|
|
|
||
|
|
return result.String()
|
||
|
|
}
|
||
|
|
wasSpace = justSpace
|
||
|
|
}
|
||
|
|
|
||
|
|
// no problematic whitespace found
|
||
|
|
return x
|
||
|
|
}
|
||
|
|
|
||
|
|
// this type is specially named to match up with the name returned by the
|
||
|
|
// dialect impl in the sql package.
|
||
|
|
type __sqlbundle_postgres struct{}
|
||
|
|
|
||
|
|
func (p __sqlbundle_postgres) Rebind(sql string) string {
|
||
|
|
type sqlParseState int
|
||
|
|
const (
|
||
|
|
sqlParseStart sqlParseState = iota
|
||
|
|
sqlParseInStringLiteral
|
||
|
|
sqlParseInQuotedIdentifier
|
||
|
|
sqlParseInComment
|
||
|
|
)
|
||
|
|
|
||
|
|
out := make([]byte, 0, len(sql)+10)
|
||
|
|
|
||
|
|
j := 1
|
||
|
|
state := sqlParseStart
|
||
|
|
for i := 0; i < len(sql); i++ {
|
||
|
|
ch := sql[i]
|
||
|
|
switch state {
|
||
|
|
case sqlParseStart:
|
||
|
|
switch ch {
|
||
|
|
case '?':
|
||
|
|
out = append(out, '$')
|
||
|
|
out = append(out, strconv.Itoa(j)...)
|
||
|
|
state = sqlParseStart
|
||
|
|
j++
|
||
|
|
continue
|
||
|
|
case '-':
|
||
|
|
if i+1 < len(sql) && sql[i+1] == '-' {
|
||
|
|
state = sqlParseInComment
|
||
|
|
}
|
||
|
|
case '"':
|
||
|
|
state = sqlParseInQuotedIdentifier
|
||
|
|
case '\'':
|
||
|
|
state = sqlParseInStringLiteral
|
||
|
|
}
|
||
|
|
case sqlParseInStringLiteral:
|
||
|
|
if ch == '\'' {
|
||
|
|
state = sqlParseStart
|
||
|
|
}
|
||
|
|
case sqlParseInQuotedIdentifier:
|
||
|
|
if ch == '"' {
|
||
|
|
state = sqlParseStart
|
||
|
|
}
|
||
|
|
case sqlParseInComment:
|
||
|
|
if ch == '\n' {
|
||
|
|
state = sqlParseStart
|
||
|
|
}
|
||
|
|
}
|
||
|
|
out = append(out, ch)
|
||
|
|
}
|
||
|
|
|
||
|
|
return string(out)
|
||
|
|
}
|
||
|
|
|
||
|
|
// this type is specially named to match up with the name returned by the
|
||
|
|
// dialect impl in the sql package.
|
||
|
|
type __sqlbundle_sqlite3 struct{}
|
||
|
|
|
||
|
|
func (s __sqlbundle_sqlite3) Rebind(sql string) string {
|
||
|
|
return sql
|
||
|
|
}
|
||
|
|
|
||
|
|
// this type is specially named to match up with the name returned by the
|
||
|
|
// dialect impl in the sql package.
|
||
|
|
type __sqlbundle_cockroach struct{}
|
||
|
|
|
||
|
|
func (p __sqlbundle_cockroach) Rebind(sql string) string {
|
||
|
|
type sqlParseState int
|
||
|
|
const (
|
||
|
|
sqlParseStart sqlParseState = iota
|
||
|
|
sqlParseInStringLiteral
|
||
|
|
sqlParseInQuotedIdentifier
|
||
|
|
sqlParseInComment
|
||
|
|
)
|
||
|
|
|
||
|
|
out := make([]byte, 0, len(sql)+10)
|
||
|
|
|
||
|
|
j := 1
|
||
|
|
state := sqlParseStart
|
||
|
|
for i := 0; i < len(sql); i++ {
|
||
|
|
ch := sql[i]
|
||
|
|
switch state {
|
||
|
|
case sqlParseStart:
|
||
|
|
switch ch {
|
||
|
|
case '?':
|
||
|
|
out = append(out, '$')
|
||
|
|
out = append(out, strconv.Itoa(j)...)
|
||
|
|
state = sqlParseStart
|
||
|
|
j++
|
||
|
|
continue
|
||
|
|
case '-':
|
||
|
|
if i+1 < len(sql) && sql[i+1] == '-' {
|
||
|
|
state = sqlParseInComment
|
||
|
|
}
|
||
|
|
case '"':
|
||
|
|
state = sqlParseInQuotedIdentifier
|
||
|
|
case '\'':
|
||
|
|
state = sqlParseInStringLiteral
|
||
|
|
}
|
||
|
|
case sqlParseInStringLiteral:
|
||
|
|
if ch == '\'' {
|
||
|
|
state = sqlParseStart
|
||
|
|
}
|
||
|
|
case sqlParseInQuotedIdentifier:
|
||
|
|
if ch == '"' {
|
||
|
|
state = sqlParseStart
|
||
|
|
}
|
||
|
|
case sqlParseInComment:
|
||
|
|
if ch == '\n' {
|
||
|
|
state = sqlParseStart
|
||
|
|
}
|
||
|
|
}
|
||
|
|
out = append(out, ch)
|
||
|
|
}
|
||
|
|
|
||
|
|
return string(out)
|
||
|
|
}
|
||
|
|
|
||
|
|
// this type is specially named to match up with the name returned by the
|
||
|
|
// dialect impl in the sql package.
|
||
|
|
type __sqlbundle_pgx struct{}
|
||
|
|
|
||
|
|
func (p __sqlbundle_pgx) Rebind(sql string) string {
|
||
|
|
type sqlParseState int
|
||
|
|
const (
|
||
|
|
sqlParseStart sqlParseState = iota
|
||
|
|
sqlParseInStringLiteral
|
||
|
|
sqlParseInQuotedIdentifier
|
||
|
|
sqlParseInComment
|
||
|
|
)
|
||
|
|
|
||
|
|
out := make([]byte, 0, len(sql)+10)
|
||
|
|
|
||
|
|
j := 1
|
||
|
|
state := sqlParseStart
|
||
|
|
for i := 0; i < len(sql); i++ {
|
||
|
|
ch := sql[i]
|
||
|
|
switch state {
|
||
|
|
case sqlParseStart:
|
||
|
|
switch ch {
|
||
|
|
case '?':
|
||
|
|
out = append(out, '$')
|
||
|
|
out = append(out, strconv.Itoa(j)...)
|
||
|
|
state = sqlParseStart
|
||
|
|
j++
|
||
|
|
continue
|
||
|
|
case '-':
|
||
|
|
if i+1 < len(sql) && sql[i+1] == '-' {
|
||
|
|
state = sqlParseInComment
|
||
|
|
}
|
||
|
|
case '"':
|
||
|
|
state = sqlParseInQuotedIdentifier
|
||
|
|
case '\'':
|
||
|
|
state = sqlParseInStringLiteral
|
||
|
|
}
|
||
|
|
case sqlParseInStringLiteral:
|
||
|
|
if ch == '\'' {
|
||
|
|
state = sqlParseStart
|
||
|
|
}
|
||
|
|
case sqlParseInQuotedIdentifier:
|
||
|
|
if ch == '"' {
|
||
|
|
state = sqlParseStart
|
||
|
|
}
|
||
|
|
case sqlParseInComment:
|
||
|
|
if ch == '\n' {
|
||
|
|
state = sqlParseStart
|
||
|
|
}
|
||
|
|
}
|
||
|
|
out = append(out, ch)
|
||
|
|
}
|
||
|
|
|
||
|
|
return string(out)
|
||
|
|
}
|
||
|
|
|
||
|
|
// this type is specially named to match up with the name returned by the
|
||
|
|
// dialect impl in the sql package.
|
||
|
|
type __sqlbundle_pgxcockroach struct{}
|
||
|
|
|
||
|
|
func (p __sqlbundle_pgxcockroach) Rebind(sql string) string {
|
||
|
|
type sqlParseState int
|
||
|
|
const (
|
||
|
|
sqlParseStart sqlParseState = iota
|
||
|
|
sqlParseInStringLiteral
|
||
|
|
sqlParseInQuotedIdentifier
|
||
|
|
sqlParseInComment
|
||
|
|
)
|
||
|
|
|
||
|
|
out := make([]byte, 0, len(sql)+10)
|
||
|
|
|
||
|
|
j := 1
|
||
|
|
state := sqlParseStart
|
||
|
|
for i := 0; i < len(sql); i++ {
|
||
|
|
ch := sql[i]
|
||
|
|
switch state {
|
||
|
|
case sqlParseStart:
|
||
|
|
switch ch {
|
||
|
|
case '?':
|
||
|
|
out = append(out, '$')
|
||
|
|
out = append(out, strconv.Itoa(j)...)
|
||
|
|
state = sqlParseStart
|
||
|
|
j++
|
||
|
|
continue
|
||
|
|
case '-':
|
||
|
|
if i+1 < len(sql) && sql[i+1] == '-' {
|
||
|
|
state = sqlParseInComment
|
||
|
|
}
|
||
|
|
case '"':
|
||
|
|
state = sqlParseInQuotedIdentifier
|
||
|
|
case '\'':
|
||
|
|
state = sqlParseInStringLiteral
|
||
|
|
}
|
||
|
|
case sqlParseInStringLiteral:
|
||
|
|
if ch == '\'' {
|
||
|
|
state = sqlParseStart
|
||
|
|
}
|
||
|
|
case sqlParseInQuotedIdentifier:
|
||
|
|
if ch == '"' {
|
||
|
|
state = sqlParseStart
|
||
|
|
}
|
||
|
|
case sqlParseInComment:
|
||
|
|
if ch == '\n' {
|
||
|
|
state = sqlParseStart
|
||
|
|
}
|
||
|
|
}
|
||
|
|
out = append(out, ch)
|
||
|
|
}
|
||
|
|
|
||
|
|
return string(out)
|
||
|
|
}
|
||
|
|
|
||
|
|
type __sqlbundle_Literal string
|
||
|
|
|
||
|
|
func (__sqlbundle_Literal) private() {}
|
||
|
|
|
||
|
|
func (l __sqlbundle_Literal) Render() string { return string(l) }
|
||
|
|
|
||
|
|
type __sqlbundle_Literals struct {
|
||
|
|
Join string
|
||
|
|
SQLs []__sqlbundle_SQL
|
||
|
|
}
|
||
|
|
|
||
|
|
func (__sqlbundle_Literals) private() {}
|
||
|
|
|
||
|
|
func (l __sqlbundle_Literals) Render() string {
|
||
|
|
var out bytes.Buffer
|
||
|
|
|
||
|
|
first := true
|
||
|
|
for _, sql := range l.SQLs {
|
||
|
|
if sql == nil {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
if !first {
|
||
|
|
out.WriteString(l.Join)
|
||
|
|
}
|
||
|
|
first = false
|
||
|
|
out.WriteString(sql.Render())
|
||
|
|
}
|
||
|
|
|
||
|
|
return out.String()
|
||
|
|
}
|
||
|
|
|
||
|
|
type __sqlbundle_Condition struct {
|
||
|
|
// set at compile/embed time
|
||
|
|
Name string
|
||
|
|
Left string
|
||
|
|
Equal bool
|
||
|
|
Right string
|
||
|
|
|
||
|
|
// set at runtime
|
||
|
|
Null bool
|
||
|
|
}
|
||
|
|
|
||
|
|
func (*__sqlbundle_Condition) private() {}
|
||
|
|
|
||
|
|
func (c *__sqlbundle_Condition) Render() string {
|
||
|
|
// TODO(jeff): maybe check if we can use placeholders instead of the
|
||
|
|
// literal null: this would make the templates easier.
|
||
|
|
|
||
|
|
switch {
|
||
|
|
case c.Equal && c.Null:
|
||
|
|
return c.Left + " is null"
|
||
|
|
case c.Equal && !c.Null:
|
||
|
|
return c.Left + " = " + c.Right
|
||
|
|
case !c.Equal && c.Null:
|
||
|
|
return c.Left + " is not null"
|
||
|
|
case !c.Equal && !c.Null:
|
||
|
|
return c.Left + " != " + c.Right
|
||
|
|
default:
|
||
|
|
panic("unhandled case")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type __sqlbundle_Hole struct {
|
||
|
|
// set at compiile/embed time
|
||
|
|
Name string
|
||
|
|
|
||
|
|
// set at runtime or possibly embed time
|
||
|
|
SQL __sqlbundle_SQL
|
||
|
|
}
|
||
|
|
|
||
|
|
func (*__sqlbundle_Hole) private() {}
|
||
|
|
|
||
|
|
func (h *__sqlbundle_Hole) Render() string {
|
||
|
|
if h.SQL == nil {
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
return h.SQL.Render()
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// end runtime support for building sql statements
|
||
|
|
//
|
||
|
|
|
||
|
|
func (obj *pgxImpl) Create_Node(ctx context.Context,
|
||
|
|
node_id Node_Id_Field,
|
||
|
|
node_name Node_Name_Field,
|
||
|
|
node_api_secret Node_ApiSecret_Field,
|
||
|
|
node_logo Node_Logo_Field) (
|
||
|
|
node *Node, err error) {
|
||
|
|
defer mon.Task()(&ctx)(&err)
|
||
|
|
__id_val := node_id.value()
|
||
|
|
__name_val := node_name.value()
|
||
|
|
__api_secret_val := node_api_secret.value()
|
||
|
|
__logo_val := node_logo.value()
|
||
|
|
|
||
|
|
var __embed_stmt = __sqlbundle_Literal("INSERT INTO nodes ( id, name, api_secret, logo ) VALUES ( ?, ?, ?, ? ) RETURNING nodes.id, nodes.name, nodes.api_secret, nodes.logo")
|
||
|
|
|
||
|
|
var __values []interface{}
|
||
|
|
__values = append(__values, __id_val, __name_val, __api_secret_val, __logo_val)
|
||
|
|
|
||
|
|
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
|
||
|
|
obj.logStmt(__stmt, __values...)
|
||
|
|
|
||
|
|
node = &Node{}
|
||
|
|
err = obj.driver.QueryRowContext(ctx, __stmt, __values...).Scan(&node.Id, &node.Name, &node.ApiSecret, &node.Logo)
|
||
|
|
if err != nil {
|
||
|
|
return nil, obj.makeErr(err)
|
||
|
|
}
|
||
|
|
return node, nil
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func (obj *pgxImpl) Get_Node_By_Id(ctx context.Context,
|
||
|
|
node_id Node_Id_Field) (
|
||
|
|
node *Node, err error) {
|
||
|
|
defer mon.Task()(&ctx)(&err)
|
||
|
|
|
||
|
|
var __embed_stmt = __sqlbundle_Literal("SELECT nodes.id, nodes.name, nodes.api_secret, nodes.logo FROM nodes WHERE nodes.id = ?")
|
||
|
|
|
||
|
|
var __values []interface{}
|
||
|
|
__values = append(__values, node_id.value())
|
||
|
|
|
||
|
|
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
|
||
|
|
obj.logStmt(__stmt, __values...)
|
||
|
|
|
||
|
|
node = &Node{}
|
||
|
|
err = obj.driver.QueryRowContext(ctx, __stmt, __values...).Scan(&node.Id, &node.Name, &node.ApiSecret, &node.Logo)
|
||
|
|
if err != nil {
|
||
|
|
return (*Node)(nil), obj.makeErr(err)
|
||
|
|
}
|
||
|
|
return node, nil
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func (impl pgxImpl) isConstraintError(err error) (
|
||
|
|
constraint string, ok bool) {
|
||
|
|
if e, ok := err.(*pgconn.PgError); ok {
|
||
|
|
if e.Code[:2] == "23" {
|
||
|
|
return e.ConstraintName, true
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return "", false
|
||
|
|
}
|
||
|
|
|
||
|
|
func (obj *pgxImpl) deleteAll(ctx context.Context) (count int64, err error) {
|
||
|
|
defer mon.Task()(&ctx)(&err)
|
||
|
|
var __res sql.Result
|
||
|
|
var __count int64
|
||
|
|
__res, err = obj.driver.ExecContext(ctx, "DELETE FROM nodes;")
|
||
|
|
if err != nil {
|
||
|
|
return 0, obj.makeErr(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
__count, err = __res.RowsAffected()
|
||
|
|
if err != nil {
|
||
|
|
return 0, obj.makeErr(err)
|
||
|
|
}
|
||
|
|
count += __count
|
||
|
|
|
||
|
|
return count, nil
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
type Rx struct {
|
||
|
|
db *DB
|
||
|
|
tx *Tx
|
||
|
|
}
|
||
|
|
|
||
|
|
func (rx *Rx) UnsafeTx(ctx context.Context) (unsafe_tx tagsql.Tx, err error) {
|
||
|
|
tx, err := rx.getTx(ctx)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return tx.Tx, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (rx *Rx) getTx(ctx context.Context) (tx *Tx, err error) {
|
||
|
|
if rx.tx == nil {
|
||
|
|
if rx.tx, err = rx.db.Open(ctx); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return rx.tx, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (rx *Rx) Rebind(s string) string {
|
||
|
|
return rx.db.Rebind(s)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (rx *Rx) Commit() (err error) {
|
||
|
|
if rx.tx != nil {
|
||
|
|
err = rx.tx.Commit()
|
||
|
|
rx.tx = nil
|
||
|
|
}
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (rx *Rx) Rollback() (err error) {
|
||
|
|
if rx.tx != nil {
|
||
|
|
err = rx.tx.Rollback()
|
||
|
|
rx.tx = nil
|
||
|
|
}
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (rx *Rx) Create_Node(ctx context.Context,
|
||
|
|
node_id Node_Id_Field,
|
||
|
|
node_name Node_Name_Field,
|
||
|
|
node_api_secret Node_ApiSecret_Field,
|
||
|
|
node_logo Node_Logo_Field) (
|
||
|
|
node *Node, err error) {
|
||
|
|
var tx *Tx
|
||
|
|
if tx, err = rx.getTx(ctx); err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
return tx.Create_Node(ctx, node_id, node_name, node_api_secret, node_logo)
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func (rx *Rx) Get_Node_By_Id(ctx context.Context,
|
||
|
|
node_id Node_Id_Field) (
|
||
|
|
node *Node, err error) {
|
||
|
|
var tx *Tx
|
||
|
|
if tx, err = rx.getTx(ctx); err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
return tx.Get_Node_By_Id(ctx, node_id)
|
||
|
|
}
|
||
|
|
|
||
|
|
type Methods interface {
|
||
|
|
Create_Node(ctx context.Context,
|
||
|
|
node_id Node_Id_Field,
|
||
|
|
node_name Node_Name_Field,
|
||
|
|
node_api_secret Node_ApiSecret_Field,
|
||
|
|
node_logo Node_Logo_Field) (
|
||
|
|
node *Node, err error)
|
||
|
|
|
||
|
|
Get_Node_By_Id(ctx context.Context,
|
||
|
|
node_id Node_Id_Field) (
|
||
|
|
node *Node, err error)
|
||
|
|
}
|
||
|
|
|
||
|
|
type TxMethods interface {
|
||
|
|
Methods
|
||
|
|
|
||
|
|
Rebind(s string) string
|
||
|
|
Commit() error
|
||
|
|
Rollback() error
|
||
|
|
}
|
||
|
|
|
||
|
|
type txMethods interface {
|
||
|
|
TxMethods
|
||
|
|
|
||
|
|
deleteAll(ctx context.Context) (int64, error)
|
||
|
|
makeErr(err error) error
|
||
|
|
}
|
||
|
|
|
||
|
|
type DBMethods interface {
|
||
|
|
Methods
|
||
|
|
|
||
|
|
Schema() string
|
||
|
|
Rebind(sql string) string
|
||
|
|
}
|
||
|
|
|
||
|
|
type dbMethods interface {
|
||
|
|
DBMethods
|
||
|
|
|
||
|
|
wrapTx(tx tagsql.Tx) txMethods
|
||
|
|
makeErr(err error) error
|
||
|
|
}
|
||
|
|
|
||
|
|
func openpgx(source string) (*sql.DB, error) {
|
||
|
|
return sql.Open("pgx", source)
|
||
|
|
}
|