msqlconn.go

package msqlconn

import (
	"context"
	"database/sql"
	"fmt"
	"os"
	"time"

	"github.com/sivaosorg/govm/common"
	"github.com/sivaosorg/govm/dbx"
	"github.com/sivaosorg/govm/logger"
	"github.com/sivaosorg/govm/mysql"
	"github.com/sivaosorg/govm/utils"

	_ "github.com/go-sql-driver/mysql"
)

var (
	_logger = logger.NewLogger()
)

func NewMySql() *MySql {
	m := &MySql{}
	return m
}

func (m *MySql) SetConn(value *sql.DB) *MySql {
	m.conn = value
	return m
}

func (m *MySql) SetConfig(value mysql.MysqlConfig) *MySql {
	m.Config = value
	return m
}

func (m *MySql) SetState(value dbx.Dbx) *MySql {
	m.State = value
	return m
}

func (m *MySql) Json() string {
	return utils.ToJson(m)
}

func (m *MySql) GetConn() *sql.DB {
	return m.conn
}

func (m *MySql) Close() error {
	return m.conn.Close()
}

func NewClient(config mysql.MysqlConfig) (*MySql, dbx.Dbx) {
	instance := NewMySql()
	s := dbx.NewDbx().SetDatabase(config.Database)
	if !config.IsEnabled {
		s.SetConnected(false).
			SetMessage("Mysql unavailable").
			SetError(fmt.Errorf(s.Message))
		instance.SetState(*s)
		return instance, *s
	}
	client, err := sql.Open(common.EntryKeyMysql, Dsn(config))
	if err != nil {
		s.SetConnected(false).SetError(err).SetMessage(err.Error())
		instance.SetState(*s)
		return instance, *s
	}
	if config.MaxOpenConn <= 0 {
		config.MaxOpenConn = 10
	}
	if config.MaxIdleConn <= 0 {
		config.MaxIdleConn = 5
	}
	if config.MaxLifeTimeMinutesConn <= 0 {
		config.MaxLifeTimeMinutesConn = 5
	}
	client.SetMaxOpenConns(config.MaxOpenConn)
	client.SetMaxIdleConns(config.MaxIdleConn)
	client.SetConnMaxLifetime(time.Duration(time.Duration(config.MaxLifeTimeMinutesConn).Minutes()))
	ctx, cancel := context.WithTimeout(context.Background(), config.Timeout)
	defer cancel()
	err = client.PingContext(ctx)
	if err != nil {
		s.SetConnected(false).SetError(err).SetMessage(err.Error())
		instance.SetState(*s)
		return instance, *s
	}
	if config.DebugMode {
		_logger.Info(fmt.Sprintf("Mysql client connection:: %s", config.Json()))
		_logger.Info(fmt.Sprintf("Connected successfully to mysql:: %s (database: %s)", Dsn(config), config.Database))
	}
	pid := os.Getpid()
	s.SetConnected(true).SetMessage("Connected successfully").SetPid(pid).SetNewInstance(true)
	instance.SetConn(client).SetConfig(config).SetState(*s)
	return instance, *s
}

func Dsn(config mysql.MysqlConfig) string {
	hostname := fmt.Sprintf("%s:%d", config.Host, config.Port)
	return fmt.Sprintf("%s:%s@tcp(%s)/%s?parseTime=true", config.Username, config.Password, hostname, config.Database)
}