|
|
@@ -1,16 +1,17 @@
|
|
|
package model
|
|
|
|
|
|
import (
|
|
|
- "github.com/glebarez/sqlite"
|
|
|
- "gorm.io/driver/mysql"
|
|
|
- "gorm.io/driver/postgres"
|
|
|
- "gorm.io/gorm"
|
|
|
"log"
|
|
|
"one-api/common"
|
|
|
"os"
|
|
|
"strings"
|
|
|
"sync"
|
|
|
"time"
|
|
|
+
|
|
|
+ "github.com/glebarez/sqlite"
|
|
|
+ "gorm.io/driver/mysql"
|
|
|
+ "gorm.io/driver/postgres"
|
|
|
+ "gorm.io/gorm"
|
|
|
)
|
|
|
|
|
|
var groupCol string
|
|
|
@@ -60,7 +61,7 @@ func chooseDB(envName string) (*gorm.DB, error) {
|
|
|
}()
|
|
|
dsn := os.Getenv(envName)
|
|
|
if dsn != "" {
|
|
|
- if strings.HasPrefix(dsn, "postgres://") {
|
|
|
+ if strings.HasPrefix(dsn, "postgres://") || strings.HasPrefix(dsn, "postgresql://") {
|
|
|
// Use PostgreSQL
|
|
|
common.SysLog("using PostgreSQL as database")
|
|
|
common.UsingPostgreSQL = true
|