diff options
Diffstat (limited to 'vendor/github.com/uptrace/bun/query_index_create.go')
-rw-r--r-- | vendor/github.com/uptrace/bun/query_index_create.go | 264 |
1 files changed, 0 insertions, 264 deletions
diff --git a/vendor/github.com/uptrace/bun/query_index_create.go b/vendor/github.com/uptrace/bun/query_index_create.go deleted file mode 100644 index 4ac4ffd10..000000000 --- a/vendor/github.com/uptrace/bun/query_index_create.go +++ /dev/null @@ -1,264 +0,0 @@ -package bun - -import ( - "context" - "database/sql" - - "github.com/uptrace/bun/internal" - "github.com/uptrace/bun/schema" -) - -type CreateIndexQuery struct { - whereBaseQuery - - unique bool - fulltext bool - spatial bool - concurrently bool - ifNotExists bool - - index schema.QueryWithArgs - using schema.QueryWithArgs - include []schema.QueryWithArgs - comment string -} - -var _ Query = (*CreateIndexQuery)(nil) - -func NewCreateIndexQuery(db *DB) *CreateIndexQuery { - q := &CreateIndexQuery{ - whereBaseQuery: whereBaseQuery{ - baseQuery: baseQuery{ - db: db, - }, - }, - } - return q -} - -func (q *CreateIndexQuery) Conn(db IConn) *CreateIndexQuery { - q.setConn(db) - return q -} - -func (q *CreateIndexQuery) Model(model interface{}) *CreateIndexQuery { - q.setModel(model) - return q -} - -func (q *CreateIndexQuery) Err(err error) *CreateIndexQuery { - q.setErr(err) - return q -} - -func (q *CreateIndexQuery) Unique() *CreateIndexQuery { - q.unique = true - return q -} - -func (q *CreateIndexQuery) Concurrently() *CreateIndexQuery { - q.concurrently = true - return q -} - -func (q *CreateIndexQuery) IfNotExists() *CreateIndexQuery { - q.ifNotExists = true - return q -} - -//------------------------------------------------------------------------------ - -func (q *CreateIndexQuery) Index(query string) *CreateIndexQuery { - q.index = schema.UnsafeIdent(query) - return q -} - -func (q *CreateIndexQuery) IndexExpr(query string, args ...interface{}) *CreateIndexQuery { - q.index = schema.SafeQuery(query, args) - return q -} - -//------------------------------------------------------------------------------ - -func (q *CreateIndexQuery) Table(tables ...string) *CreateIndexQuery { - for _, table := range tables { - q.addTable(schema.UnsafeIdent(table)) - } - return q -} - -func (q *CreateIndexQuery) TableExpr(query string, args ...interface{}) *CreateIndexQuery { - q.addTable(schema.SafeQuery(query, args)) - return q -} - -func (q *CreateIndexQuery) ModelTableExpr(query string, args ...interface{}) *CreateIndexQuery { - q.modelTableName = schema.SafeQuery(query, args) - return q -} - -func (q *CreateIndexQuery) Using(query string, args ...interface{}) *CreateIndexQuery { - q.using = schema.SafeQuery(query, args) - return q -} - -//------------------------------------------------------------------------------ - -func (q *CreateIndexQuery) Column(columns ...string) *CreateIndexQuery { - for _, column := range columns { - q.addColumn(schema.UnsafeIdent(column)) - } - return q -} - -func (q *CreateIndexQuery) ColumnExpr(query string, args ...interface{}) *CreateIndexQuery { - q.addColumn(schema.SafeQuery(query, args)) - return q -} - -func (q *CreateIndexQuery) ExcludeColumn(columns ...string) *CreateIndexQuery { - q.excludeColumn(columns) - return q -} - -//------------------------------------------------------------------------------ - -func (q *CreateIndexQuery) Include(columns ...string) *CreateIndexQuery { - for _, column := range columns { - q.include = append(q.include, schema.UnsafeIdent(column)) - } - return q -} - -func (q *CreateIndexQuery) IncludeExpr(query string, args ...interface{}) *CreateIndexQuery { - q.include = append(q.include, schema.SafeQuery(query, args)) - return q -} - -//------------------------------------------------------------------------------ - -func (q *CreateIndexQuery) Where(query string, args ...interface{}) *CreateIndexQuery { - q.addWhere(schema.SafeQueryWithSep(query, args, " AND ")) - return q -} - -func (q *CreateIndexQuery) WhereOr(query string, args ...interface{}) *CreateIndexQuery { - q.addWhere(schema.SafeQueryWithSep(query, args, " OR ")) - return q -} - -//------------------------------------------------------------------------------ - -// Comment adds a comment to the query, wrapped by /* ... */. -func (q *CreateIndexQuery) Comment(comment string) *CreateIndexQuery { - q.comment = comment - return q -} - -//------------------------------------------------------------------------------ - -func (q *CreateIndexQuery) Operation() string { - return "CREATE INDEX" -} - -func (q *CreateIndexQuery) AppendQuery(fmter schema.Formatter, b []byte) (_ []byte, err error) { - if q.err != nil { - return nil, q.err - } - - b = appendComment(b, q.comment) - - b = append(b, "CREATE "...) - - if q.unique { - b = append(b, "UNIQUE "...) - } - if q.fulltext { - b = append(b, "FULLTEXT "...) - } - if q.spatial { - b = append(b, "SPATIAL "...) - } - - b = append(b, "INDEX "...) - - if q.concurrently { - b = append(b, "CONCURRENTLY "...) - } - if q.ifNotExists { - b = append(b, "IF NOT EXISTS "...) - } - - b, err = q.index.AppendQuery(fmter, b) - if err != nil { - return nil, err - } - - b = append(b, " ON "...) - b, err = q.appendFirstTable(fmter, b) - if err != nil { - return nil, err - } - - if !q.using.IsZero() { - b = append(b, " USING "...) - b, err = q.using.AppendQuery(fmter, b) - if err != nil { - return nil, err - } - } - - b = append(b, " ("...) - for i, col := range q.columns { - if i > 0 { - b = append(b, ", "...) - } - b, err = col.AppendQuery(fmter, b) - if err != nil { - return nil, err - } - } - b = append(b, ')') - - if len(q.include) > 0 { - b = append(b, " INCLUDE ("...) - for i, col := range q.include { - if i > 0 { - b = append(b, ", "...) - } - b, err = col.AppendQuery(fmter, b) - if err != nil { - return nil, err - } - } - b = append(b, ')') - } - - if len(q.where) > 0 { - b = append(b, " WHERE "...) - b, err = appendWhere(fmter, b, q.where) - if err != nil { - return nil, err - } - } - - return b, nil -} - -//------------------------------------------------------------------------------ - -func (q *CreateIndexQuery) 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) - - res, err := q.exec(ctx, q, query) - if err != nil { - return nil, err - } - - return res, nil -} |