package conf

import (
	"context"
	"database/sql"
	"fmt"
	_ "github.com/go-sql-driver/mysql"
	"sync"
	"time"
)

var db *sql.DB
var once sync.Once

/*
主要用于获取一个新的数据库连接。
处理连接池的配置，设置最大打开连接数、最大空闲连接数等。
使用 context 进行连接测试，确保连接的可用性。
返回一个新的数据库连接和可能的错误。
*/
func (m *MySQL) getDBConn() (*sql.DB, error) {
	var err error
	dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8&multiStatements=true", m.UserName, m.Password, m.Host, m.Port, m.Database)

	conn, err := sql.Open("mysql", dsn)
	if err != nil {
		return nil, fmt.Errorf("GetDBConn failed.. ")
	}

	// 连接池参数配置
	//最大打开连接数、最大空闲连接数、连接的最大生命周期以及连接的最大空闲时间。
	conn.SetMaxOpenConns(m.MaxOpenConn)
	conn.SetMaxIdleConns(m.MaxIdleConn)
	conn.SetConnMaxLifetime(time.Duration(m.MaxLifeTime) * time.Second)
	conn.SetConnMaxIdleTime(time.Duration(m.MaxIdleTime) * time.Second)

	//context 用于处理请求的上下文的标准库，它提供了一种在不同 Goroutine 之间传递请求特定数据、取消信号和截止时间的方法。
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	if err = conn.PingContext(ctx); err != nil {
		return nil, fmt.Errorf("ping mysql failed")
	}
	return conn, nil
}

// 主要用于获取数据库连接，并维护一个全局的数据库连接，确保多次调用返回的是同一个连接。
func (m *MySQL) GetDB() (*sql.DB, error) {
	var err error
	once.Do(func() {
		conn, e := m.getDBConn()
		if e != nil {
			//fmt.Println("getDBConn failed: %s\n", err)
			err = e
			return
		}
		db = conn
	})
	return db, err
}

/*func D() *sql.DB {
	if db == nil {
		panic("db empty..")
	}
	return db
}*/