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

package database

import (
	"log"
	"os"
	"strconv"
	"strings"
	"time"

	"soul-api/internal/model"

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

var db *gorm.DB

// Init 使用 DSN 连接 MySQL，供 handler 通过 DB() 使用
func Init(dsn string) error {
	// 慢查询阈值：默认 5 秒，避免 GORM 默认 200ms 导致控制台刷屏；可通过 SLOW_SQL_THRESHOLD_MS 覆盖
	slowMs := 5000
	if s := os.Getenv("SLOW_SQL_THRESHOLD_MS"); s != "" {
		if n, e := strconv.Atoi(s); e == nil && n > 0 {
			slowMs = n
		}
	}
	gormLogger := logger.New(
		log.New(os.Stdout, "\r\n", log.LstdFlags),
		logger.Config{
			SlowThreshold:             time.Duration(slowMs) * time.Millisecond,
			IgnoreRecordNotFoundError: true,
			Colorful:                  true,
		},
	)

	var err error
	db, err = gorm.Open(mysql.Open(dsn), &gorm.Config{Logger: gormLogger})
	if err != nil {
		return err
	}

	skipMigrate := strings.ToLower(strings.TrimSpace(os.Getenv("SKIP_AUTO_MIGRATE")))
	if skipMigrate == "1" || skipMigrate == "true" || skipMigrate == "yes" {
		log.Println("database: SKIP_AUTO_MIGRATE enabled, skipping schema migration")
		log.Println("database: connected")
		return nil
	}

	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.CkbSubmitRecord{}); err != nil {
		log.Printf("database: ckb_submit_records migrate warning: %v", err)
	}
	if err := db.AutoMigrate(&model.CkbLeadRecord{}); err != nil {
		log.Printf("database: ckb_lead_records migrate warning: %v", err)
	}
	if err := db.AutoMigrate(&model.Person{}); err != nil {
		log.Printf("database: persons migrate warning: %v", err)
	}
	if err := db.AutoMigrate(&model.LinkTag{}); err != nil {
		log.Printf("database: link_tags migrate warning: %v", err)
	}
	if err := db.AutoMigrate(&model.UserBalance{}); err != nil {
		log.Printf("database: user_balances migrate warning: %v", err)
	}
	if err := db.AutoMigrate(&model.BalanceTransaction{}); err != nil {
		log.Printf("database: balance_transactions migrate warning: %v", err)
	}
	if err := db.AutoMigrate(&model.GiftUnlock{}); err != nil {
		log.Printf("database: gift_unlocks migrate warning: %v", err)
	}
	// 以下表业务大量使用，必须参与 AutoMigrate，否则旧库缺字段会导致订单/用户/VIP 等接口报错
	if err := db.AutoMigrate(&model.User{}); err != nil {
		log.Printf("database: users migrate warning: %v", err)
	}
	if err := db.AutoMigrate(&model.SystemConfig{}); err != nil {
		log.Printf("database: system_config migrate warning: %v", err)
	}
	if err := db.AutoMigrate(&model.Chapter{}); err != nil {
		log.Printf("database: chapters migrate warning: %v", err)
	}
	if err := db.AutoMigrate(&model.UserRule{}); err != nil {
		log.Printf("database: user_rules migrate warning: %v", err)
	}
	if err := db.AutoMigrate(&model.UserTrack{}); err != nil {
		log.Printf("database: user_tracks migrate warning: %v", err)
	}
	seedDefaultRules(db)
	seedHistoryTracks(db)
	fixBrokenImageUrls(db)
	log.Println("database: connected")
	return nil
}

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

func seedDefaultRules(d *gorm.DB) {
	var count int64
	d.Model(&model.UserRule{}).Count(&count)
	if count > 0 {
		return
	}
	defaults := []model.UserRule{
		{Title: "注册完成 → 填写头像", Description: "用户完成注册后，引导填写头像和昵称", Trigger: "注册", Sort: 10, Enabled: true},
		{Title: "完成匹配 → 补充个人资料", Description: "完成派对房匹配后，引导填写 MBTI、行业、职位", Trigger: "完成匹配", Sort: 20, Enabled: true},
		{Title: "首次浏览章节 → 绑定手机号", Description: "点击阅读收费章节时，引导绑定手机号", Trigger: "点击收费章节", Sort: 30, Enabled: true},
		{Title: "付款 ¥1980 → 填写完整信息", Description: "购买全书后，需填写完整信息以进入 VIP 群", Trigger: "完成付款", Sort: 40, Enabled: true},
		{Title: "加入派对房 → 填写项目介绍", Description: "进入派对房前，引导填写项目介绍和核心需求", Trigger: "加入派对房", Sort: 50, Enabled: true},
		{Title: "浏览 5 个章节 → 分享推广", Description: "累计阅读 5 个章节后，触发分享引导", Trigger: "累计浏览5章节", Sort: 60, Enabled: true},
		{Title: "绑定微信 → 开启分销", Description: "绑定微信后，引导开启分销功能", Trigger: "绑定微信", Sort: 70, Enabled: true},
		{Title: "收益达到 ¥50 → 申请提现", Description: "累计分销收益超过 50 元时引导提现", Trigger: "收益满50元", Sort: 80, Enabled: true},
		{Title: "完善存客宝信息 → 进入流量池", Description: "引导授权存客宝信息同步，进入微信流量池", Trigger: "手动触发", Sort: 90, Enabled: true},
		{Title: "浏览导师主页 → 预约咨询", Description: "浏览导师详情页超过 30 秒，引导预约咨询", Trigger: "浏览导师页", Sort: 100, Enabled: true},
	}
	if err := d.CreateInBatches(&defaults, len(defaults)).Error; err != nil {
		log.Printf("database: seed user_rules warning: %v", err)
	}
}

// fixBrokenImageUrls 修复数据库中 URL 缺少冒号的脏数据（"https//..." → "https://..."）
func fixBrokenImageUrls(d *gorm.DB) {
	cols := []struct{ table, col string }{
		{"users", "avatar"},
		{"users", "vip_avatar"},
		{"author_config", "avatar_img"},
		{"mentors", "avatar"},
	}
	for _, c := range cols {
		res := d.Exec(
			"UPDATE "+c.table+" SET "+c.col+" = REPLACE("+c.col+", 'https//', 'https://') WHERE "+c.col+" LIKE 'https//%'",
		)
		if res.RowsAffected > 0 {
			log.Printf("database: fixed %d broken URL(s) in %s.%s", res.RowsAffected, c.table, c.col)
		}
		res = d.Exec(
			"UPDATE "+c.table+" SET "+c.col+" = REPLACE("+c.col+", 'http//', 'http://') WHERE "+c.col+" LIKE 'http//%'",
		)
		if res.RowsAffected > 0 {
			log.Printf("database: fixed %d broken http URL(s) in %s.%s", res.RowsAffected, c.table, c.col)
		}
	}
}

func seedHistoryTracks(d *gorm.DB) {
	var trackCount int64
	d.Model(&model.UserTrack{}).Count(&trackCount)
	if trackCount > 5 {
		return
	}
	// 为所有已有用户回填 register track
	d.Exec(`INSERT IGNORE INTO user_tracks (id, user_id, action, created_at)
		SELECT CONCAT('seed_reg_', id), id, 'register', created_at FROM users
		WHERE id NOT IN (SELECT user_id FROM user_tracks WHERE action = 'register')`)
	// 为已绑定手机的用户回填 bind_phone track
	d.Exec(`INSERT IGNORE INTO user_tracks (id, user_id, action, created_at)
		SELECT CONCAT('seed_phone_', id), id, 'bind_phone', updated_at FROM users
		WHERE phone IS NOT NULL AND phone != ''
		AND id NOT IN (SELECT user_id FROM user_tracks WHERE action = 'bind_phone')`)
	// 为有订单的用户回填 purchase track
	d.Exec(`INSERT IGNORE INTO user_tracks (id, user_id, action, created_at)
		SELECT CONCAT('seed_pay_', o.user_id), o.user_id, 'purchase', MIN(o.created_at)
		FROM orders o WHERE o.status IN ('paid','success','completed')
		AND o.user_id NOT IN (SELECT user_id FROM user_tracks WHERE action = 'purchase')
		GROUP BY o.user_id`)
	log.Println("database: seeded history tracks from existing data")
}