File tree 2 files changed +15
-2
lines changed
2 files changed +15
-2
lines changed Original file line number Diff line number Diff line change 6
6
"crypto/tls"
7
7
"database/sql"
8
8
"fmt"
9
+
9
10
"github.com/go-sql-driver/mysql"
10
11
_ "github.com/go-sql-driver/mysql"
11
12
_ "github.com/mattn/go-sqlite3"
@@ -40,7 +41,7 @@ func getConn() *sql.DB {
40
41
viper .GetString ("mysql.password" ),
41
42
viper .GetString ("mysql.host" ),
42
43
viper .GetInt ("mysql.port" ),
43
- viper . GetString ("mysql.db" ),
44
+ utils . GetStringConfs ("mysql.db" , "mysql.database " ),
44
45
)
45
46
if viper .GetBool ("mysql.tls" ) {
46
47
mysql .RegisterTLSConfig ("tls" , & tls.Config {
Original file line number Diff line number Diff line change @@ -2,9 +2,11 @@ package utils
2
2
3
3
import (
4
4
"fmt"
5
- "github.com/goccy/go-json"
6
5
"math/rand"
7
6
"time"
7
+
8
+ "github.com/goccy/go-json"
9
+ "github.com/spf13/viper"
8
10
)
9
11
10
12
func Intn (n int ) int {
@@ -332,3 +334,13 @@ func Range(start int, end int) []int {
332
334
}
333
335
return res
334
336
}
337
+
338
+ func GetStringConfs (key ... string ) string {
339
+ for _ , k := range key {
340
+ if v := viper .GetString (k ); len (v ) > 0 {
341
+ return v
342
+ }
343
+ }
344
+
345
+ return ""
346
+ }
You can’t perform that action at this time.
0 commit comments