fix: gen models

2024-12-25 20:16:41 +08:00
parent 905fb3f9f7
commit 7dde1820c3
4 changed files with 63 additions and 19 deletions
--- a/templates/project/database/database.go.tpl
+++ b/templates/project/database/database.go.tpl
@@ -21,7 +21,7 @@ func FromContext(ctx context.Context, db *sql.DB) qrm.DB {
 	return db
 }

-func TruncateAllTables(ctx context.Context, db *sql.DB, tableName ...string) error {
+func Truncate(ctx context.Context, db *sql.DB, tableName ...string) error {
 	for _, name := range tableName {
 		sql := fmt.Sprintf("TRUNCATE TABLE %s RESTART IDENTITY", name)
 		if _, err := db.ExecContext(ctx, sql); err != nil {