// Copyright (c) 2020 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

/**
*@describe:
*@author wfl19/Kristas
*@date 2021/04/10
 */

package datasource_processor

import (
	"database/sql"
	"fmt"
	"gitee.com/kristas/booting-go/framework/data"
	"github.com/go-sql-driver/mysql"
	"time"
)

func init() {
	UseDataBaseHandler("mysql", buildMysqlConn)
}

func buildMysqlConn(d *DatasourceProcessor, configure data.Configure) data.DatasourceWrapper {
	switch configure.Config().Tunnel {
	case "ssh":
		mysql.RegisterDialContext("mysql+ssh", d.sshDialer.Dial)
	case "tcp":
		mysql.RegisterDialContext("mysql+tcp", d.tcpDialer.Dial)
	}
	config := configure.Config()
	dsn := fmt.Sprintf("%s:%s@mysql+%s(%s)%s",
		config.Username, config.Password, config.Tunnel, config.Host, config.Database)
	conn, err := sql.Open("mysql", dsn)
	if err != nil {
		d.Log.Panicf("Failed to open mysql connection:\ncaused by:\ndsn: %s\n%s", dsn, err.Error())
	}
	if err = conn.Ping(); err != nil {
		d.Log.Panicf("ping mysql failed:\ncaused by:\n%s", err.Error())
	}
	initConn(config, conn)

	d.Log.Info("build mysql connection")
	datasourceWrapper := data.NewDatasourceWrapper(conn, configure)
	return datasourceWrapper
}

func initConn(config data.Config, conn *sql.DB) {
	if config.ConnMaxLifeTime != 0 {
		conn.SetConnMaxLifetime(time.Second * time.Duration(config.ConnMaxLifeTime))
	}
	if config.ConnMaxIdleTime != 0 {
		conn.SetConnMaxIdleTime(time.Second * time.Duration(config.ConnMaxIdleTime))
	}
	if config.MaxIdleConns != 0 {
		conn.SetMaxIdleConns(config.MaxIdleConns)
	}
	if config.MaxOpenConns != 0 {
		conn.SetMaxOpenConns(config.MaxOpenConns)
	}
}