package postgresql

import (
	"gitee.com/netany/quick/console"
	"gitee.com/netany/quick/data/db"
	"gitee.com/netany/quick/helper"
	loggerx "gitee.com/netany/quick/logger"
	"github.com/gookit/goutil/errorx"
	"gorm.io/driver/postgres"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
	"gorm.io/gorm/schema"
)

var DB *gorm.DB
var err error

// NewPostgresqlDb 初始化pgsql数据库
func NewPostgresqlDb(conf *Postgresql, logConfig *loggerx.LogConfig) *gorm.DB {
	gormConf := &gorm.Config{
		SkipDefaultTransaction: true, // 关闭默认单个创建、更新、删除操作使用事务
		NamingStrategy: schema.NamingStrategy{
			SingularTable: true,
			TablePrefix:   conf.DbPrefix,
		},
		Logger: db.NewGormLogger(logger.Config{
			LogLevel: helper.LoggerLevelToDbLogLevel(logConfig.Level),
			Colorful: true,
		}, logConfig.Output),
	}
	DB, err = gorm.Open(postgres.Open(conf.DSN), gormConf)

	if err != nil {
		panic(errorx.With(err, "postgresql启动异常"))
	}

	conn, err := DB.DB()
	if err != nil {
		panic(errorx.With(err, "postgresql启动异常"))
	}
	console.Logger.Info("postgresql init success")
	conn.SetMaxIdleConns(10)
	conn.SetMaxOpenConns(151)
	return DB
}

type Postgresql struct {
	DSN      string `json:"dsn"`
	DbPrefix string `json:"dbPrefix"`
}