micro-bundle/pkg/db/mysql.go

package db

import (
	"fmt"
	"log"
	"time"

	"github.com/gin-gonic/gin"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
	"gorm.io/gorm/schema"

	"micro-bundle/internal/model"
)

func loadMysqlConn(conn string) *gorm.DB {
	var ormLogger logger.Interface
	if gin.Mode() == "debug" {
		ormLogger = logger.Default.LogMode(logger.Info)
	} else {
		ormLogger = logger.Default
	}
	db, err := gorm.Open(mysql.New(mysql.Config{
		DSN:                       conn,  // DSN data source name
		DefaultStringSize:         256,   // string 类型字段的默认长度
		DisableDatetimePrecision:  true,  // 禁用 datetime 精度，MySQL 5.6 之前的数据库不支持
		DontSupportRenameIndex:    true,  // 重命名索引时采用删除并新建的方式，MySQL 5.7 之前的数据库和 MariaDB 不支持重命名索引
		DontSupportRenameColumn:   true,  // 用 `change` 重命名列，MySQL 8 之前的数据库和 MariaDB 不支持重命名列
		SkipInitializeWithVersion: false, // 根据版本自动配置
	}), &gorm.Config{
		Logger: ormLogger,
		NamingStrategy: schema.NamingStrategy{
			SingularTable: true,
		},
		DisableForeignKeyConstraintWhenMigrating: true,
	})
	if err != nil {
		panic(err)
	}
	fmt.Println("[BundleDB] 数据库连接成功, 连接字符串: %s", conn)
	sqlDB, _ := db.DB()
	sqlDB.SetMaxIdleConns(20)  //设置连接池，空闲
	sqlDB.SetMaxOpenConns(100) //打开
	sqlDB.SetConnMaxLifetime(time.Second * 30)

	// Bundle数据库的自动迁移
	err = db.AutoMigrate(
		&model.BundleProfile{},
		&model.BundleOrderRecords{},
		&model.ValueAddBundleProfile{},
		//&model.ValueAddBundleRecord{}
		&model.BundleProfileLang{},
		&model.ValueAddService{},
		&model.ValueAddServiceLang{},
		&model.BundleToValueAddService{},
		&model.BundleProfileHistory{},
		&model.ValueAddServiceHistory{},
		&model.BundleOrderValueAdd{},
		&model.BundleExtensionRecords{},
		&model.BundleBalance{},
		&model.Reconciliation{},
	)

	if err != nil {
		// return nil
		panic(err)
	}
	return db
}

// TaskBench数据库的自动迁移函数
func loadTaskBenchMysqlConn(conn string) *gorm.DB {
	var ormLogger logger.Interface
	if gin.Mode() == "debug" {
		ormLogger = logger.Default.LogMode(logger.Info)
	} else {
		ormLogger = logger.Default
	}
	db, err := gorm.Open(mysql.New(mysql.Config{
		DSN:                       conn,
		DefaultStringSize:         256,
		DisableDatetimePrecision:  true,
		DontSupportRenameIndex:    true,
		DontSupportRenameColumn:   true,
		SkipInitializeWithVersion: false,
	}), &gorm.Config{
		Logger: ormLogger,
		NamingStrategy: schema.NamingStrategy{
			SingularTable: true,
		},
		DisableForeignKeyConstraintWhenMigrating: true,
	})
	if err != nil {
		panic(err)
	}
	sqlDB, _ := db.DB()
	sqlDB.SetMaxIdleConns(20)
	sqlDB.SetMaxOpenConns(100)
	sqlDB.SetConnMaxLifetime(time.Second * 30)

	// 自动迁移任务管理台相关数据库表
	err = db.AutoMigrate(
		&model.TaskManagement{},
		&model.TaskAssignRecords{},
	)

	if err != nil {
		log.Printf("[TaskBenchDB] 数据库迁移失败: %v", err)
		panic(err)
	}

	log.Printf("[TaskBenchDB] 数据库迁移成功，已创建表: task_management, task_assign_records")
	return db
}

func DBMigration() {
}

func addColumn(dst interface{}, column string) {
	return
}