gorm连接mysql的初始化配置

包含mysql配置、gorm配置、连接池配置、log日志配置

init_db_log.go文件代码

package main

import (
	"fmt"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
	"time"
)

const (
	host   = "xxx"
	port   = 3306
	user   = "root"
	passwd = "xxxx"
	dbName = "topgoer_gorm"
	other  = "charset=utf8mb4&parseTime=True&loc=Local"
)

type Writer struct {
}

func (w Writer) Printf(format string, args ...interface{}) {
	slowSql := fmt.Sprintf(format, args...)
	// 出可以将慢查询信息写入到sentry日志当中去
	fmt.Println(slowSql)
	/*
		C:/Users/mayanan/Desktop/pro_go/topgoer_gorm/main.go:53 SLOW SQL >= 80ms
		[235.827ms] [rows:3] SELECT * FROM `users` WHERE `users`.`deleted` IS NULL
	*/
}

func InitDB() *gorm.DB {
	dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?%s", user, passwd, host, port, dbName, other)

	newLogger := logger.New(
		Writer{},
		logger.Config{
			SlowThreshold:             200 * time.Millisecond, // Slow SQL threshold
			LogLevel:                  logger.Warn,           // Log level
			IgnoreRecordNotFoundError: true,                  // Ignore ErrRecordNotFound error for logger
			Colorful:                  true,                 // Disable color
		},
	)

	// MySQl 驱动程序提供了 一些高级配置 可以在初始化过程中使用
	db, err := gorm.Open(mysql.New(mysql.Config{
		DSN:                       dsn,
		DefaultStringSize:         256,   // string 类型字段的默认长度
		DisableDatetimePrecision:  true,  // 禁用 datetime 精度，MySQL 5.6 之前的数据库不支持
		DontSupportRenameIndex:    true,  // 重命名索引时采用删除并新建的方式，MySQL 5.7 之前的数据库和 MariaDB 不支持重命名索引
		DontSupportRenameColumn:   true,  // 用 `change` 重命名列，MySQL 8 之前的数据库和 MariaDB 不支持重命名列
		SkipInitializeWithVersion: false, // 根据当前 MySQL 版本自动配置
	}), &gorm.Config{
		// 使用CreateBatchSize 选项初始化 GORM 时，所有的创建& 关联 INSERT 都将遵循该选项
		CreateBatchSize: 1000,
		// 全局模式：执行任何 SQL 时都创建并缓存预编译语句，可以提高后续的调用速度
		PrepareStmt: true,
		// 注意 QueryFields 模式会根据当前 model 的所有字段名称进行 select。
		//QueryFields: true,
		Logger: newLogger,
		// 表名加前缀和禁用复数表名
		NamingStrategy: schema.NamingStrategy{
			TablePrefix: "t_",   // 表名前缀，`User`表为`t_users`
			SingularTable: true, // 使用单数表名，启用该选项后，`User` 表将是`user`
		},
	})

	if err != nil {
		fmt.Println("conn mysql fail", err)
		return nil
	}

	sqlDb, _ := db.DB()
	// 对于中小型 web 应用程序，我通常使用以下设置作为起点，然后根据负载测试结果和实际吞吐量级别进行优化。
	// SetMaxIdleConns: 设置空闲连接池中链接的最大数量
	sqlDb.SetMaxIdleConns(25)
	// SetMaxOpenConns: 设置打开数据库链接的最大数量
	sqlDb.SetMaxOpenConns(25)
	// SetConnMaxLifetime: 设置链接可复用的最大时间
	sqlDb.SetConnMaxLifetime(5 * time.Minute)

	return db

}

main.py文件

package main

import (
	"database/sql"
	"fmt"
	uuid "github.com/satori/go.uuid"
	"gorm.io/gorm"
	"time"
)

func myPrint(p interface{}) {
	fmt.Printf("type: %T\tvalue: %v\n", p, p)
}

type User struct {
	ID        int            `gorm:"primaryKey;autoIncrement"`
	Name      sql.NullString `gorm:"default:'隔壁老王'"`
	Age       uint8          `gorm:"default:55"`
	UUID      uuid.UUID
	CreatedAt time.Time `gorm:"autoCreateTime"`
	Deleted gorm.DeletedAt `gorm:"autoDeleteTime"`
}
/*
	func (u *User) AfterFind(tx *gorm.DB) (err error) {
		fmt.Println("User AfterFind执行了")
		return
	}
*/
type Email struct {
	UserID int
	Email string
}
type SlowSql struct {
	FileRow string
	Slow string
	Ms float64
	Rows int64
	Sql string
	gorm.Model
}

var db *gorm.DB
func main() {
	db = InitDB()

	// 自动检查User结构是否发生变化，变化则进行迁移
	db.AutoMigrate(&User{})
	db.AutoMigrate(&Email{})
	db.AutoMigrate(&SlowSql{})

	// Context
	var users []User
	db.Debug().Find(&users)

}