package bun import ( "context" "database/sql" "github.com/uptrace/bun/internal" "github.com/uptrace/bun/schema" ) type DropIndexQuery struct { baseQuery cascadeQuery concurrently bool ifExists bool index schema.QueryWithArgs } func NewDropIndexQuery(db *DB) *DropIndexQuery { q := &DropIndexQuery{ baseQuery: baseQuery{ db: db, conn: db.DB, }, } return q } func (q *DropIndexQuery) Conn(db IConn) *DropIndexQuery { q.setConn(db) return q } func (q *DropIndexQuery) Model(model interface{}) *DropIndexQuery { q.setTableModel(model) return q } //------------------------------------------------------------------------------ func (q *DropIndexQuery) Concurrently() *DropIndexQuery { q.concurrently = true return q } func (q *DropIndexQuery) IfExists() *DropIndexQuery { q.ifExists = true return q } func (q *DropIndexQuery) Restrict() *DropIndexQuery { q.restrict = true return q } func (q *DropIndexQuery) Index(query string, args ...interface{}) *DropIndexQuery { q.index = schema.SafeQuery(query, args) return q } //------------------------------------------------------------------------------ func (q *DropIndexQuery) Operation() string { return "DROP INDEX" } func (q *DropIndexQuery) AppendQuery(fmter schema.Formatter, b []byte) (_ []byte, err error) { if q.err != nil { return nil, q.err } b = append(b, "DROP INDEX "...) if q.concurrently { b = append(b, "CONCURRENTLY "...) } if q.ifExists { b = append(b, "IF EXISTS "...) } b, err = q.index.AppendQuery(fmter, b) if err != nil { return nil, err } b = q.appendCascade(fmter, b) return b, nil } //------------------------------------------------------------------------------ func (q *DropIndexQuery) 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 }