From a286c631462b9b023d65b25fd7d8c356135b7c18 Mon Sep 17 00:00:00 2001 From: eatmoreapple Date: Mon, 14 Nov 2022 11:05:54 +0800 Subject: [PATCH] update HotReloadStorage --- bot.go | 4 ++-- stroage.go | 13 ++++++------- 2 files changed, 8 insertions(+), 9 deletions(-) diff --git a/bot.go b/bot.go index 6d9f2ec..723470e 100644 --- a/bot.go +++ b/bot.go @@ -85,7 +85,7 @@ func (b *Bot) HotLogin(storage HotReloadStorage, retry ...bool) error { return b.Login() } - if err = b.hotLoginInit(*item); err != nil { + if err = b.hotLoginInit(item); err != nil { return err } @@ -98,7 +98,7 @@ func (b *Bot) HotLogin(storage HotReloadStorage, retry ...bool) error { } // 热登陆初始化 -func (b *Bot) hotLoginInit(item HotReloadStorageItem) error { +func (b *Bot) hotLoginInit(item *HotReloadStorageItem) error { cookies := item.Cookies for u, ck := range cookies { path, err := url.Parse(u) diff --git a/stroage.go b/stroage.go index 8f01ef9..e641988 100644 --- a/stroage.go +++ b/stroage.go @@ -28,7 +28,6 @@ type HotReloadStorage io.ReadWriter // JsonFileHotReloadStorage 实现HotReloadStorage接口 // 默认以json文件的形式存储 -// Deprecated type JsonFileHotReloadStorage struct { FileName string file *os.File @@ -45,21 +44,21 @@ func (j *JsonFileHotReloadStorage) Read(p []byte) (n int, err error) { } func (j *JsonFileHotReloadStorage) Write(p []byte) (n int, err error) { - j.file, err = os.Create(j.FileName) - if err != nil { - return 0, err + if j.file == nil { + j.file, err = os.Create(j.FileName) + if err != nil { + return 0, err + } } return j.file.Write(p) } // NewJsonFileHotReloadStorage 创建JsonFileHotReloadStorage -// Deprecated -// use os.File instead func NewJsonFileHotReloadStorage(filename string) HotReloadStorage { return &JsonFileHotReloadStorage{FileName: filename} } -var _ HotReloadStorage = &JsonFileHotReloadStorage{} +var _ HotReloadStorage = (*JsonFileHotReloadStorage)(nil) func NewHotReloadStorageItem(storage HotReloadStorage) (*HotReloadStorageItem, error) { if storage == nil {