diff options
| author | ᴜɴᴋɴᴡᴏɴ <u@gogs.io> | 2020-09-06 10:11:08 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-09-06 10:11:08 +0800 |
| commit | 519e59b5778571ace3f681b81a21b92a38ede890 (patch) | |
| tree | 373a2021f3adcf8b2c44e6f28282733e00909d88 /internal/db/models.go | |
| parent | 771d3673f5dc1ba96ed9bf2ebe22c7f3066ec31d (diff) | |
db: migrate to GORM v2 (#6309)
Diffstat (limited to 'internal/db/models.go')
| -rw-r--r-- | internal/db/models.go | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/internal/db/models.go b/internal/db/models.go index d2868bb6..52ef4495 100644 --- a/internal/db/models.go +++ b/internal/db/models.go @@ -13,7 +13,7 @@ import ( "strings" "time" - "github.com/jinzhu/gorm" + "gorm.io/gorm" log "unknwon.dev/clog/v2" "xorm.io/core" "xorm.io/xorm" @@ -68,6 +68,7 @@ func getEngine() (*xorm.Engine, error) { Param = "&" } + driver := conf.Database.Type connStr := "" switch conf.Database.Type { case "mysql": @@ -92,6 +93,7 @@ func getEngine() (*xorm.Engine, error) { connStr = fmt.Sprintf("postgres://%s:%s@%s:%s/%s%ssslmode=%s", url.QueryEscape(conf.Database.User), url.QueryEscape(conf.Database.Password), host, port, conf.Database.Name, Param, conf.Database.SSLMode) } + driver = "pgx" case "mssql": conf.UseMSSQL = true @@ -108,7 +110,7 @@ func getEngine() (*xorm.Engine, error) { default: return nil, fmt.Errorf("unknown database type: %s", conf.Database.Type) } - return xorm.NewEngine(conf.Database.Type, connStr) + return xorm.NewEngine(driver, connStr) } func NewTestEngine() error { |