From dc3669dcad6ce4f8e7c9a12347e8483906347545 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=A4=9A=E5=90=83=E7=82=B9=E8=8B=B9=E6=9E=9C?= <73388495+eatmoreapple@users.noreply.github.com> Date: Thu, 12 Jan 2023 12:47:42 +0800 Subject: [PATCH] =?UTF-8?q?[refactor]=20pushLogin=20=E8=AE=BE=E7=BD=AE?= =?UTF-8?q?=E4=BA=8C=E7=BB=B4=E7=A0=81=E5=9B=9E=E8=B0=83=E5=92=8C=E6=89=AB?= =?UTF-8?q?=E7=A0=81=E5=9B=9E=E8=B0=83=E9=BB=98=E8=AE=A4=E4=B8=8D=E8=A7=A6?= =?UTF-8?q?=E5=8F=91=20(#197)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- bot.go | 19 +++++----- bot_login.go | 100 +++++++++------------------------------------------ 2 files changed, 24 insertions(+), 95 deletions(-) diff --git a/bot.go b/bot.go index e53a14f..2908d17 100644 --- a/bot.go +++ b/bot.go @@ -29,6 +29,7 @@ type Bot struct { hotReloadStorage HotReloadStorage uuid string deviceId string // 设备Id + loginOptionGroup BotOptionGroup } // Alive 判断当前用户是否正常在线 @@ -67,8 +68,8 @@ func (b *Bot) GetCurrentUser() (*Self, error) { } // login 这里对进行一些对登录前后的hook -func (b *Bot) login(login BotLogin, opts ...BotLoginOption) (err error) { - opt := BotOptionGroup(opts) +func (b *Bot) login(login BotLogin) (err error) { + opt := b.loginOptionGroup opt.Prepare(b) if err = login.Login(b); err != nil { err = opt.OnError(b, err) @@ -81,11 +82,7 @@ func (b *Bot) login(login BotLogin, opts ...BotLoginOption) (err error) { // Login 用户登录 func (b *Bot) Login() error { - scanLogin := &SacnLogin{ - UUIDCallback: b.UUIDCallback, - ScanCallBack: b.ScanCallBack, - LoginCallBack: b.LoginCallBack, - } + scanLogin := &SacnLogin{} return b.login(scanLogin) } @@ -94,8 +91,8 @@ func (b *Bot) HotLogin(storage HotReloadStorage, opts ...BotLoginOption) error { hotLogin := &HotLogin{storage: storage} // 进行相关设置。 // 如果相对默认的行为进行修改，在opts里面进行追加即可。 - opts = append(hotLoginDefaultOptions[:], opts...) - return b.login(hotLogin, opts...) + b.loginOptionGroup = append(hotLoginDefaultOptions[:], opts...) + return b.login(hotLogin) } // PushLogin 免扫码登录 @@ -106,8 +103,8 @@ func (b *Bot) PushLogin(storage HotReloadStorage, opts ...BotLoginOption) error pushLogin := &PushLogin{storage: storage} // 进行相关设置。 // 如果相对默认的行为进行修改，在opts里面进行追加即可。 - opts = append(pushLoginDefaultOptions[:], opts...) - return b.login(pushLogin, opts...) + b.loginOptionGroup = append(pushLoginDefaultOptions[:], opts...) + return b.login(pushLogin) } // Logout 用户退出 diff --git a/bot_login.go b/bot_login.go index 344c8e5..18151ba 100644 --- a/bot_login.go +++ b/bot_login.go @@ -64,24 +64,22 @@ var DoNothingBotLoginOption = &BaseBotLoginOption{} // RetryLoginOption 在登录失败后进行扫码登录 type RetryLoginOption struct { BaseBotLoginOption - SacnLogin -} - -// Prepare 实现了 BotLoginOption 接口 -func (r *RetryLoginOption) Prepare(bot *Bot) { - r.UUIDCallback = bot.UUIDCallback - r.LoginCallBack = bot.LoginCallBack - r.ScanCallBack = bot.ScanCallBack + MaxRetryCount int + currentRetryTime int } // OnError 实现了 BotLoginOption 接口 // 当登录失败后，会调用此方法进行扫码登录 -func (r *RetryLoginOption) OnError(bot *Bot, _ error) error { - return r.Login(bot) +func (r *RetryLoginOption) OnError(bot *Bot, err error) error { + if r.currentRetryTime >= r.MaxRetryCount { + return err + } + r.currentRetryTime++ + return bot.Login() } func NewRetryLoginOption() BotLoginOption { - return &RetryLoginOption{} + return &RetryLoginOption{MaxRetryCount: 1} } // SyncReloadDataLoginOption 在登录成功后进行数据定时同步到指定的storage中 @@ -105,38 +103,6 @@ func NewSyncReloadDataLoginOption(duration time.Duration) BotLoginOption { return &SyncReloadDataLoginOption{SyncLoopDuration: duration} } -// WithoutLoginCallbackOption 不使用登录回调 -type WithoutLoginCallbackOption struct{ BaseBotLoginOption } - -// Prepare 实现了 BotLoginOption 接口 -// 将设置的 LoginCallback 置为 nil -func (w WithoutLoginCallbackOption) Prepare(b *Bot) { b.LoginCallBack = nil } - -func NewWithoutLoginCallbackOption() BotLoginOption { - return &WithoutLoginCallbackOption{} -} - -// WithoutScanCallbackOption 不使用扫码回调 -type WithoutScanCallbackOption struct{ BaseBotLoginOption } - -// Prepare 实现了 BotLoginOption 接口 -func (w WithoutScanCallbackOption) Prepare(b *Bot) { b.ScanCallBack = nil } - -func NewWithoutScanCallbackOption() BotLoginOption { - return &WithoutScanCallbackOption{} -} - -// WithoutUUIDCallbackOption 不使用UUID回调 -type WithoutUUIDCallbackOption struct{ BaseBotLoginOption } - -// Prepare 实现了 BotLoginOption 接口 -// 将设置的 UUIDCallback 置为 nil -func (w WithoutUUIDCallbackOption) Prepare(bot *Bot) { bot.UUIDCallback = nil } - -func NewWithoutUUIDCallbackOption() BotLoginOption { - return &WithoutUUIDCallbackOption{} -} - // WithModeOption 指定使用哪种客户端模式 type WithModeOption struct { mode Mode @@ -168,11 +134,7 @@ type BotLogin interface { } // SacnLogin 扫码登录 -type SacnLogin struct { - UUIDCallback func(uuid string) - LoginCallBack func(body []byte) - ScanCallBack func(body []byte) -} +type SacnLogin struct{} // Login 实现了 BotLogin 接口 func (s *SacnLogin) Login(bot *Bot) error { @@ -189,9 +151,9 @@ func (s *SacnLogin) checkLogin(bot *Bot, uuid string) error { loginChecker := &LoginChecker{ Bot: bot, Tip: "0", - UUIDCallback: s.UUIDCallback, - LoginCallBack: s.LoginCallBack, - ScanCallBack: s.ScanCallBack, + UUIDCallback: bot.UUIDCallback, + LoginCallBack: bot.LoginCallBack, + ScanCallBack: bot.ScanCallBack, } return loginChecker.CheckLogin() } @@ -200,10 +162,7 @@ var ( hotLoginDefaultOptions = [...]BotLoginOption{ NewSyncReloadDataLoginOption(defaultHotStorageSyncDuration), } - pushLoginDefaultOptions = [...]BotLoginOption{ - NewSyncReloadDataLoginOption(defaultHotStorageSyncDuration), - NewWithoutScanCallbackOption(), - } + pushLoginDefaultOptions = hotLoginDefaultOptions ) // HotLogin 热登录模式 @@ -252,12 +211,12 @@ func (p *PushLogin) pushLoginInit(bot *Bot) error { // checkLogin 登录检查 func (p *PushLogin) checkLogin(bot *Bot, uuid string) error { bot.uuid = uuid + // 为什么把 UUIDCallback 和 ScanCallBack 置为nil呢? + // 因为这两个对用户是无感知的。 loginChecker := &LoginChecker{ Bot: bot, Tip: "1", - UUIDCallback: bot.UUIDCallback, LoginCallBack: bot.LoginCallBack, - ScanCallBack: bot.ScanCallBack, } return loginChecker.CheckLogin() } @@ -329,33 +288,6 @@ func HotLoginWithSyncReloadData(duration time.Duration) BotLoginOption { return NewSyncReloadDataLoginOption(duration) } -// Deprecated: 请使用 NewWithoutLoginCallbackOption 代替 -// PushLoginWithoutUUIDCallback 免扫码登录模式，不执行UUID回调 -func PushLoginWithoutUUIDCallback(flag bool) BotLoginOption { - if !flag { - return DoNothingBotLoginOption - } - return NewWithoutLoginCallbackOption() -} - -// Deprecated: 请使用 NewWithoutScanCallbackOption 代替 -// PushLoginWithoutScanCallback 免扫码登录模式，不执行扫码回调 -func PushLoginWithoutScanCallback(flag bool) BotLoginOption { - if !flag { - return DoNothingBotLoginOption - } - return NewWithoutScanCallbackOption() -} - -// Deprecated: 请使用 NewWithoutLoginCallbackOption 代替 -// PushLoginWithoutLoginCallback 免扫码登录模式，不执行登录回调 -func PushLoginWithoutLoginCallback(flag bool) BotLoginOption { - if !flag { - return DoNothingBotLoginOption - } - return NewWithoutLoginCallbackOption() -} - // Deprecated: 请使用 NewRetryLoginOption 代替 // PushLoginWithRetry 免扫码登录模式，如果登录失败会重试 func PushLoginWithRetry(flag bool) BotLoginOption {