1
package utils
2

3
import (
4
  "fmt"
5
  "github.com/spf13/viper"
6
  "gorm.io/driver/mysql"
7
  "gorm.io/gorm"
8
  "gorm.io/gorm/logger"
9
  "log"
10
  "os"
11
  "time"
12
)
13

14
var DB *gorm.DB
15

16
var globalLogger = logger.New(
17
  log.New(os.Stdout, "\r\n", log.LstdFlags),
18
  logger.Config{
19
    SlowThreshold:             time.Second,
20
        // 自己切换logger的优先级，logger.Silent为不打印log
21
    LogLevel:                  logger.Info,
22
    IgnoreRecordNotFoundError: false,
23
    Colorful:                  true,
24
  },
25
)
26

27
func InitMySQL() {
28
  // 获取配置项
29
  username := viper.GetString("datasource.username")
30
  password := viper.GetString("datasource.password")
31
  host := viper.GetString("datasource.host")
32
  port := viper.GetString("datasource.port")
33
  database := viper.GetString("datasource.database")
34
  // 连接MySQL
35
  db, err := gorm.Open(mysql.New(mysql.Config{
36
    DSN: fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=true&loc=Local", username, password, host, port, database),
37
  }), &gorm.Config{
38
    // 开启logger
39
    Logger: globalLogger,
40
  })
41
  db = db.Session(&gorm.Session{Logger: globalLogger})
42
  if err != nil {
43
    panic(fmt.Sprintf("MySQL连接失败：%v", err))
44
  }
45
  fmt.Println("MySQL连接成功")
46
  DB = db
47
}
48

49
func GetMySQL() *gorm.DB {
50
  return DB
51
}