fix: try to fix pgsql #682

2024-12-31 02:06:30 +08:00
parent b4f17543cb
commit 87d5e286d5
1 changed files with 4 additions and 1 deletions
--- 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://") {