re_configuration.go

package mysql

import (
	"context"
	"database/sql"
	"github.com/D3rus/go-library/configurator"
	"github.com/D3rus/go-library/data_base/client/driver"
	"github.com/D3rus/go-library/data_base/client/limiter"
	"github.com/D3rus/go-library/logger"
	"sync"
)

type ReConfiguration struct {
	mysql *ReConnect

	logger  logger.Logger
	limiter limiter.Limiter

	config  *Config
	rwMutex *sync.RWMutex
}

func (reConfiguration *ReConfiguration) DriverName() string {
	return driver.MySQL
}

func NewReConfigurationWithConfigurator(logger logger.Logger, limiter limiter.Limiter, config *Config, configurator configurator.Configurator) *ReConfiguration {
	return NewReConfiguration(logger, limiter, Configuration(config, configurator))
}

func NewReConfiguration(logger logger.Logger, limiter limiter.Limiter, config *Config) *ReConfiguration {
	return &ReConfiguration{
		logger:  logger,
		limiter: limiter,
		config:  config,
		rwMutex: &sync.RWMutex{},
		mysql:   NewReConnect(config, logger, limiter),
	}
}

func (reConfiguration *ReConfiguration) MySQL() *ReConnect {
	reConfiguration.rwMutex.RLock()
	defer reConfiguration.rwMutex.RUnlock()

	return reConfiguration.mysql
}

func (reConfiguration *ReConfiguration) Close() error {
	return reConfiguration.MySQL().Close()
}

func (reConfiguration *ReConfiguration) Query(s string, i ...interface{}) (*sql.Rows, error) {
	return reConfiguration.MySQL().Query(s, i...)
}

func (reConfiguration *ReConfiguration) QueryRow(s string, i ...interface{}) *sql.Row {
	return reConfiguration.MySQL().QueryRow(s, i...)
}

func (reConfiguration *ReConfiguration) Exec(s string, i ...interface{}) (sql.Result, error) {
	return reConfiguration.MySQL().Exec(s, i...)
}

func (reConfiguration *ReConfiguration) QueryContext(ctx context.Context, s string, i ...interface{}) (*sql.Rows, error) {
	return reConfiguration.MySQL().QueryContext(ctx, s, i...)
}

func (reConfiguration *ReConfiguration) QueryRowContext(ctx context.Context, s string, i ...interface{}) *sql.Row {
	return reConfiguration.MySQL().QueryRowContext(ctx, s, i...)
}

func (reConfiguration *ReConfiguration) ExecContext(ctx context.Context, s string, i ...interface{}) (sql.Result, error) {
	return reConfiguration.MySQL().ExecContext(ctx, s, i...)
}

func (reConfiguration *ReConfiguration) SQL() *sql.DB {
	return reConfiguration.MySQL().SQL()
}

func (reConfiguration *ReConfiguration) ReConfiguration(configurator configurator.Configurator) error {
	isConfigChange := false

	config := Configuration(NewConfig(), configurator)

	if config.Port != reConfiguration.config.Port {
		reConfiguration.logger.Infof("data_base.mysql: old port '%d', new port '%d'", reConfiguration.config.Port, config.Port)

		reConfiguration.config.Port = config.Port
		isConfigChange = true
	}

	if config.Host != reConfiguration.config.Host {
		reConfiguration.logger.Infof("data_base.mysql: old host '%s', new host '%s'", reConfiguration.config.Host, config.Host)

		reConfiguration.config.Host = config.Host
		isConfigChange = true
	}

	if config.User != reConfiguration.config.User {
		reConfiguration.logger.Info("data_base.mysql: change user name")

		reConfiguration.config.User = config.User
		isConfigChange = true
	}

	if config.Password != reConfiguration.config.Password {
		reConfiguration.logger.Info("data_base.mysql: change password")

		reConfiguration.config.Password = config.Password
		isConfigChange = true
	}

	if config.DataBase != reConfiguration.config.DataBase {
		reConfiguration.logger.Infof("data_base.mysql: old data base name '%s', new data base name '%s'", reConfiguration.config.DataBase, config.DataBase)

		reConfiguration.config.DataBase = config.DataBase
		isConfigChange = true
	}

	if isConfigChange {
		reConfiguration.logger.Info("data_base.mysql: config change detected")

		mysql := NewReConnect(reConfiguration.config, reConfiguration.logger, reConfiguration.limiter)

		reConfiguration.rwMutex.Lock()
		defer reConfiguration.rwMutex.Unlock()

		if err := reConfiguration.mysql.Close(); err != nil {
			reConfiguration.logger.Error(err)
		}

		reConfiguration.mysql = mysql
	}

	return nil
}