Opt: mysql ssl mode

synctv-org · Nov 17, 2023 · 1bb24e7 · 1bb24e7
1 parent cc23248
commit 1bb24e7
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/internal/bootstrap/db.go b/internal/bootstrap/db.go
@@ -28,20 +28,22 @@ func InitDatabase(ctx context.Context) error {
 		if conf.Conf.Database.CustomDSN != "" {
 			dsn = conf.Conf.Database.CustomDSN
 		} else if conf.Conf.Database.Port == 0 {
-			dsn = fmt.Sprintf("%s:%s@unix(%s)/%s?charset=utf8mb4&parseTime=True&loc=Local&interpolateParams=true",
+			dsn = fmt.Sprintf("%s:%s@unix(%s)/%s?charset=utf8mb4&parseTime=True&loc=Local&interpolateParams=true&tls=%s",
 				conf.Conf.Database.User,
 				conf.Conf.Database.Password,
 				conf.Conf.Database.Host,
 				conf.Conf.Database.DBName,
+				conf.Conf.Database.SslMode,
 			)
 			log.Infof("mysql database unix socket: %s", conf.Conf.Database.Host)
 		} else {
-			dsn = fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local&interpolateParams=true",
+			dsn = fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local&interpolateParams=true&tls=%s",
 				conf.Conf.Database.User,
 				conf.Conf.Database.Password,
 				conf.Conf.Database.Host,
 				conf.Conf.Database.Port,
 				conf.Conf.Database.DBName,
+				conf.Conf.Database.SslMode,
 			)
 			log.Infof("mysql database tcp: %s:%d", conf.Conf.Database.Host, conf.Conf.Database.Port)
 		}

diff --git a/internal/conf/db.go b/internal/conf/db.go
@@ -15,7 +15,7 @@ type DatabaseConfig struct {
 	User     string       `yaml:"user" env:"DATABASE_USER"`
 	Password string       `yaml:"password" env:"DATABASE_PASSWORD"`
 	DBName   string       `yaml:"db_name" lc:"default: synctv" hc:"when type is sqlite3, it will use sqlite db file or memory" env:"DATABASE_DB_NAME"`
-	SslMode  string       `yaml:"ssl_mode" env:"DATABASE_SSL_MODE"`
+	SslMode  string       `yaml:"ssl_mode" env:"DATABASE_SSL_MODE" hc:"mysql: true, false, skip-verify, preferred, <name> postgres: disable, require, verify-ca, verify-full"`
 
 	CustomDSN string `yaml:"custom_dsn" hc:"when not empty, it will ignore other config" env:"DATABASE_CUSTOM_DSN"`
 
@@ -27,10 +27,9 @@ type DatabaseConfig struct {
 
 func DefaultDatabaseConfig() DatabaseConfig {
 	return DatabaseConfig{
-		Type:    DatabaseTypeSqlite3,
-		Host:    "",
-		DBName:  "synctv",
-		SslMode: "disable",
+		Type:   DatabaseTypeSqlite3,
+		Host:   "",
+		DBName: "synctv",
 
 		MaxIdleConns:    4,
 		MaxOpenConns:    64,