package database

import (
	"log"

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

	fwh_pro_query "fwh-toolbox-server/models/fwh_pro/query"
)

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),
	})
	if err != nil {
		log.Printf("failed to connect to MySQL: %v", err)
		return nil, err
	}
	MySQL = db

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

	return db, nil
}

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