From c4ec53b4d0c3577e63b17ea450dabce0cadf2644 Mon Sep 17 00:00:00 2001
From: yanghao05 <yanghao05@qianxin.com>
Date: Thu, 6 Jul 2023 19:05:29 +0800
Subject: [PATCH] fix issues

---
 utils/db/db.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/utils/db/db.go b/utils/db/db.go
index 6a99220..dc7d355 100644
--- 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))
 }