admin777/new-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: try to fix pgsql #682

Browse Source
This commit is contained in:
CalciumIon 2024-12-31 02:06:30 +08:00
parent 2de4de0006
commit e0b7300239
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
model/main.go
View File

@ -16,7 +16,7 @@ import (
var groupCol string var groupCol string
var keyCol string var keyCol string
func init() { func initCol() {
if common.UsingPostgreSQL { if common.UsingPostgreSQL {
groupCol = `"group"` groupCol = `"group"`
keyCol = `"key"` keyCol = `"key"`
@ -55,6 +55,9 @@ func createRootAccountIfNeed() error {
} }
func chooseDB(envName string) (*gorm.DB, error) { func chooseDB(envName string) (*gorm.DB, error) {
defer func() {
initCol()
}()
dsn := os.Getenv(envName) dsn := os.Getenv(envName)
if dsn != "" { if dsn != "" {
if strings.HasPrefix(dsn, "postgres://") { if strings.HasPrefix(dsn, "postgres://") {