修改HotReloadStorage定义

This commit is contained in:
eatmoreapple 2021-12-14 14:23:35 +08:00
parent 406b0f4be0
commit fd73ef6696
2 changed files with 2 additions and 18 deletions

11
bot.go
View File

@ -75,8 +75,6 @@ func (b *Bot) HotLogin(storage HotReloadStorage, retry ...bool) error {
return b.Login()
}
defer storage.Close()
if err = b.hotLoginInit(*item); err != nil {
return err
}
@ -340,14 +338,7 @@ func (b *Bot) DumpHotReloadStorage() error {
WechatDomain: b.Caller.Client.Domain,
}
data, err := json.Marshal(item)
if err != nil {
return err
}
if _, err = b.HotReloadStorage.Write(data); err != nil {
return err
}
return b.HotReloadStorage.Close()
return json.NewEncoder(b.HotReloadStorage).Encode(item)
}
// OnLogin is a setter for LoginCallBack

View File

@ -23,7 +23,7 @@ type HotReloadStorageItem struct {
}
// HotReloadStorage 热登陆存储接口
type HotReloadStorage io.ReadWriteCloser
type HotReloadStorage io.ReadWriter
// JsonFileHotReloadStorage 实现HotReloadStorage接口
// 默认以json文件的形式存储
@ -51,13 +51,6 @@ func (j *JsonFileHotReloadStorage) Write(p []byte) (n int, err error) {
return j.file.Write(p)
}
func (j *JsonFileHotReloadStorage) Close() error {
if j.file != nil {
return j.file.Close()
}
return nil
}
// NewJsonFileHotReloadStorage 创建JsonFileHotReloadStorage
// Deprecated
// use os.File instead