瀏覽代碼

fix: try to fix pgsql #682

CalciumIon 1 年之前
父節點
當前提交
87d5e286d5
共有 1 個文件被更改,包括 4 次插入1 次删除
  1. 4 1
      model/main.go

+ 4 - 1
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://") {