update HotReloadStorage

This commit is contained in:
eatmoreapple 2022-11-14 11:05:54 +08:00
parent 80efb966db
commit a286c63146
2 changed files with 8 additions and 9 deletions

4
bot.go
View File

@ -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)

View File

@ -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 {