summaryrefslogtreecommitdiff
path: root/vendor/github.com/uptrace/bun/query_table_truncate.go
diff options
context:
space:
mode:
authorLibravatar kim <89579420+NyaaaWhatsUpDoc@users.noreply.github.com>2025-01-14 14:23:28 +0000
committerLibravatar GitHub <noreply@github.com>2025-01-14 14:23:28 +0000
commitb8ef9fc4bcccc6c024edaa8e9c91a6bf87f83dd9 (patch)
tree68eaf966c80237e18993e887c8583355f0943ca7 /vendor/github.com/uptrace/bun/query_table_truncate.go
parent[chore] better dns validation (#3644) (diff)
downloadgotosocial-b8ef9fc4bcccc6c024edaa8e9c91a6bf87f83dd9.tar.xz
bump uptrace/bun dependencies from 1.2.6 to 1.2.8 (#3645)
Diffstat (limited to 'vendor/github.com/uptrace/bun/query_table_truncate.go')
-rw-r--r--vendor/github.com/uptrace/bun/query_table_truncate.go13
1 files changed, 12 insertions, 1 deletions
diff --git a/vendor/github.com/uptrace/bun/query_table_truncate.go b/vendor/github.com/uptrace/bun/query_table_truncate.go
index 9ac5599d9..7ee5d2a8d 100644
--- a/vendor/github.com/uptrace/bun/query_table_truncate.go
+++ b/vendor/github.com/uptrace/bun/query_table_truncate.go
@@ -14,6 +14,7 @@ type TruncateTableQuery struct {
cascadeQuery
continueIdentity bool
+ comment string
}
var _ Query = (*TruncateTableQuery)(nil)
@@ -81,6 +82,14 @@ func (q *TruncateTableQuery) Restrict() *TruncateTableQuery {
//------------------------------------------------------------------------------
+// Comment adds a comment to the query, wrapped by /* ... */.
+func (q *TruncateTableQuery) Comment(comment string) *TruncateTableQuery {
+ q.comment = comment
+ return q
+}
+
+//------------------------------------------------------------------------------
+
func (q *TruncateTableQuery) Operation() string {
return "TRUNCATE TABLE"
}
@@ -92,6 +101,8 @@ func (q *TruncateTableQuery) AppendQuery(
return nil, q.err
}
+ b = appendComment(b, q.comment)
+
if !fmter.HasFeature(feature.TableTruncate) {
b = append(b, "DELETE FROM "...)
@@ -110,7 +121,7 @@ func (q *TruncateTableQuery) AppendQuery(
return nil, err
}
- if q.db.features.Has(feature.TableIdentity) {
+ if q.db.HasFeature(feature.TableIdentity) {
if q.continueIdentity {
b = append(b, " CONTINUE IDENTITY"...)
} else {