106 lines
2.4 KiB
Go
106 lines
2.4 KiB
Go
package openwechat
|
|
|
|
import (
|
|
"io"
|
|
"os"
|
|
"time"
|
|
)
|
|
|
|
// Storage 身份信息, 维持整个登陆的Session会话
|
|
type Storage struct {
|
|
LoginInfo *LoginInfo
|
|
Request *BaseRequest
|
|
Response *WebInitResponse
|
|
}
|
|
|
|
type HotReloadStorageItem struct {
|
|
Jar *Jar
|
|
BaseRequest *BaseRequest
|
|
LoginInfo *LoginInfo
|
|
WechatDomain WechatDomain
|
|
UUID string
|
|
}
|
|
|
|
// HotReloadStorage 热登陆存储接口
|
|
type HotReloadStorage io.ReadWriter
|
|
|
|
// jsonFileHotReloadStorage 实现HotReloadStorage接口
|
|
// 默认以json文件的形式存储
|
|
type jsonFileHotReloadStorage struct {
|
|
filename string
|
|
file *os.File
|
|
}
|
|
|
|
func (j *jsonFileHotReloadStorage) Read(p []byte) (n int, err error) {
|
|
if j.file == nil {
|
|
j.file, err = os.OpenFile(j.filename, os.O_RDWR, 0600)
|
|
if os.IsNotExist(err) {
|
|
return 0, ErrInvalidStorage
|
|
}
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
}
|
|
return j.file.Read(p)
|
|
}
|
|
|
|
func (j *jsonFileHotReloadStorage) Write(p []byte) (n int, err error) {
|
|
if j.file == nil {
|
|
j.file, err = os.Create(j.filename)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
}
|
|
// 为什么这里要对文件进行Truncate操作呢?
|
|
// 这是为了方便每次Dump的时候对文件进行重新写入, 而不是追加
|
|
// json序列化写入只会调用一次Write方法, 所以不要把这个方法当成io.Writer的Write方法
|
|
if _, err = j.file.Seek(0, io.SeekStart); err != nil {
|
|
return
|
|
}
|
|
if err = j.file.Truncate(0); err != nil {
|
|
return
|
|
}
|
|
return j.file.Write(p)
|
|
}
|
|
|
|
func (j *jsonFileHotReloadStorage) Close() error {
|
|
if j.file == nil {
|
|
return nil
|
|
}
|
|
return j.file.Close()
|
|
}
|
|
|
|
// NewJsonFileHotReloadStorage 创建JsonFileHotReloadStorage
|
|
func NewJsonFileHotReloadStorage(filename string) io.ReadWriteCloser {
|
|
return &jsonFileHotReloadStorage{filename: filename}
|
|
}
|
|
|
|
var _ HotReloadStorage = (*jsonFileHotReloadStorage)(nil)
|
|
|
|
type HotReloadStorageSyncer struct {
|
|
duration time.Duration
|
|
bot *Bot
|
|
}
|
|
|
|
// Sync 定时同步数据到登陆存储中
|
|
func (h *HotReloadStorageSyncer) Sync() error {
|
|
// 定时器
|
|
ticker := time.NewTicker(h.duration)
|
|
for {
|
|
select {
|
|
case <-ticker.C:
|
|
// 每隔一段时间, 将数据同步到storage中
|
|
if err := h.bot.DumpHotReloadStorage(); err != nil {
|
|
return err
|
|
}
|
|
case <-h.bot.Context().Done():
|
|
// 当Bot关闭的时候, 退出循环
|
|
return nil
|
|
}
|
|
}
|
|
}
|
|
|
|
func NewHotReloadStorageSyncer(bot *Bot, duration time.Duration) *HotReloadStorageSyncer {
|
|
return &HotReloadStorageSyncer{duration: duration, bot: bot}
|
|
}
|