package starter

import (
	"gitee.com/kristas/booting-go/framework/bean/bean_factory"
	. "gitee.com/kristas/booting-go/framework/common/database/mysql"
	ssh_client "gitee.com/kristas/booting-go/framework/common/database/net_conn/ssh"
	"gorm.io/gorm"
)

type GormDataSourceStarter struct {
	DataSources []DataSource   `value:"datasource"`
	SSH         ssh_client.SSH `value:"ssh"`
	LoggingLevel string `value:"logging.level"`
}

func (r *GormDataSourceStarter) Bean() string {
	return ""
}

func (r *GormDataSourceStarter) AfterPropertiesSet() {
	for i := range r.DataSources {
		dataSource := r.DataSources[i]
		bean_factory.RegisterWithName(dataSource.Tag, new(gorm.DB))
	}
}

func (r *GormDataSourceStarter) Init() {
	if len(r.DataSources) > 0 {
		var isUseSSH = false
		if r.SSH.IP != "" && r.SSH.Username != "" && r.SSH.Password != "" {
			ssh_client.InitSSHTunnel(r.SSH)
			isUseSSH = true
		}
		InitGormDatasource(r.DataSources,isUseSSH,r.LoggingLevel)
	}
}