From 4c160cadc6ac92e507f3b9ea2bd7467f0539519e Mon Sep 17 00:00:00 2001
From: lsgdut <lhwywh@live.com>
Date: Tue, 3 May 2022 20:31:00 +0800
Subject: [PATCH] fix:support mysql/postgres ssl/tls connect

---
 models/init.go | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/models/init.go b/models/init.go
index 83d5d16c..a1066d74 100644
--- a/models/init.go
+++ b/models/init.go
@@ -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,