增加mysql的loglevel参数

Author: chenxi2015

config/config.go

@@ -4,6 +4,7 @@ import (
 	"bytes"
 	"embed"
 	"github.com/spf13/viper"
+	"gorm.io/gorm/logger"
 )
 
 // ConfEnv env环境变量
@@ -37,15 +38,16 @@ type (
 		DirPath  string `yaml:"dirPath" default:"runtime/logs"`
 	}
 	Mysql []struct {
-		Host         string `yaml:"host" default:"127.0.0.1"`
-		Port         string `yaml:"port" default:"3306"`
-		User         string `yaml:"user" default:"root"`
-		Password     string `yaml:"password" default:"123456"`
-		DbName       string `yaml:"dbname"`
-		Prefix       string `yaml:"prefix" default:""`
-		MaxIdleConns int    `yaml:"maxIdleConns" default:"10"`
-		MaxOpenConns int    `yaml:"maxOpenConns" default:"100"`
-		MaxLifeTime  int    `yaml:"maxLifeTime" default:"60"`
+		Host         string          `yaml:"host" default:"127.0.0.1"`
+		Port         string          `yaml:"port" default:"3306"`
+		User         string          `yaml:"user" default:"root"`
+		Password     string          `yaml:"password" default:"123456"`
+		DbName       string          `yaml:"dbname"`
+		Prefix       string          `yaml:"prefix" default:""`
+		MaxIdleConns int             `yaml:"maxIdleConns" default:"10"`
+		MaxOpenConns int             `yaml:"maxOpenConns" default:"100"`
+		MaxLifeTime  int             `yaml:"maxLifeTime" default:"60"`
+		LogLevel     logger.LogLevel `yaml:"logLevel" default:"1"`
 	}
 	Redis struct {
 		Host        string `yaml:"host" default:"127.0.0.1"`

config/yaml/config.dev.yaml

@@ -22,6 +22,7 @@ mysql:
     maxIdleConns: 10 # 设置空闲连接池中连接的最大数量
     maxOpenConns: 100 # 设置打开数据库连接的最大数量
     maxLifeTime: 60 # 设置了连接可复用的最大时间（分钟）
+    logLevel: 4 # 日志模式 1:slient 2:error 3:warn 4:info
 redis:
   host: '127.0.0.1'
   port: '6379'

config/yaml/config.prod.yaml

@@ -22,6 +22,7 @@ mysql:
     maxIdleConns: 10 # 设置空闲连接池中连接的最大数量
     maxOpenConns: 100 # 设置打开数据库连接的最大数量
     maxLifeTime: 60 # 设置了连接可复用的最大时间（分钟）
+    logLevel: 1 # 日志模式 1:slient 2:error 3:warn 4:info
 redis:
   host: '127.0.0.1'
   port: '6379'

config/yaml/config.test.yaml

@@ -22,6 +22,7 @@ mysql:
     maxIdleConns: 10 # 设置空闲连接池中连接的最大数量
     maxOpenConns: 100 # 设置打开数据库连接的最大数量
     maxLifeTime: 60 # 设置了连接可复用的最大时间（分钟）
+    logLevel: 3 # 日志模式 1:slient 2:error 3:warn 4:info
 redis:
   host: '127.0.0.1'
   port: '6379'

pkg/lib/mysql.go

@@ -1,9 +1,11 @@
 package lib
 
 import (
+	"database/sql"
 	"fmt"
 	"gorm.io/driver/mysql"
 	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
 	"gorm.io/gorm/schema"
 	"time"
 )
@@ -18,6 +20,7 @@ type DatabaseConfig struct {
 	MaxIdleConns int
 	MaxOpenConns int
 	MaxLifeTime  int // 分钟
+	LogLevel     logger.LogLevel
 }
 
 // NewMysql 数据库连接
@@ -28,6 +31,7 @@ func NewMysql(config DatabaseConfig) (*gorm.DB, error) {
 			TablePrefix:   config.Prefix,
 			SingularTable: true, // 是否设置单数表名，设置为 是
 		},
+		Logger: logger.Default.LogMode(config.LogLevel),
 	})
 	if err != nil {
 		return nil, fmt.Errorf("Unable to connect to the database, please check the MySQL configuration information first,the error details are:" + err.Error())
@@ -42,3 +46,25 @@ func NewMysql(config DatabaseConfig) (*gorm.DB, error) {
 	sqlDB.SetConnMaxLifetime(time.Duration(config.MaxLifeTime))
 	return db, nil
 }
+
+// NewNullString
+func NewNullString(s string) sql.NullString {
+	if len(s) == 0 {
+		return sql.NullString{}
+	}
+	return sql.NullString{
+		String: s,
+		Valid:  true,
+	}
+}
+
+// NewNullInt64
+func NewNullInt64(s int64) sql.NullInt64 {
+	if s == 0 {
+		return sql.NullInt64{}
+	}
+	return sql.NullInt64{
+		Int64: s,
+		Valid: true,
+	}
+}