@@ -1166,16 +1166,21 @@ func getDataDBSession() (map[string]*gorm.DB, []model.DatabasesMap) {
1166
1166
keepAlive = viper .GetBool (keyData + ".keepalive" )
1167
1167
}
1168
1168
1169
+ customcli := ""
1170
+ if viper .IsSet (keyData + ".custom_cli" ) {
1171
+ customcli = viper .GetString (keyData + ".custom_cli" )
1172
+ }
1173
+
1169
1174
sslMode := "disable"
1170
1175
if viper .IsSet (keyData + ".sslmode" ) {
1171
1176
sslMode = viper .GetString (keyData + ".sslmode" )
1172
1177
} else if viper .IsSet (keyData + ".usessl" ) && viper .GetBool (keyData + ".usessl" ) {
1173
1178
sslMode = "require"
1174
1179
}
1175
1180
1176
- logger .Info (fmt .Sprintf ("Connecting to [%s, %s, %s, %s, %d, ssl: %s]\n " , host , user , name , node , port , sslMode ))
1181
+ logger .Info (fmt .Sprintf ("Connecting to [%s, %s, %s, %s, %d, ssl: %s, custom_cli: %s ]\n " , host , user , name , node , port , sslMode , customcli ))
1177
1182
1178
- connectString := fmt .Sprintf ("host=%s user=%s dbname=%s sslmode=%s password=%s" , host , user , name , sslMode , password )
1183
+ connectString := fmt .Sprintf ("host=%s user=%s dbname=%s sslmode=%s password=%s %s " , host , user , name , sslMode , password , customcli )
1179
1184
1180
1185
if port != 0 {
1181
1186
connectString += fmt .Sprintf (" port=%d" , port )
@@ -1273,7 +1278,12 @@ func getDataDBSession() (map[string]*gorm.DB, []model.DatabasesMap) {
1273
1278
keepAlive = viper .GetBool ("database_data.keepalive" )
1274
1279
}
1275
1280
1276
- logger .Info (fmt .Sprintf ("Connecting to the old way: [%s, %s, %s, %d]\n " , host , user , name , port ))
1281
+ customcli := ""
1282
+ if viper .IsSet ("database_data.custom_cli" ) {
1283
+ customcli = viper .GetString ("database_data.custom_cli" )
1284
+ }
1285
+
1286
+ logger .Info (fmt .Sprintf ("Connecting to the old way: [%s, %s, %s, %d, %s]\n " , host , user , name , port , customcli ))
1277
1287
1278
1288
sslMode := "disable"
1279
1289
if viper .IsSet ("database_data.sslmode" ) {
@@ -1282,7 +1292,7 @@ func getDataDBSession() (map[string]*gorm.DB, []model.DatabasesMap) {
1282
1292
sslMode = "require"
1283
1293
}
1284
1294
1285
- connectString := fmt .Sprintf ("host=%s user=%s dbname=%s sslmode=%s password=%s" , host , user , name , sslMode , password )
1295
+ connectString := fmt .Sprintf ("host=%s user=%s dbname=%s sslmode=%s password=%s %s " , host , user , name , sslMode , password , customcli )
1286
1296
1287
1297
if port != 0 {
1288
1298
connectString += fmt .Sprintf (" port=%d" , port )
@@ -1357,7 +1367,12 @@ func getConfigDBSession() *gorm.DB {
1357
1367
sslMode = "require"
1358
1368
}
1359
1369
1360
- connectString := fmt .Sprintf ("host=%s user=%s dbname=%s sslmode=%s password=%s" , host , user , name , sslMode , password )
1370
+ customcli := ""
1371
+ if viper .IsSet ("database_config.custom_cli" ) {
1372
+ customcli = viper .GetString ("database_config.custom_cli" )
1373
+ }
1374
+
1375
+ connectString := fmt .Sprintf ("host=%s user=%s dbname=%s sslmode=%s password=%s %s" , host , user , name , sslMode , password , customcli )
1361
1376
1362
1377
if port != 0 {
1363
1378
connectString += fmt .Sprintf (" port=%d" , port )
0 commit comments