|
|
|
@ -2,6 +2,7 @@ package model
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"fmt"
|
|
|
|
|
"strings"
|
|
|
|
|
"time"
|
|
|
|
|
|
|
|
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/conf"
|
|
|
|
@ -36,12 +37,19 @@ func Init() {
|
|
|
|
|
// 未指定数据库或者明确指定为 sqlite 时,使用 SQLite3 数据库
|
|
|
|
|
db, err = gorm.Open("sqlite3", util.RelativePath(conf.DatabaseConfig.DBFile))
|
|
|
|
|
case "postgres":
|
|
|
|
|
db, err = gorm.Open(conf.DatabaseConfig.Type, fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%d sslmode=disable",
|
|
|
|
|
var sslmode = "disable"
|
|
|
|
|
if strings.EqualFold(conf.DatabaseConfig.TLS, "true") {
|
|
|
|
|
sslmode = "require"
|
|
|
|
|
} else {
|
|
|
|
|
sslmode = "disable"
|
|
|
|
|
}
|
|
|
|
|
db, err = gorm.Open(conf.DatabaseConfig.Type, fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%d sslmode=%s",
|
|
|
|
|
conf.DatabaseConfig.Host,
|
|
|
|
|
conf.DatabaseConfig.User,
|
|
|
|
|
conf.DatabaseConfig.Password,
|
|
|
|
|
conf.DatabaseConfig.Name,
|
|
|
|
|
conf.DatabaseConfig.Port))
|
|
|
|
|
conf.DatabaseConfig.Port,
|
|
|
|
|
sslmode))
|
|
|
|
|
case "mysql", "mssql":
|
|
|
|
|
db, err = gorm.Open(conf.DatabaseConfig.Type, fmt.Sprintf("%s:%s@(%s:%d)/%s?charset=%s&parseTime=True&loc=Local&tls=%s",
|
|
|
|
|
conf.DatabaseConfig.User,
|
|
|
|
|