fix: try to fix pgsql #682

Calcium-Ion · Dec 30, 2024 · 87d5e28 · 87d5e28
1 parent b4f1754
commit 87d5e28
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/model/main.go b/model/main.go
@@ -16,7 +16,7 @@ import (
 var groupCol string
 var keyCol string
 
-func init() {
+func initCol() {
 	if common.UsingPostgreSQL {
 		groupCol = `"group"`
 		keyCol = `"key"`
@@ -55,6 +55,9 @@ func createRootAccountIfNeed() error {
 }
 
 func chooseDB(envName string) (*gorm.DB, error) {
+	defer func() {
+		initCol()
+	}()
 	dsn := os.Getenv(envName)
 	if dsn != "" {
 		if strings.HasPrefix(dsn, "postgres://") {