package util

import (
	"log"

	elastic "app.yhyue.com/moapp/jybase/esv7"
	"app.yhyue.com/moapp/jybase/mysql"
	"gorm.io/gorm"
)

type MysqlDBConfig struct {
	DriverName      string
	DataSourceName  string
	MaxOpenConn     int
	MaxIdleConn     int
	MaxConnLifeTime int
}

type EsConfig struct {
	Addr string
	Pool int
}

var (
	jyDocsDB *gorm.DB
)

func InitDB(url, driverName string, maxOpenConn, maxIdle int) {
	jyDocsDB = mysql.GormMysql(url, driverName, maxOpenConn, maxIdle, nil)
	if jyDocsDB != nil {
		log.Printf("----------->【jy_docs】 DB :[%s] 初始化成功!<--------------", url)
	} else {
		log.Fatalf("----------->【jy_docs】 DB初始化失败<--------------")
	}

}
func InitEs(addr string, poolSize int) {
	log.Printf("----------->【jy_docs】 elastic :[%s] init<--------------", addr)
	elastic.InitElasticSize(addr, poolSize)
}
func GetJyDocsDB() *gorm.DB {
	return jyDocsDB
}