package data

import (
	"database/sql"
	"gitee.com/kristas/booting-go/framework/core/bean"
)

type Configure interface {
	bean.Bean
	Config() Config
}

type Config struct {
	Type            string `yaml:"type"`
	Host            string `yaml:"host"`
	Database        string `yaml:"database"`
	Username        string `yaml:"username"`
	Password        string `yaml:"password"`
	Tunnel          string `yaml:"tunnel"`
	ConnMaxLifeTime int64  `yaml:"max_life_time"`
	ConnMaxIdleTime int64  `yaml:"max_idle_time"`
	MaxIdleConns    int    `yaml:"max_idle_conn"`
	MaxOpenConns    int    `yaml:"max_open_conn"`
}

type DatasourceWrapper interface {
	DB() *sql.DB
	Configure() Configure
}

func NewDatasourceWrapper(DB *sql.DB, configure Configure) DatasourceWrapper {
	return &DatasourceWrapperImpl{
		db:        DB,
		configure: configure,
	}
}

type DatasourceWrapperImpl struct {
	db        *sql.DB
	configure Configure
}

func (d *DatasourceWrapperImpl) DB() *sql.DB {
	return d.db
}

func (d *DatasourceWrapperImpl) Configure() Configure {
	return d.configure
}