diff options
Diffstat (limited to 'vendor/github.com/uptrace/bun/query_delete.go')
-rw-r--r-- | vendor/github.com/uptrace/bun/query_delete.go | 26 |
1 files changed, 21 insertions, 5 deletions
diff --git a/vendor/github.com/uptrace/bun/query_delete.go b/vendor/github.com/uptrace/bun/query_delete.go index c5f3d7103..5899c9ba6 100644 --- a/vendor/github.com/uptrace/bun/query_delete.go +++ b/vendor/github.com/uptrace/bun/query_delete.go @@ -287,12 +287,32 @@ func (q *DeleteQuery) afterDeleteHook(ctx context.Context) error { return nil } +func (q *DeleteQuery) String() string { + buf, err := q.AppendQuery(q.db.Formatter(), nil) + if err != nil { + panic(err) + } + + return string(buf) +} + //------------------------------------------------------------------------------ + +func (q *DeleteQuery) QueryBuilder() QueryBuilder { + return &deleteQueryBuilder{q} +} + +func (q *DeleteQuery) ApplyQueryBuilder(fn func(QueryBuilder) QueryBuilder) *DeleteQuery { + return fn(q.QueryBuilder()).Unwrap().(*DeleteQuery) +} + type deleteQueryBuilder struct { *DeleteQuery } -func (q *deleteQueryBuilder) WhereGroup(sep string, fn func(QueryBuilder) QueryBuilder) QueryBuilder { +func (q *deleteQueryBuilder) WhereGroup( + sep string, fn func(QueryBuilder) QueryBuilder, +) QueryBuilder { q.DeleteQuery = q.DeleteQuery.WhereGroup(sep, func(qs *DeleteQuery) *DeleteQuery { return fn(q).(*deleteQueryBuilder).DeleteQuery }) @@ -327,7 +347,3 @@ func (q *deleteQueryBuilder) WherePK(cols ...string) QueryBuilder { func (q *deleteQueryBuilder) Unwrap() interface{} { return q.DeleteQuery } - -func (q *DeleteQuery) Query() QueryBuilder { - return &deleteQueryBuilder{q} -} |