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

增加manage,方便快速搭建项目,试用

Browse Source
This commit is contained in:
injoyai
2025-02-21 15:13:51 +08:00
parent 2f46a3fb4b
commit c7c3e42b61
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
manage.go
View File

@@ -12,8 +12,8 @@ func NewManage(cfg *ManageConfig, op ...client.Option) (*Manage, error) {
if len(cfg.Hosts) == 0 { if len(cfg.Hosts) == 0 {
cfg.Hosts = Hosts cfg.Hosts = Hosts
} }
if cfg.Database == "" { if cfg.Dir == "" {
cfg.Database = "./data/" cfg.Dir = "./data/"
} }
//连接池 //连接池
@@ -30,7 +30,7 @@ func NewManage(cfg *ManageConfig, op ...client.Option) (*Manage, error) {
return nil, err return nil, err
} }
codesClient.Wait.SetTimeout(time.Second * 5) codesClient.Wait.SetTimeout(time.Second * 5)
codes, err := NewCodes(codesClient, filepath.Join(cfg.Database, "database/codes.db")) codes, err := NewCodes(codesClient, filepath.Join(cfg.Dir, "database/codes.db"))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@@ -41,7 +41,7 @@ func NewManage(cfg *ManageConfig, op ...client.Option) (*Manage, error) {
return nil, err return nil, err
} }
workdayClient.Wait.SetTimeout(time.Second * 5) workdayClient.Wait.SetTimeout(time.Second * 5)
workday, err := NewWorkday(workdayClient, filepath.Join(cfg.Database, "database/codes.db")) workday, err := NewWorkday(workdayClient, filepath.Join(cfg.Dir, "database/codes.db"))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@@ -71,7 +71,7 @@ func (this *Manage) AddWorkdayTask(spec string, f func(m *Manage)) {
} }
type ManageConfig struct { type ManageConfig struct {
Hosts []string //服务端IP Hosts []string //服务端IP
Number int //客户端数量 Number int //客户端数量
Database string //数据位置 Dir string //数据位置
} }