diff options
Diffstat (limited to 'vendor/github.com/uptrace/bun/query_table_truncate.go')
-rw-r--r-- | vendor/github.com/uptrace/bun/query_table_truncate.go | 132 |
1 files changed, 0 insertions, 132 deletions
diff --git a/vendor/github.com/uptrace/bun/query_table_truncate.go b/vendor/github.com/uptrace/bun/query_table_truncate.go deleted file mode 100644 index 870c776c6..000000000 --- a/vendor/github.com/uptrace/bun/query_table_truncate.go +++ /dev/null @@ -1,132 +0,0 @@ -package bun - -import ( - "context" - "database/sql" - - "github.com/uptrace/bun/dialect/feature" - "github.com/uptrace/bun/internal" - "github.com/uptrace/bun/schema" -) - -type TruncateTableQuery struct { - baseQuery - cascadeQuery - - continueIdentity bool -} - -var _ Query = (*TruncateTableQuery)(nil) - -func NewTruncateTableQuery(db *DB) *TruncateTableQuery { - q := &TruncateTableQuery{ - baseQuery: baseQuery{ - db: db, - conn: db.DB, - }, - } - return q -} - -func (q *TruncateTableQuery) Conn(db IConn) *TruncateTableQuery { - q.setConn(db) - return q -} - -func (q *TruncateTableQuery) Model(model interface{}) *TruncateTableQuery { - q.setTableModel(model) - return q -} - -//------------------------------------------------------------------------------ - -func (q *TruncateTableQuery) Table(tables ...string) *TruncateTableQuery { - for _, table := range tables { - q.addTable(schema.UnsafeIdent(table)) - } - return q -} - -func (q *TruncateTableQuery) TableExpr(query string, args ...interface{}) *TruncateTableQuery { - q.addTable(schema.SafeQuery(query, args)) - return q -} - -//------------------------------------------------------------------------------ - -func (q *TruncateTableQuery) ContinueIdentity() *TruncateTableQuery { - q.continueIdentity = true - return q -} - -func (q *TruncateTableQuery) Cascade() *TruncateTableQuery { - q.cascade = true - return q -} - -func (q *TruncateTableQuery) Restrict() *TruncateTableQuery { - q.restrict = true - return q -} - -//------------------------------------------------------------------------------ - -func (q *TruncateTableQuery) Operation() string { - return "TRUNCATE TABLE" -} - -func (q *TruncateTableQuery) AppendQuery( - fmter schema.Formatter, b []byte, -) (_ []byte, err error) { - if q.err != nil { - return nil, q.err - } - - if !fmter.HasFeature(feature.TableTruncate) { - b = append(b, "DELETE FROM "...) - - b, err = q.appendTables(fmter, b) - if err != nil { - return nil, err - } - - return b, nil - } - - b = append(b, "TRUNCATE TABLE "...) - - b, err = q.appendTables(fmter, b) - if err != nil { - return nil, err - } - - if q.db.features.Has(feature.TableIdentity) { - if q.continueIdentity { - b = append(b, " CONTINUE IDENTITY"...) - } else { - b = append(b, " RESTART IDENTITY"...) - } - } - - b = q.appendCascade(fmter, b) - - return b, nil -} - -//------------------------------------------------------------------------------ - -func (q *TruncateTableQuery) 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 -} |