package db

import (
	"database/sql"
	"fmt"
	message "m2pool-payment/internal/msg"

	_ "github.com/go-sql-driver/mysql"
)

type MySQLPool struct {
	db *sql.DB
}

// NewMySQLPool 初始化连接池
func NewMySQLPool(cfg message.MysqlConfig) (*MySQLPool, error) {
	dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local",
		cfg.User, cfg.Password, cfg.Host, cfg.Port, cfg.Database)
	db, err := sql.Open("mysql", dsn)
	if err != nil {
		return nil, err
	}

	// 设置连接池参数
	db.SetMaxOpenConns(cfg.MaxOpenConns)
	db.SetMaxIdleConns(cfg.MaxIdleConns)
	db.SetConnMaxLifetime(cfg.ConnMaxLife)

	// 测试连接
	if err := db.Ping(); err != nil {
		return nil, err
	}

	return &MySQLPool{db: db}, nil
}

// Exec 执行 INSERT/UPDATE/DELETE
func (p *MySQLPool) Exec(query string, args ...any) (sql.Result, error) {
	return p.db.Exec(query, args...)
}

// Query 查询多行
func (p *MySQLPool) Query(query string, args ...any) (*sql.Rows, error) {
	return p.db.Query(query, args...)
}

// QueryRow 查询单行
func (p *MySQLPool) QueryRow(query string, args ...any) *sql.Row {
	return p.db.QueryRow(query, args...)
}

// Transaction 执行事务
func (p *MySQLPool) Transaction(fn func(tx *sql.Tx) error) error {
	tx, err := p.db.Begin()
	if err != nil {
		return err
	}

	if err := fn(tx); err != nil {
		_ = tx.Rollback()
		return err
	}

	return tx.Commit()
}

// Close 关闭连接池
func (p *MySQLPool) Close() error {
	return p.db.Close()
}