package mysql

import (
	"fmt"
	"gitee.com/eden-framework/envconfig"
	"gitee.com/eden-framework/sqlx"
	"gitee.com/eden-framework/sqlx/mysqlconnector"
	"io"
	"net/url"
	"time"
)

type MySQL struct {
	Name            string
	Host            string
	Port            int
	User            string
	Password        envconfig.Password
	Extra           string
	PoolSize        int
	ConnMaxLifetime envconfig.Duration
	Retry
	Database *sqlx.Database `json:"-" ignored:"true"`

	*sqlx.DB `json:"-" ignored:"true"`
}

func (m *MySQL) SetDefaults() {
	if m.Port == 0 {
		m.Port = 3306
	}

	if m.PoolSize == 0 {
		m.PoolSize = 10
	}

	if m.ConnMaxLifetime == 0 {
		m.ConnMaxLifetime = envconfig.Duration(1 * time.Hour)
	}

	if m.Extra == "" {
		values := url.Values{}
		values.Set("charset", "utf8")
		values.Set("parseTime", "true")
		values.Set("interpolateParams", "true")
		values.Set("autocommit", "true")
		values.Set("loc", "Local")
		m.Extra = values.Encode()
	}
}

func (m *MySQL) URL() string {
	password := m.Password
	if password != "" {
		password = ":" + password
	}
	return fmt.Sprintf("%s%s@tcp(%s:%d)", m.User, password, m.Host, m.Port)
}

func (m *MySQL) Connect() error {
	m.SetDefaults()
	db := m.Database.OpenDB(
		&mysqlconnector.MysqlConnector{
			Host:  m.URL(),
			Extra: m.Extra,
		},
	)
	db.SetMaxOpenConns(m.PoolSize)
	db.SetMaxIdleConns(m.PoolSize / 2)
	db.SetConnMaxLifetime(time.Duration(m.ConnMaxLifetime))
	m.DB = db

	return nil
}

func (m *MySQL) Init() {
	if m.DB == nil {
		m.Do(m.Connect)
	}
}

func (m *MySQL) Refresh() {
	// TODO use pool to prevent close the connection immediately
	if m.DB != nil {
		if closer, ok := m.DB.SqlExecutor.(io.Closer); ok {
			closer.Close()
		}
	}
	m.Do(m.Connect)
}

func (m *MySQL) Get() *sqlx.DB {
	if m.DB == nil {
		panic(fmt.Errorf("get db before init"))
	}
	return m.DB
}