github/injoyai-tdx
1
0
Fork 0
mirror of https://github.com/injoyai/tdx.git synced 2025-11-26 21:25:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

可以自定义连接方式

Browse Source
This commit is contained in:
injoyai
2025-03-24 16:33:52 +08:00
parent ff7fc6aba0
commit e76043dc29
1 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
manage.go
View File

@@ -16,19 +16,19 @@ func NewManage(cfg *ManageConfig, op ...client.Option) (*Manage, error) {
if cfg == nil {
cfg = &ManageConfig{}
}
if len(cfg.Hosts) == 0 {
cfg.Hosts = Hosts
}
if cfg.CodesDir == "" {
cfg.CodesDir = DefaultDatabaseDir
}
if cfg.WorkdayDir == "" {
cfg.WorkdayDir = DefaultDatabaseDir
}
if cfg.Dial == nil {
cfg.Dial = DialDefault
}
//代码
DefaultCodes = &Codes{}
codesClient, err := DialHostsRange(cfg.Hosts, op...)
codesClient, err := cfg.Dial(op...)
if err != nil {
return nil, err
}
@@ -41,14 +41,14 @@ func NewManage(cfg *ManageConfig, op ...client.Option) (*Manage, error) {
//连接池
p, err := NewPool(func() (*Client, error) {
return DialHostsRange(cfg.Hosts, op...)
return cfg.Dial(op...)
}, cfg.Number)
if err != nil {
return nil, err
}
//工作日
workdayClient, err := DialHostsRange(cfg.Hosts, op...)
workdayClient, err := cfg.Dial(op...)
if err != nil {
return nil, err
}
@@ -85,8 +85,8 @@ func (this *Manage) AddWorkdayTask(spec string, f func(m *Manage)) {
}
type ManageConfig struct {
Hosts []string //服务端IP
Number int //客户端数量
CodesDir string //代码数据库位置
WorkdayDir string //工作日数据库位置
Number int //客户端数量
CodesDir string //代码数据库位置
WorkdayDir string //工作日数据库位置
Dial func(op ...client.Option) (cli *Client, err error) //默认连接方式
}