From c7c3e42b61f1bfb39f2a178a558af501c93f58f7 Mon Sep 17 00:00:00 2001 From: injoyai <1113655791@qq.com> Date: Fri, 21 Feb 2025 15:13:51 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0manage,=E6=96=B9=E4=BE=BF?= =?UTF-8?q?=E5=BF=AB=E9=80=9F=E6=90=AD=E5=BB=BA=E9=A1=B9=E7=9B=AE,?= =?UTF-8?q?=E8=AF=95=E7=94=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- manage.go | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/manage.go b/manage.go index 62463d0..8b1fb59 100644 --- a/manage.go +++ b/manage.go @@ -12,8 +12,8 @@ func NewManage(cfg *ManageConfig, op ...client.Option) (*Manage, error) { if len(cfg.Hosts) == 0 { cfg.Hosts = Hosts } - if cfg.Database == "" { - cfg.Database = "./data/" + if cfg.Dir == "" { + cfg.Dir = "./data/" } //连接池 @@ -30,7 +30,7 @@ func NewManage(cfg *ManageConfig, op ...client.Option) (*Manage, error) { return nil, err } 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 { return nil, err } @@ -41,7 +41,7 @@ func NewManage(cfg *ManageConfig, op ...client.Option) (*Manage, error) { return nil, err } 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 { return nil, err } @@ -71,7 +71,7 @@ func (this *Manage) AddWorkdayTask(spec string, f func(m *Manage)) { } type ManageConfig struct { - Hosts []string //服务端IP - Number int //客户端数量 - Database string //数据位置 + Hosts []string //服务端IP + Number int //客户端数量 + Dir string //数据位置 }