78 lines
1.8 KiB
Go
78 lines
1.8 KiB
Go
package openwechat
|
|
|
|
import (
|
|
"io"
|
|
"os"
|
|
)
|
|
|
|
// 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)
|