fix:support mysql/postgres ssl/tls connect

pull/1255/head
lsgdut 3 years ago
parent 5f1da5644d
commit 4c160cadc6

@ -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,

Loading…
Cancel
Save