package sqlitex

import (
	"github.com/jinzhu/gorm"
	_ "github.com/jinzhu/gorm/dialects/sqlite"
	"time"
)

type Conf struct {
	LogMode               bool
	MaxIdleCon            int64
	MaxOpenCon            int64
	ConnMaxLifeTimeSecond int64
	ConnMaxIdleTimeSecond int64
	Dsn                   string
	Host                  string
}

var openDb *gorm.DB

func Init(conf *Conf) error {
	db, err := gorm.Open("sqlite3", conf.Dsn)
	if err != nil {
		return err
	}
	sqlDb := db.DB()
	sqlDb.SetMaxIdleConns(int(conf.MaxIdleCon))
	sqlDb.SetMaxOpenConns(int(conf.MaxOpenCon))
	sqlDb.SetConnMaxLifetime(time.Duration(conf.ConnMaxLifeTimeSecond) * time.Second)
	sqlDb.SetConnMaxIdleTime(time.Duration(conf.ConnMaxIdleTimeSecond) * time.Second)
	openDb = db
	db.LogMode(true)
	return nil
}

func GetDB() *gorm.DB {
	return openDb
}