rogee/atom
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix issues

Browse Source
This commit is contained in:
yanghao05
2023-07-06 19:05:29 +08:00
parent f75985f797
commit c4ec53b4d0
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
utils/db/db.go
View File

@@ -7,5 +7,9 @@ import (
) )
func TruncateTable(db *gorm.DB, table string) { 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)) db.Exec(fmt.Sprintf("TRUNCATE TABLE %s", table))
} }