hoteas
7f7b585ffb
- 在应用程序中新增对达梦数据库（DM）的配置和连接支持 - 实现 SetDmDB 函数以配置达梦数据库连接 - 更新数据库操作逻辑，支持达梦特有的 SQL 语法和功能 - 在相关文件中添加达梦数据库的处理逻辑，包括表创建、数据插入和查询 - 更新 go.mod 和 go.sum 文件以引入达梦数据库驱动 - 增强文档，详细说明达梦数据库的配置和使用方法
66 lines
1.6 KiB
Go
66 lines
1.6 KiB
Go
/*
|
|
* Copyright (c) 2000-2018, 达梦数据库有限公司.
|
|
* All rights reserved.
|
|
*/
|
|
|
|
package dm
|
|
|
|
import "sort"
|
|
|
|
const (
|
|
TYPE_WELL_DISTRIBUTE = 0
|
|
TYPE_HEAD_FIRST = 1
|
|
)
|
|
|
|
type epSelector struct {
|
|
dbs []*ep
|
|
}
|
|
|
|
func newEPSelector(dbs []*ep) *epSelector {
|
|
return &epSelector{dbs}
|
|
}
|
|
|
|
func (s *epSelector) sortDBList(first bool) []*ep {
|
|
if !first {
|
|
// 按sort从大到小排序，相同sort值顺序不变
|
|
sort.Slice(s.dbs, func(i, j int) bool {
|
|
return s.dbs[i].getSort(first) > s.dbs[j].getSort(first)
|
|
})
|
|
}
|
|
return s.dbs
|
|
}
|
|
|
|
func (s *epSelector) checkServerMode(conn *DmConnection, last bool) (bool, error) {
|
|
// 只连dsc control节点
|
|
if conn.dmConnector.loginDscCtrl && !conn.dscControl {
|
|
conn.close()
|
|
return false, ECGO_INVALID_SERVER_MODE.throw()
|
|
}
|
|
// 模式不匹配, 这里使用的是连接之前的sort，连接之后server的状态可能发生改变sort也可能改变
|
|
if conn.dmConnector.loginStatus > 0 && int(conn.SvrStat) != conn.dmConnector.loginStatus {
|
|
conn.close()
|
|
return false, ECGO_INVALID_SERVER_MODE.throw()
|
|
}
|
|
if last {
|
|
switch conn.dmConnector.loginMode {
|
|
case LOGIN_MODE_PRIMARY_ONLY:
|
|
return conn.SvrMode == SERVER_MODE_PRIMARY, nil
|
|
case LOGIN_MODE_STANDBY_ONLY:
|
|
return conn.SvrMode == SERVER_MODE_STANDBY, nil
|
|
default:
|
|
return true, nil
|
|
}
|
|
}
|
|
switch conn.dmConnector.loginMode {
|
|
case LOGIN_MODE_NORMAL_FIRST:
|
|
return conn.SvrMode == SERVER_MODE_NORMAL, nil
|
|
case LOGIN_MODE_PRIMARY_FIRST, LOGIN_MODE_PRIMARY_ONLY:
|
|
return conn.SvrMode == SERVER_MODE_PRIMARY, nil
|
|
case LOGIN_MODE_STANDBY_FIRST, LOGIN_MODE_STANDBY_ONLY:
|
|
return conn.SvrMode == SERVER_MODE_STANDBY, nil
|
|
default:
|
|
break
|
|
}
|
|
return false, nil
|
|
}
|