diff options
Diffstat (limited to 'vendor/github.com/uptrace/bun/query_column_add.go')
-rw-r--r-- | vendor/github.com/uptrace/bun/query_column_add.go | 116 |
1 files changed, 0 insertions, 116 deletions
diff --git a/vendor/github.com/uptrace/bun/query_column_add.go b/vendor/github.com/uptrace/bun/query_column_add.go deleted file mode 100644 index 0105fdb38..000000000 --- a/vendor/github.com/uptrace/bun/query_column_add.go +++ /dev/null @@ -1,116 +0,0 @@ -package bun - -import ( - "context" - "database/sql" - "fmt" - - "github.com/uptrace/bun/internal" - "github.com/uptrace/bun/schema" -) - -type AddColumnQuery struct { - baseQuery - - ifNotExists bool -} - -var _ Query = (*AddColumnQuery)(nil) - -func NewAddColumnQuery(db *DB) *AddColumnQuery { - q := &AddColumnQuery{ - baseQuery: baseQuery{ - db: db, - conn: db.DB, - }, - } - return q -} - -func (q *AddColumnQuery) Conn(db IConn) *AddColumnQuery { - q.setConn(db) - return q -} - -func (q *AddColumnQuery) Model(model interface{}) *AddColumnQuery { - q.setTableModel(model) - return q -} - -//------------------------------------------------------------------------------ - -func (q *AddColumnQuery) Table(tables ...string) *AddColumnQuery { - for _, table := range tables { - q.addTable(schema.UnsafeIdent(table)) - } - return q -} - -func (q *AddColumnQuery) TableExpr(query string, args ...interface{}) *AddColumnQuery { - q.addTable(schema.SafeQuery(query, args)) - return q -} - -func (q *AddColumnQuery) ModelTableExpr(query string, args ...interface{}) *AddColumnQuery { - q.modelTableName = schema.SafeQuery(query, args) - return q -} - -//------------------------------------------------------------------------------ - -func (q *AddColumnQuery) ColumnExpr(query string, args ...interface{}) *AddColumnQuery { - q.addColumn(schema.SafeQuery(query, args)) - return q -} - -func (q *AddColumnQuery) IfNotExists() *AddColumnQuery { - q.ifNotExists = true - return q -} - -//------------------------------------------------------------------------------ - -func (q *AddColumnQuery) Operation() string { - return "ADD COLUMN" -} - -func (q *AddColumnQuery) AppendQuery(fmter schema.Formatter, b []byte) (_ []byte, err error) { - if q.err != nil { - return nil, q.err - } - if len(q.columns) != 1 { - return nil, fmt.Errorf("bun: AddColumnQuery requires exactly one column") - } - - b = append(b, "ALTER TABLE "...) - - b, err = q.appendFirstTable(fmter, b) - if err != nil { - return nil, err - } - - b = append(b, " ADD "...) - - if q.ifNotExists { - b = append(b, "IF NOT EXISTS "...) - } - - b, err = q.columns[0].AppendQuery(fmter, b) - if err != nil { - return nil, err - } - - return b, nil -} - -//------------------------------------------------------------------------------ - -func (q *AddColumnQuery) Exec(ctx context.Context, dest ...interface{}) (sql.Result, error) { - queryBytes, err := q.AppendQuery(q.db.fmter, q.db.makeQueryBytes()) - if err != nil { - return nil, err - } - - query := internal.String(queryBytes) - return q.exec(ctx, q, query) -} |