feat: uniq tenant_users

This commit is contained in:
Rogee
2024-12-02 11:29:15 +08:00
parent d1a2a80859
commit 2b4cfb1a1e
7 changed files with 198 additions and 37 deletions

View File

@@ -3,6 +3,7 @@ package db
import (
"context"
"database/sql"
"fmt"
"backend/common/consts"
@@ -15,3 +16,13 @@ func FromContext(ctx context.Context, db *sql.DB) qrm.DB {
}
return db
}
func TruncateAllTables(ctx context.Context, db *sql.DB, tableName ...string) error {
for _, name := range tableName {
sql := fmt.Sprintf("TRUNCATE TABLE %s CASCADE", name)
if _, err := db.ExecContext(ctx, sql); err != nil {
return err
}
}
return nil
}