github/openwechat
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

添加Bot存活控制

Browse Source
This commit is contained in:
eatmoreapple 2022-12-09 13:33:16 +08:00
parent 8ba44602ae
commit 8a201f1461
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
bot.go
View File

@ -386,11 +386,12 @@ func (b *Bot) OnLogout(f func(bot *Bot)) {
} }
// NewBot Bot的构造方法 // NewBot Bot的构造方法
func NewBot() *Bot { // 接收外部的 context.Context，用于控制Bot的存活
func NewBot(c context.Context) *Bot {
caller := DefaultCaller() caller := DefaultCaller()
// 默认行为为桌面模式 // 默认行为为桌面模式
caller.Client.SetMode(Normal) caller.Client.SetMode(Normal)
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(c)
return &Bot{Caller: caller, Storage: &Storage{}, context: ctx, cancel: cancel} return &Bot{Caller: caller, Storage: &Storage{}, context: ctx, cancel: cancel}
} }
@ -399,7 +400,7 @@ func NewBot() *Bot {
// //
// bot := openwechat.DefaultBot(openwechat.Desktop) // bot := openwechat.DefaultBot(openwechat.Desktop)
func DefaultBot(modes ...Mode) *Bot { func DefaultBot(modes ...Mode) *Bot {
bot := NewBot() bot := NewBot(context.Background())
if len(modes) > 0 { if len(modes) > 0 {
bot.Caller.Client.SetMode(modes[0]) bot.Caller.Client.SetMode(modes[0])
} }