hoteas
b43f968b6c
- 在应用程序中新增对达梦数据库（DM）的配置和连接支持 - 实现 SetDmDB 函数以配置达梦数据库连接 - 更新数据库操作逻辑，支持达梦特有的 SQL 语法和功能 - 在相关文件中添加达梦数据库的处理逻辑，包括表创建、数据插入和查询 - 更新 go.mod 和 go.sum 文件以引入达梦数据库驱动 - 增强文档，详细说明达梦数据库的配置和使用方法
479 lines
14 KiB
Go
479 lines
14 KiB
Go
/*
|
|
* Copyright (c) 2000-2018, 达梦数据库有限公司.
|
|
* All rights reserved.
|
|
*/
|
|
|
|
package dm
|
|
|
|
import (
|
|
"bufio"
|
|
"io"
|
|
"os"
|
|
"runtime"
|
|
"strconv"
|
|
"strings"
|
|
"sync"
|
|
|
|
"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淘汰方式
|
|
|
|
ClobToBytesDef = false //Clob是否转换为bytes
|
|
)
|
|
|
|
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 sync.Map
|
|
|
|
GlobalProperties = NewProperties()
|
|
)
|
|
|
|
// filePath: dm_svc.conf 文件路径
|
|
func load(filePath string) {
|
|
if filePath == "" {
|
|
// 如果设置了环境变量则环境变量优先
|
|
filePath = os.Getenv("DM_SVC_PATH")
|
|
if filePath == "" {
|
|
// 否则使用默认
|
|
switch runtime.GOOS {
|
|
case "windows":
|
|
filePath = os.Getenv("SystemRoot") + "\\system32\\dm_svc.conf"
|
|
case "linux":
|
|
filePath = "/etc/dm_svc.conf"
|
|
default:
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
// 如果设置了连接串上svcConfPath，则串上优先
|
|
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.Load(groupName)
|
|
if groupName == "" || !ok {
|
|
continue
|
|
}
|
|
groupProps = dbGroup.(*epGroup).props
|
|
if groupProps.IsNil() {
|
|
groupProps = NewProperties()
|
|
groupProps.SetProperties(GlobalProperties)
|
|
dbGroup.(*epGroup).props = groupProps
|
|
}
|
|
|
|
} else {
|
|
cfgInfo := [2]string{}
|
|
if index := strings.Index(line, "="); index > 0 {
|
|
cfgInfo[0] = line[0:index]
|
|
cfgInfo[1] = line[index+1:]
|
|
} else {
|
|
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.Store(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 == "CATALOG" {
|
|
props.Set(CatalogKey, 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 if key == "SERVER_OPTION" {
|
|
props.Set(ServerOptionKey, value)
|
|
} else if key == "CLOB_TO_BYTES" {
|
|
props.Set(ClobToBytesKey, 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]
|
|
tmpName = v[begin : end+1]
|
|
|
|
// port
|
|
if portIndex := strings.IndexByte(v[end:], ':'); portIndex != -1 {
|
|
tmpPort, _ = strconv.Atoi(strings.TrimSpace(v[end+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
|
|
}
|