fix issues

2023-07-06 19:05:29 +08:00
parent f75985f797
commit c4ec53b4d0
1 changed files with 4 additions and 0 deletions
--- a/utils/db/db.go
+++ b/utils/db/db.go
@@ -7,5 +7,9 @@ import (
 )

 func TruncateTable(db *gorm.DB, table string) {
+	if db.Dialector.Name() == "postgres" {
+		db.Exec(fmt.Sprintf("TRUNCATE TABLE %s RESTART IDENTITY", table))
+		return
+	}
 	db.Exec(fmt.Sprintf("TRUNCATE TABLE %s", table))
 }