From 50c997f26fed0d39e46ddaf8b11b5c434fe7ec4f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=92=B1=E7=BA=AF=E5=87=80?= <1113655791@qq.com>
Date: Mon, 3 Mar 2025 19:35:56 +0800
Subject: [PATCH] =?UTF-8?q?=E7=BB=86=E8=8A=82=E4=BC=98=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 manage.go | 20 ++++++++++++--------
 1 file changed, 12 insertions(+), 8 deletions(-)

diff --git a/manage.go b/manage.go
index 6b42c32..f1dba7b 100644
--- a/manage.go
+++ b/manage.go
@@ -15,8 +15,11 @@ func NewManage(cfg *ManageConfig, op ...client.Option) (*Manage, error) {
 	if len(cfg.Hosts) == 0 {
 		cfg.Hosts = Hosts
 	}
-	if cfg.Dir == "" {
-		cfg.Dir = "./data/"
+	if cfg.CodesDir == "" {
+		cfg.CodesDir = "./data/database"
+	}
+	if cfg.WorkdayDir == "" {
+		cfg.WorkdayDir = "./data/database"
 	}
 
 	//连接池
@@ -33,18 +36,18 @@ 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.Dir, "database/codes.db"))
+	codes, err := NewCodes(codesClient, filepath.Join(cfg.CodesDir, "codes.db"))
 	if err != nil {
 		return nil, err
 	}
 
-	///工作日
+	//工作日
 	workdayClient, err := DialHosts(cfg.Hosts, op...)
 	if err != nil {
 		return nil, err
 	}
 	workdayClient.Wait.SetTimeout(time.Second * 5)
-	workday, err := NewWorkday(workdayClient, filepath.Join(cfg.Dir, "database/codes.db"))
+	workday, err := NewWorkday(workdayClient, filepath.Join(cfg.WorkdayDir, "workday.db"))
 	if err != nil {
 		return nil, err
 	}
@@ -76,7 +79,8 @@ func (this *Manage) AddWorkdayTask(spec string, f func(m *Manage)) {
 }
 
 type ManageConfig struct {
-	Hosts  []string //服务端IP
-	Number int      //客户端数量
-	Dir    string   //数据位置
+	Hosts      []string //服务端IP
+	Number     int      //客户端数量
+	CodesDir   string   //代码数据库位置
+	WorkdayDir string   //工作日数据库位置
 }