soul-yongping/soul-api/internal/database/database.go

package database

import (
	"log"

	"soul-api/internal/model"

	"gorm.io/driver/mysql"
	"gorm.io/gorm"
)

var db *gorm.DB

// Init 使用 DSN 连接 MySQL，供 handler 通过 DB() 使用
func Init(dsn string) error {
	var err error
	db, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
	if err != nil {
		return err
	}
	if err := db.AutoMigrate(&model.WechatCallbackLog{}); err != nil {
		log.Printf("database: wechat_callback_logs migrate warning: %v", err)
	}
	if err := db.AutoMigrate(&model.Withdrawal{}); err != nil {
		log.Printf("database: withdrawals migrate warning: %v", err)
	}
	if err := db.AutoMigrate(&model.MatchRecord{}); err != nil {
		log.Printf("database: match_records migrate warning: %v", err)
	}
	if err := db.AutoMigrate(&model.UserAddress{}); err != nil {
		log.Printf("database: user_addresses migrate warning: %v", err)
	}
	if err := db.AutoMigrate(&model.VipRole{}); err != nil {
		log.Printf("database: vip_roles migrate warning: %v", err)
	}
	if err := db.AutoMigrate(&model.Order{}); err != nil {
		log.Printf("database: orders migrate warning: %v", err)
	}
	if err := db.AutoMigrate(&model.Mentor{}); err != nil {
		log.Printf("database: mentors migrate warning: %v", err)
	}
	if err := db.AutoMigrate(&model.MentorConsultation{}); err != nil {
		log.Printf("database: mentor_consultations migrate warning: %v", err)
	}
	if err := db.AutoMigrate(&model.AuthorConfig{}); err != nil {
		log.Printf("database: author_config migrate warning: %v", err)
	}
	if err := db.AutoMigrate(&model.AdminUser{}); err != nil {
		log.Printf("database: admin_users migrate warning: %v", err)
	}
	if err := db.AutoMigrate(&model.UserRule{}); err != nil {
		log.Printf("database: user_rules migrate warning: %v", err)
	}
	log.Println("database: connected")
	return nil
}

// DB 返回全局 *gorm.DB，仅在 Init 成功后调用
func DB() *gorm.DB {
	return db
}