FranklinWH-Toolbox/service/database/mysql.go

package database

import (
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"

	zapLogger "fwh-toolbox-server/logger"
	"fwh-toolbox-server/models/fwh_pro/query_fwh_pro"
)

var MySQL *gorm.DB

func InitMySQL(dsn string) (*gorm.DB, error) {
	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{
		// Logger: logger.Default.LogMode(logger.Info),
		Logger: NewGormZapLogger(zapLogger.Sugar, logger.Info),
	})
	if err != nil {
		zapLogger.Sugar.Errorf("failed to connect to MySQL: %v", err)
		return nil, err
	}
	MySQL = db

	// 初始化Query
	query_fwh_pro.SetDefault(db)

	return db, nil
}

// 示例DSN格式:
// user:pass@tcp(127.0.0.1:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local&readTimeout=30s