From d2504639da3c3328329498d5efb178179daf435c Mon Sep 17 00:00:00 2001 From: wood chen Date: Wed, 18 Sep 2024 15:49:26 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D(service/init.go):=20?= =?UTF-8?q?=E4=BC=98=E5=8C=96=E5=88=9D=E5=A7=8B=E5=8C=96=E6=B5=81=E7=A8=8B?= =?UTF-8?q?=EF=BC=8C=E5=A2=9E=E5=BC=BA=E6=9C=8D=E5=8A=A1=E7=A8=B3=E5=AE=9A?= =?UTF-8?q?=E6=80=A7=EF=BC=8C=E6=B7=BB=E5=8A=A0=E9=94=99=E8=AF=AF=E5=A4=84?= =?UTF-8?q?=E7=90=86=E6=9C=BA=E5=88=B6=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.go | 5 ++++- service/guard.go | 2 +- service/init.go | 10 +++++++++- 3 files changed, 14 insertions(+), 3 deletions(-) diff --git a/main.go b/main.go index 98790bb..cc69652 100644 --- a/main.go +++ b/main.go @@ -18,7 +18,10 @@ func main() { log.Fatalf("Invalid ADMIN_ID: %v", err) } - service.Init(botToken, adminID) + err = service.Init(botToken, adminID) + if err != nil { + log.Fatalf("Failed to initialize service: %v", err) + } go service.RunGuard() go service.RunBinance() diff --git a/service/guard.go b/service/guard.go index af50a68..223cc8f 100644 --- a/service/guard.go +++ b/service/guard.go @@ -91,7 +91,7 @@ func startBot() error { return fmt.Errorf("error registering commands: %w", err) } - linkFilter, err := NewLinkFilter(core.DB_FILE) + linkFilter, err := NewLinkFilter() if err != nil { return fmt.Errorf("failed to create LinkFilter: %v", err) } diff --git a/service/init.go b/service/init.go index b6eee12..4b646c4 100644 --- a/service/init.go +++ b/service/init.go @@ -6,10 +6,18 @@ import ( "github.com/woodchen-ink/Q58Bot/core" ) -func Init(botToken string, adminID int64) { +func Init(botToken string, adminID int64) error { core.InitGlobalVariables(botToken, adminID) + // 初始化提示词服务 + err := InitPromptService() + if err != nil { + return err + } + // 设置时区 loc := time.FixedZone("Asia/Singapore", 8*60*60) time.Local = loc + + return nil }