添加获取热存储对象的方法
This commit is contained in:
parent
0d72f37d31
commit
3e14a0d5ed
15
bot.go
15
bot.go
@ -1,7 +1,6 @@
|
|||||||
package openwechat
|
package openwechat
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
|
||||||
"context"
|
"context"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"errors"
|
"errors"
|
||||||
@ -67,19 +66,15 @@ func (b *Bot) HotLogin(storage HotReloadStorage, retry ...bool) error {
|
|||||||
|
|
||||||
// 如果load出错了,就执行正常登陆逻辑
|
// 如果load出错了,就执行正常登陆逻辑
|
||||||
// 第一次没有数据load都会出错的
|
// 第一次没有数据load都会出错的
|
||||||
var buffer bytes.Buffer
|
item, err := NewHotReloadStorageItem(storage)
|
||||||
if _, err := buffer.ReadFrom(storage); err != nil {
|
|
||||||
|
if err != nil {
|
||||||
return b.Login()
|
return b.Login()
|
||||||
}
|
}
|
||||||
defer b.HotReloadStorage.Close()
|
|
||||||
|
|
||||||
var item HotReloadStorageItem
|
defer storage.Close()
|
||||||
|
|
||||||
if err = json.NewDecoder(&buffer).Decode(&item); err != nil {
|
if err = b.hotLoginInit(*item); err != nil {
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
if err = b.hotLoginInit(item); err != nil {
|
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
19
stroage.go
19
stroage.go
@ -1,6 +1,9 @@
|
|||||||
package openwechat
|
package openwechat
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
"io"
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
@ -60,3 +63,19 @@ func NewJsonFileHotReloadStorage(filename string) HotReloadStorage {
|
|||||||
}
|
}
|
||||||
|
|
||||||
var _ HotReloadStorage = &JsonFileHotReloadStorage{}
|
var _ HotReloadStorage = &JsonFileHotReloadStorage{}
|
||||||
|
|
||||||
|
func NewHotReloadStorageItem(storage HotReloadStorage) (*HotReloadStorageItem, error) {
|
||||||
|
if storage == nil {
|
||||||
|
return nil, errors.New("storage can't be nil")
|
||||||
|
}
|
||||||
|
var buffer bytes.Buffer
|
||||||
|
if _, err := buffer.ReadFrom(storage); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
var item HotReloadStorageItem
|
||||||
|
|
||||||
|
if err := json.NewDecoder(&buffer).Decode(&item); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &item, nil
|
||||||
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user