package go_common

import (
	"database/sql"
	"fmt"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
	"gorm.io/gorm/schema"
	"time"
)
type BaseModel struct {
	Id int64 `gorm:"PRIMARY_KEY;AUTO_INCREMENT" json:"id,omitempty" `
	CreatedAt *MyTime `json:"created_at,omitempty"`
	UpdatedAt *MyTime `json:"updated_at,omitempty"`
	DeletedAt *gorm.DeletedAt `gorm:"index" json:"deleted_at,omitempty"`
}
type CommModel struct {
	Id int64 `gorm:"PRIMARY_KEY;AUTO_INCREMENT" json:"id,omitempty" `
	CreatedAt *MyTime `json:"created_at,omitempty"`
	UpdatedAt *MyTime `json:"updated_at,omitempty"`
}
var db *gorm.DB
var sqlDB *sql.DB

func InitDataBase(username,password,host,database, prefix string, logMode logger.LogLevel, maxIdle, maxOpen int, models ...interface{}) (err error) {
	dsn := fmt.Sprintf("%v:%v@tcp(%v)/%v?charset=utf8mb4&parseTime=True&loc=Local", username, password,host,database)
	db, err = gorm.Open(mysql.Open(dsn), &gorm.Config{
		NamingStrategy: schema.NamingStrategy{
			TablePrefix: prefix,
			SingularTable: true,
		},
		Logger: logger.Default.LogMode(logMode),
	})
	if err != nil {
		return err
	}
	// 自动迁移
	if err = db.AutoMigrate(models ...); err != nil {
		return err
	}
	// 连接池设置
	if sqlDB, err = db.DB(); err != nil {
		return err
	} else {
		sqlDB.SetConnMaxIdleTime(time.Hour)
		sqlDB.SetMaxIdleConns(maxIdle)
		sqlDB.SetMaxOpenConns(maxOpen)
	}

	return nil
}
func GetDb() *gorm.DB {

	return db
}

func CloseDb()  {
	if sqlDB.Ping() != nil {
		return
	}
	sqlDB.Close()
}