package data

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

type Datasource interface {
	bean.Bean
	component.ConfigurationProperties
	GetType() string
	GetHost() string
	GetDatabase() string
	GetUsername() string
	GetPassword() string
	GetTunnel() string

	GetConnMaxLifeTime() int64
	GetConnMaxIdleTime() int64
	GetMaxIdleConns() int
	GetMaxOpenConns() int
	IsLazyFetch() bool
}

type DatasourceConfig struct {
	BeanName         string
	PropertiesPrefix string

	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"`
	LazyFetch       bool  `yaml:"lazy_fetch"`
}

func (d *DatasourceConfig) Bean() string {
	return presentDefault(d.BeanName, "datasource")
}

func (d *DatasourceConfig) Prefix() string {
	return presentDefault(d.PropertiesPrefix, "application.datasource")
}

func (d *DatasourceConfig) GetType() string {
	return d.Type
}

func (d *DatasourceConfig) GetHost() string {
	return d.Host
}

func (d *DatasourceConfig) GetDatabase() string {
	return d.Database
}

func (d *DatasourceConfig) GetUsername() string {
	return d.Username
}

func (d *DatasourceConfig) GetPassword() string {
	return d.Password
}

func (d *DatasourceConfig) GetTunnel() string {
	return d.Tunnel
}

func (d *DatasourceConfig) GetConnMaxLifeTime() int64 {
	return d.ConnMaxLifeTime
}

func (d *DatasourceConfig) GetConnMaxIdleTime() int64 {
	return d.ConnMaxIdleTime
}

func (d *DatasourceConfig) GetMaxIdleConns() int {
	return d.MaxIdleConns
}

func (d *DatasourceConfig) GetMaxOpenConns() int {
	return d.MaxOpenConns
}

func (d *DatasourceConfig) IsLazyFetch() bool {
	return d.LazyFetch
}

func presentDefault(s, defaults string) string {
	if s == "" {
		s = defaults
	}
	return s
}