Cloudpods/backend/vendor/gitee.com/chunanyong/dm/zzm.go

/*
 * Copyright (c) 2000-2018, 达梦数据库有限公司.
 * All rights reserved.
 */

package dm

import (
	"bufio"
	"io"
	"os"
	"runtime"
	"strconv"
	"strings"

	"gitee.com/chunanyong/dm/util"
)

var LogDirDef, _ = os.Getwd()

var StatDirDef, _ = os.Getwd()

const (
	DEFAULT_PORT int32 = 5236

	//log level
	LOG_OFF int = 0

	LOG_ERROR int = 1

	LOG_WARN int = 2

	LOG_SQL int = 3

	LOG_INFO int = 4

	LOG_DEBUG int = 5

	LOG_ALL int = 9

	//stat
	STAT_SQL_REMOVE_LATEST int = 0

	STAT_SQL_REMOVE_OLDEST int = 1

	// 编码字符集
	ENCODING_UTF8 string = "UTF-8"

	ENCODING_EUCKR string = "EUC-KR"

	ENCODING_GB18030 string = "GB18030"

	ENCODING_BIG5 string = "BIG5"

	DbAliveCheckFreqDef = 0

	LocaleDef = LANGUAGE_CN

	// log
	LogLevelDef = LOG_OFF // 日志级别：off, error, warn, sql, info, all

	LogFlushFreqDef = 10 // 日志刷盘时间s (>=0)

	LogFlushQueueSizeDef = 100 //日志队列大小

	LogBufferSizeDef = 32 * 1024 // 日志缓冲区大小 (>0)

	// stat
	StatEnableDef = false //

	StatFlushFreqDef = 3 // 日志刷盘时间s (>=0)

	StatSlowSqlCountDef = 100 // 慢sql top行数，(0-1000)

	StatHighFreqSqlCountDef = 100 // 高频sql top行数， (0-1000)

	StatSqlMaxCountDef = 100000 // sql 统计最大值(0-100000)

	StatSqlRemoveModeDef = STAT_SQL_REMOVE_LATEST // 记录sql数超过最大值时，sql淘汰方式
)

var (
	DbAliveCheckFreq = DbAliveCheckFreqDef

	Locale = LocaleDef // 0:简体中文 1：英文 2:繁体中文

	// log
	LogLevel = LogLevelDef // 日志级别：off, error, warn, sql, info, all

	LogDir = LogDirDef

	LogFlushFreq = LogFlushFreqDef // 日志刷盘时间s (>=0)

	LogFlushQueueSize = LogFlushQueueSizeDef

	LogBufferSize = LogBufferSizeDef // 日志缓冲区大小 (>0)

	// stat
	StatEnable = StatEnableDef //

	StatDir = StatDirDef // jdbc工作目录,所有生成的文件都在该目录下

	StatFlushFreq = StatFlushFreqDef // 日志刷盘时间s (>=0)

	StatSlowSqlCount = StatSlowSqlCountDef // 慢sql top行数，(0-1000)

	StatHighFreqSqlCount = StatHighFreqSqlCountDef // 高频sql top行数， (0-1000)

	StatSqlMaxCount = StatSqlMaxCountDef // sql 统计最大值(0-100000)

	StatSqlRemoveMode = StatSqlRemoveModeDef // 记录sql数超过最大值时，sql淘汰方式

	/*---------------------------------------------------------------*/
	ServerGroupMap = make(map[string]*epGroup)

	GlobalProperties = NewProperties()
)

// filePath: dm_svc.conf 文件路径
func load(filePath string) {
	if filePath == "" {
		switch runtime.GOOS {
		case "windows":
			filePath = os.Getenv("SystemRoot") + "\\system32\\dm_svc.conf"
		case "linux":
			filePath = "/etc/dm_svc.conf"
		default:
			return
		}
	}
	file, err := os.Open(filePath)
	defer file.Close()
	if err != nil {
		return
	}
	fileReader := bufio.NewReader(file)

	// GlobalProperties = NewProperties()
	var groupProps *Properties
	var line string //dm_svc.conf读取到的一行

	for line, err = fileReader.ReadString('\n'); line != "" && (err == nil || err == io.EOF); line, err = fileReader.ReadString('\n') {
		// 去除#标记的注释
		if notesIndex := strings.IndexByte(line, '#'); notesIndex != -1 {
			line = line[:notesIndex]
		}
		// 去除前后多余的空格
		line = strings.TrimSpace(line)
		if line == "" {
			continue
		}

		if strings.HasPrefix(line, "[") && strings.HasSuffix(line, "]") {
			groupName := strings.ToLower(line[1 : len(line)-1])
			dbGroup, ok := ServerGroupMap[groupName]
			if groupName == "" || !ok {
				continue
			}
			groupProps = dbGroup.props
			if groupProps.IsNil() {
				groupProps = NewProperties()
				groupProps.SetProperties(GlobalProperties)
				dbGroup.props = groupProps
			}

		} else {
			cfgInfo := strings.Split(line, "=")
			if len(cfgInfo) < 2 {
				continue
			}
			key := strings.TrimSpace(cfgInfo[0])
			value := strings.TrimSpace(cfgInfo[1])
			if strings.HasPrefix(value, "(") && strings.HasSuffix(value, ")") {
				value = strings.TrimSpace(value[1 : len(value)-1])
			}
			if key == "" || value == "" {
				continue
			}
			// 区分属性是全局的还是组的
			var success bool
			if groupProps.IsNil() {
				success = SetServerGroupProperties(GlobalProperties, key, value)
			} else {
				success = SetServerGroupProperties(groupProps, key, value)
			}
			if !success {
				var serverGroup = parseServerName(key, value)
				if serverGroup != nil {
					serverGroup.props = NewProperties()
					serverGroup.props.SetProperties(GlobalProperties)
					ServerGroupMap[strings.ToLower(key)] = serverGroup
				}
			}
		}
	}
}

func SetServerGroupProperties(props *Properties, key string, value string) bool {
	key = strings.ToUpper(key)
	if key == "ADDRESS_REMAP" {
		tmp := props.GetString(AddressRemapKey, "")
		props.Set(AddressRemapKey, tmp+"("+value+")")
	} else if key == "ALWAYS_ALLOW_COMMIT" {
		props.Set(AlwayseAllowCommitKey, value)
	} else if key == "APP_NAME" {
		props.Set(AppNameKey, value)
	} else if key == "AUTO_COMMIT" {
		props.Set(AutoCommitKey, value)
	} else if key == "BATCH_ALLOW_MAX_ERRORS" {
		props.Set(BatchAllowMaxErrorsKey, value)
	} else if key == "BATCH_CONTINUE_ON_ERROR" ||
		key == "CONTINUE_BATCH_ON_ERROR" {
		props.Set(ContinueBatchOnErrorKey, value)
	} else if key == "BATCH_NOT_ON_CALL" {
		props.Set(BatchNotOnCallKey, value)
	} else if key == "BATCH_TYPE" {
		props.Set(BatchTypeKey, value)
	} else if key == "BUF_PREFETCH" {
		props.Set(BufPrefetchKey, value)
	} else if key == "CIPHER_PATH" {
		props.Set(CipherPathKey, value)
	} else if key == "CLUSTER" {
		props.Set(ClusterKey, value)
	} else if key == "COLUMN_NAME_UPPER_CASE" {
		props.Set(ColumnNameUpperCaseKey, value)
	} else if key == "COLUMN_NAME_CASE" {
		props.Set(ColumnNameCaseKey, value)
	} else if key == "COMPATIBLE_MODE" {
		props.Set(CompatibleModeKey, value)
	} else if key == "COMPRESS" ||
		key == "COMPRESS_MSG" {
		props.Set(CompressKey, value)
	} else if key == "COMPRESS_ID" {
		props.Set(CompressIdKey, value)
	} else if key == "CONNECT_TIMEOUT" {
		props.Set(ConnectTimeoutKey, value)
	} else if key == "DO_SWITCH" ||
		key == "AUTO_RECONNECT" {
		props.Set(DoSwitchKey, value)
	} else if key == "ENABLE_RS_CACHE" {
		props.Set(EnRsCacheKey, value)
	} else if key == "EP_SELECTION" {
		props.Set(EpSelectorKey, value)
	} else if key == "ESCAPE_PROCESS" {
		props.Set(EscapeProcessKey, value)
	} else if key == "IS_BDTA_RS" {
		props.Set(IsBdtaRSKey, value)
	} else if key == "KEY_WORDS" ||
		key == "KEYWORDS" {
		props.Set(KeywordsKey, value)
	} else if key == "LANGUAGE" {
		props.Set(LanguageKey, value)
	} else if key == "LOB_MODE" {
		props.Set(LobModeKey, value)
	} else if key == "LOG_BUFFER_SIZE" {
		props.Set(LogBufferSizeKey, value)
	} else if key == "LOG_DIR" {
		props.Set(LogDirKey, value)
	} else if key == "LOG_FLUSH_FREQ" {
		props.Set(LogFlushFreqKey, value)
	} else if key == "LOG_FLUSHER_QUEUESIZE" {
		props.Set(LogFlusherQueueSizeKey, value)
	} else if key == "LOG_LEVEL" {
		props.Set(LogLevelKey, value)
	} else if key == "LOGIN_DSC_CTRL" {
		props.Set(LoginDscCtrlKey, value)
	} else if key == "LOGIN_ENCRYPT" {
		props.Set(LoginEncryptKey, value)
	} else if key == "LOGIN_MODE" {
		props.Set(LoginModeKey, value)
	} else if key == "LOGIN_STATUS" {
		props.Set(LoginStatusKey, value)
	} else if key == "MAX_ROWS" {
		props.Set(MaxRowsKey, value)
	} else if key == "MPP_LOCAL" {
		props.Set(MppLocalKey, value)
	} else if key == "OS_NAME" {
		props.Set(OsNameKey, value)
	} else if key == "RS_CACHE_SIZE" {
		props.Set(RsCacheSizeKey, value)
	} else if key == "RS_REFRESH_FREQ" {
		props.Set(RsRefreshFreqKey, value)
	} else if key == "RW_HA" {
		props.Set(RwHAKey, value)
	} else if key == "RW_IGNORE_SQL" {
		props.Set(RwIgnoreSqlKey, value)
	} else if key == "RW_PERCENT" {
		props.Set(RwPercentKey, value)
	} else if key == "RW_SEPARATE" {
		props.Set(RwSeparateKey, value)
	} else if key == "RW_STANDBY_RECOVER_TIME" {
		props.Set(RwStandbyRecoverTimeKey, value)
	} else if key == "SCHEMA" {
		props.Set(SchemaKey, value)
	} else if key == "SESS_ENCODE" {
		if IsSupportedCharset(value) {
			props.Set("sessEncode", value)
		}
	} else if key == "SESSION_TIMEOUT" {
		props.Set(SessionTimeoutKey, value)
	} else if key == "SOCKET_TIMEOUT" {
		props.Set(SocketTimeoutKey, value)
	} else if key == "SSL_CERT_PATH" {
		props.Set(SslCertPathKey, value)
	} else if key == "SSL_FILES_PATH" {
		props.Set(SslFilesPathKey, value)
	} else if key == "SSL_KEY_PATH" {
		props.Set(SslKeyPathKey, value)
	} else if key == "STAT_DIR" {
		props.Set(StatDirKey, value)
	} else if key == "STAT_ENABLE" {
		props.Set(StatEnableKey, value)
	} else if key == "STAT_FLUSH_FREQ" {
		props.Set(StatFlushFreqKey, value)
	} else if key == "STAT_HIGH_FREQ_SQL_COUNT" {
		props.Set(StatHighFreqSqlCountKey, value)
	} else if key == "STAT_SLOW_SQL_COUNT" {
		props.Set(StatSlowSqlCountKey, value)
	} else if key == "STAT_SQL_MAX_COUNT" {
		props.Set(StatSqlMaxCountKey, value)
	} else if key == "STAT_SQL_REMOVE_MODE" {
		props.Set(StatSqlRemoveModeKey, value)
	} else if key == "SWITCH_INTERVAL" {
		props.Set(SwitchIntervalKey, value)
	} else if key == "SWITCH_TIME" ||
		key == "SWITCH_TIMES" {
		props.Set(SwitchTimesKey, value)
	} else if key == "TIME_ZONE" {
		props.Set(TimeZoneKey, value)
		props.Set("localTimezone", value)
	} else if key == "USER_REMAP" {
		tmp := props.GetString(UserRemapKey, "")
		props.Set(UserRemapKey, tmp+"("+value+")")
	} else {
		return false
	}
	return true
}

func parseServerName(name string, value string) *epGroup {
	values := strings.Split(value, ",")

	var tmpVals []string
	var tmpName string
	var tmpPort int
	var svrList = make([]*ep, 0, len(values))

	for _, v := range values {

		var tmp *ep
		// 先查找IPV6,以[]包括
		begin := strings.IndexByte(v, '[')
		end := -1
		if begin != -1 {
			end = strings.IndexByte(v[begin:], ']')
		}
		if end != -1 {
			tmpName = v[begin+1 : end]

			// port
			if portIndex := strings.IndexByte(v[end:], ':'); portIndex != -1 {
				tmpPort, _ = strconv.Atoi(strings.TrimSpace(v[portIndex+1:]))
			} else {
				tmpPort = int(DEFAULT_PORT)
			}
			tmp = newEP(tmpName, int32(tmpPort))
			svrList = append(svrList, tmp)
			continue
		}
		// IPV4
		tmpVals = strings.Split(v, ":")
		tmpName = strings.TrimSpace(tmpVals[0])
		if len(tmpVals) >= 2 {
			tmpPort, _ = strconv.Atoi(tmpVals[1])
		} else {
			tmpPort = int(DEFAULT_PORT)
		}
		tmp = newEP(tmpName, int32(tmpPort))
		svrList = append(svrList, tmp)
	}

	if len(svrList) == 0 {
		return nil
	}
	return newEPGroup(name, svrList)
}

func setDriverAttributes(props *Properties) {
	if props == nil || props.Len() == 0 {
		return
	}

	parseLanguage(props.GetString(LanguageKey, "cn"))
	DbAliveCheckFreq = props.GetInt(DbAliveCheckFreqKey, DbAliveCheckFreqDef, 1, int(INT32_MAX))

	//// log
	//LogLevel = ParseLogLevel(props)
	//LogDir = util.StringUtil.FormatDir(props.GetTrimString(LogDirKey, LogDirDef))
	//LogBufferSize = props.GetInt(LogBufferSizeKey, LogBufferSizeDef, 1, int(INT32_MAX))
	//LogFlushFreq = props.GetInt(LogFlushFreqKey, LogFlushFreqDef, 1, int(INT32_MAX))
	//LogFlushQueueSize = props.GetInt(LogFlusherQueueSizeKey, LogFlushQueueSizeDef, 1, int(INT32_MAX))
	//
	//// stat
	//StatEnable = props.GetBool(StatEnableKey, StatEnableDef)
	//StatDir = util.StringUtil.FormatDir(props.GetTrimString(StatDirKey, StatDirDef))
	//StatFlushFreq = props.GetInt(StatFlushFreqKey, StatFlushFreqDef, 1, int(INT32_MAX))
	//StatHighFreqSqlCount = props.GetInt(StatHighFreqSqlCountKey, StatHighFreqSqlCountDef, 0, 1000)
	//StatSlowSqlCount = props.GetInt(StatSlowSqlCountKey, StatSlowSqlCountDef, 0, 1000)
	//StatSqlMaxCount = props.GetInt(StatSqlMaxCountKey, StatSqlMaxCountDef, 0, 100000)
	//parseStatSqlRemoveMode(props)
}

func parseLanguage(value string) {
	if util.StringUtil.EqualsIgnoreCase("cn", value) {
		Locale = 0
	} else if util.StringUtil.EqualsIgnoreCase("en", value) {
		Locale = 1
	} else if util.StringUtil.EqualsIgnoreCase("cnt_hk", value) ||
		util.StringUtil.EqualsIgnoreCase("hk", value) ||
		util.StringUtil.EqualsIgnoreCase("tw", value) {
		Locale = 2
	}
}

func IsSupportedCharset(charset string) bool {
	if util.StringUtil.EqualsIgnoreCase(ENCODING_UTF8, charset) ||
		util.StringUtil.EqualsIgnoreCase(ENCODING_GB18030, charset) ||
		util.StringUtil.EqualsIgnoreCase(ENCODING_EUCKR, charset) ||
		util.StringUtil.EqualsIgnoreCase(ENCODING_BIG5, charset) {
		return true
	}
	return false
}

func ParseLogLevel(props *Properties) int {
	logLevel := LOG_OFF
	value := props.GetString(LogLevelKey, "")
	if value != "" && !util.StringUtil.IsDigit(value) {
		if util.StringUtil.EqualsIgnoreCase("debug", value) {
			logLevel = LOG_DEBUG
		} else if util.StringUtil.EqualsIgnoreCase("info", value) {
			logLevel = LOG_INFO
		} else if util.StringUtil.EqualsIgnoreCase("sql", value) {
			logLevel = LOG_SQL
		} else if util.StringUtil.EqualsIgnoreCase("warn", value) {
			logLevel = LOG_WARN
		} else if util.StringUtil.EqualsIgnoreCase("error", value) {
			logLevel = LOG_ERROR
		} else if util.StringUtil.EqualsIgnoreCase("off", value) {
			logLevel = LOG_OFF
		} else if util.StringUtil.EqualsIgnoreCase("all", value) {
			logLevel = LOG_ALL
		}
	} else {
		logLevel = props.GetInt(LogLevelKey, logLevel, LOG_OFF, LOG_INFO)
	}

	return logLevel
}