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

update HotReloadStorage

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
bot.go
View File

@ -85,7 +85,7 @@ func (b *Bot) HotLogin(storage HotReloadStorage, retry ...bool) error {
return b.Login() return b.Login()
} }
if err = b.hotLoginInit(*item); err != nil { if err = b.hotLoginInit(item); err != nil {
return err 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 cookies := item.Cookies
for u, ck := range cookies { for u, ck := range cookies {
path, err := url.Parse(u) path, err := url.Parse(u)

13
stroage.go
View File

@ -28,7 +28,6 @@ type HotReloadStorage io.ReadWriter
// JsonFileHotReloadStorage 实现HotReloadStorage接口 // JsonFileHotReloadStorage 实现HotReloadStorage接口
// 默认以json文件的形式存储 // 默认以json文件的形式存储
// Deprecated
type JsonFileHotReloadStorage struct { type JsonFileHotReloadStorage struct {
FileName string FileName string
file *os.File 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) { func (j *JsonFileHotReloadStorage) Write(p []byte) (n int, err error) {
j.file, err = os.Create(j.FileName) if j.file == nil {
if err != nil { j.file, err = os.Create(j.FileName)
return 0, err if err != nil {
return 0, err
}
} }
return j.file.Write(p) return j.file.Write(p)
} }
// NewJsonFileHotReloadStorage 创建JsonFileHotReloadStorage // NewJsonFileHotReloadStorage 创建JsonFileHotReloadStorage
// Deprecated
// use os.File instead
func NewJsonFileHotReloadStorage(filename string) HotReloadStorage { func NewJsonFileHotReloadStorage(filename string) HotReloadStorage {
return &JsonFileHotReloadStorage{FileName: filename} return &JsonFileHotReloadStorage{FileName: filename}
} }
var _ HotReloadStorage = &JsonFileHotReloadStorage{} var _ HotReloadStorage = (*JsonFileHotReloadStorage)(nil)
func NewHotReloadStorageItem(storage HotReloadStorage) (*HotReloadStorageItem, error) { func NewHotReloadStorageItem(storage HotReloadStorage) (*HotReloadStorageItem, error) {
if storage == nil { if storage == nil {