Merge branch 'develop'
This commit is contained in:
commit
013436ab03
|
@ -19,6 +19,10 @@ func (slf *Player[ID]) GetID() ID {
|
||||||
return slf.id
|
return slf.id
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (slf *Player[ID]) GetConn() *server.Conn {
|
||||||
|
return slf.conn
|
||||||
|
}
|
||||||
|
|
||||||
func (slf *Player[ID]) UseConn(conn *server.Conn) {
|
func (slf *Player[ID]) UseConn(conn *server.Conn) {
|
||||||
if conn == nil {
|
if conn == nil {
|
||||||
return
|
return
|
||||||
|
|
|
@ -6,6 +6,8 @@ import "github.com/kercylan98/minotaur/server"
|
||||||
type Player[ID comparable] interface {
|
type Player[ID comparable] interface {
|
||||||
// GetID 获取玩家ID
|
// GetID 获取玩家ID
|
||||||
GetID() ID
|
GetID() ID
|
||||||
|
// GetConn 获取玩家连接
|
||||||
|
GetConn() *server.Conn
|
||||||
// UseConn 指定连接
|
// UseConn 指定连接
|
||||||
UseConn(conn *server.Conn)
|
UseConn(conn *server.Conn)
|
||||||
// Close 关闭玩家并且释放其资源
|
// Close 关闭玩家并且释放其资源
|
||||||
|
|
|
@ -0,0 +1,51 @@
|
||||||
|
package storage
|
||||||
|
|
||||||
|
// NewGlobalData 创建全局数据
|
||||||
|
func NewGlobalData[T any](name string, storage GlobalDataStorage[T]) *GlobalData[T] {
|
||||||
|
data := &GlobalData[T]{
|
||||||
|
storage: storage,
|
||||||
|
name: name,
|
||||||
|
data: storage.Load(name),
|
||||||
|
}
|
||||||
|
globalDataSaveHandles = append(globalDataSaveHandles, data.SaveData)
|
||||||
|
return data
|
||||||
|
}
|
||||||
|
|
||||||
|
// GlobalData 全局数据
|
||||||
|
type GlobalData[T any] struct {
|
||||||
|
storage GlobalDataStorage[T] // 全局数据存储器
|
||||||
|
name string // 全局数据名称
|
||||||
|
data T // 数据
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetName 获取名称
|
||||||
|
func (slf *GlobalData[T]) GetName() string {
|
||||||
|
return slf.name
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetData 获取数据
|
||||||
|
func (slf *GlobalData[T]) GetData() T {
|
||||||
|
return slf.data
|
||||||
|
}
|
||||||
|
|
||||||
|
// LoadData 加载数据
|
||||||
|
func (slf *GlobalData[T]) LoadData() {
|
||||||
|
slf.data = slf.storage.Load(slf.GetName())
|
||||||
|
}
|
||||||
|
|
||||||
|
// SaveData 保存数据
|
||||||
|
func (slf *GlobalData[T]) SaveData() error {
|
||||||
|
return slf.storage.Save(slf.GetName(), slf.GetData())
|
||||||
|
}
|
||||||
|
|
||||||
|
// Handle 处理数据
|
||||||
|
func (slf *GlobalData[T]) Handle(handler func(name string, data T)) *GlobalData[T] {
|
||||||
|
handler(slf.GetName(), slf.GetData())
|
||||||
|
return slf
|
||||||
|
}
|
||||||
|
|
||||||
|
// HandleWithCallback 处理数据
|
||||||
|
func (slf *GlobalData[T]) HandleWithCallback(handler func(name string, data T) error, callback func(err error)) *GlobalData[T] {
|
||||||
|
callback(handler(slf.GetName(), slf.GetData()))
|
||||||
|
return slf
|
||||||
|
}
|
|
@ -0,0 +1,10 @@
|
||||||
|
package storage
|
||||||
|
|
||||||
|
// GlobalDataStorage 全局数据存储器接口
|
||||||
|
type GlobalDataStorage[T any] interface {
|
||||||
|
// Load 加载全局数据
|
||||||
|
// - 当全局数据不存在时,应当返回新的全局数据实例
|
||||||
|
Load(name string) T
|
||||||
|
// Save 保存全局数据
|
||||||
|
Save(name string, data T) error
|
||||||
|
}
|
|
@ -0,0 +1,92 @@
|
||||||
|
package storage
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/kercylan98/minotaur/utils/generic"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// NewIndexData 创建索引数据
|
||||||
|
func NewIndexData[I generic.Ordered, T IndexDataItem[I]](name string, storage IndexDataStorage[I, T]) *IndexData[I, T] {
|
||||||
|
data := &IndexData[I, T]{
|
||||||
|
storage: storage,
|
||||||
|
name: name,
|
||||||
|
data: storage.LoadAll(name),
|
||||||
|
}
|
||||||
|
return data
|
||||||
|
}
|
||||||
|
|
||||||
|
// IndexData 全局数据
|
||||||
|
type IndexData[I generic.Ordered, T IndexDataItem[I]] struct {
|
||||||
|
storage IndexDataStorage[I, T] // 存储器
|
||||||
|
name string // 数据组名称
|
||||||
|
data map[I]T // 数据
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetName 获取名称
|
||||||
|
func (slf *IndexData[I, T]) GetName() string {
|
||||||
|
return slf.name
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetData 获取数据
|
||||||
|
func (slf *IndexData[I, T]) GetData(index I) T {
|
||||||
|
data, exist := slf.data[index]
|
||||||
|
if !exist {
|
||||||
|
slf.LoadData(index)
|
||||||
|
data = slf.data[index]
|
||||||
|
}
|
||||||
|
return data
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetAllData 获取所有数据
|
||||||
|
func (slf *IndexData[I, T]) GetAllData() map[I]T {
|
||||||
|
return slf.data
|
||||||
|
}
|
||||||
|
|
||||||
|
// LoadData 加载数据
|
||||||
|
func (slf *IndexData[I, T]) LoadData(index I) {
|
||||||
|
slf.data[index] = slf.storage.Load(slf.GetName(), index)
|
||||||
|
}
|
||||||
|
|
||||||
|
// LoadAllData 加载所有数据
|
||||||
|
func (slf *IndexData[I, T]) LoadAllData() {
|
||||||
|
slf.data = slf.storage.LoadAll(slf.GetName())
|
||||||
|
}
|
||||||
|
|
||||||
|
// SaveData 保存数据
|
||||||
|
func (slf *IndexData[I, T]) SaveData(index I) error {
|
||||||
|
return slf.storage.Save(slf.GetName(), index, slf.GetData(index))
|
||||||
|
}
|
||||||
|
|
||||||
|
// SaveAllData 保存所有数据
|
||||||
|
// - errHandle 错误处理中如果返回 false 将重试,否则跳过当前保存下一个
|
||||||
|
func (slf *IndexData[I, T]) SaveAllData(errHandle func(err error) bool, retryInterval time.Duration) {
|
||||||
|
slf.storage.SaveAll(slf.GetName(), slf.GetAllData(), errHandle, retryInterval)
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteData 删除数据
|
||||||
|
func (slf *IndexData[I, T]) DeleteData(index I) *IndexData[I, T] {
|
||||||
|
slf.storage.Delete(slf.GetName(), index)
|
||||||
|
delete(slf.data, index)
|
||||||
|
return slf
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteAllData 删除所有数据
|
||||||
|
func (slf *IndexData[I, T]) DeleteAllData() *IndexData[I, T] {
|
||||||
|
slf.storage.DeleteAll(slf.GetName())
|
||||||
|
for k := range slf.data {
|
||||||
|
delete(slf.data, k)
|
||||||
|
}
|
||||||
|
return slf
|
||||||
|
}
|
||||||
|
|
||||||
|
// Handle 处理数据
|
||||||
|
func (slf *IndexData[I, T]) Handle(index I, handler func(name string, index I, data T)) *IndexData[I, T] {
|
||||||
|
handler(slf.GetName(), index, slf.GetData(index))
|
||||||
|
return slf
|
||||||
|
}
|
||||||
|
|
||||||
|
// HandleWithCallback 处理数据
|
||||||
|
func (slf *IndexData[I, T]) HandleWithCallback(index I, handler func(name string, index I, data T) error, callback func(err error)) *IndexData[I, T] {
|
||||||
|
callback(handler(slf.GetName(), index, slf.GetData(index)))
|
||||||
|
return slf
|
||||||
|
}
|
|
@ -0,0 +1,7 @@
|
||||||
|
package storage
|
||||||
|
|
||||||
|
import "github.com/kercylan98/minotaur/utils/generic"
|
||||||
|
|
||||||
|
type IndexDataItem[I generic.Ordered] interface {
|
||||||
|
GetIndex() I
|
||||||
|
}
|
|
@ -0,0 +1,23 @@
|
||||||
|
package storage
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/kercylan98/minotaur/utils/generic"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// IndexDataStorage 全局数据存储器接口
|
||||||
|
type IndexDataStorage[I generic.Ordered, T IndexDataItem[I]] interface {
|
||||||
|
// Load 加载特定索引数据
|
||||||
|
// - 通常情况下当数据不存在时,应当返回空指针
|
||||||
|
Load(name string, index I) T
|
||||||
|
// LoadAll 加载所有数据
|
||||||
|
LoadAll(name string) map[I]T
|
||||||
|
// Save 保存特定索引数据
|
||||||
|
Save(name string, index I, data T) error
|
||||||
|
// SaveAll 保存所有数据
|
||||||
|
SaveAll(name string, data map[I]T, errHandle func(err error) bool, retryInterval time.Duration)
|
||||||
|
// Delete 删除特定索引数据
|
||||||
|
Delete(name string, index I)
|
||||||
|
// DeleteAll 删除所有数据
|
||||||
|
DeleteAll(name string)
|
||||||
|
}
|
|
@ -0,0 +1,26 @@
|
||||||
|
package storage
|
||||||
|
|
||||||
|
import "time"
|
||||||
|
|
||||||
|
var (
|
||||||
|
// globalDataSaveHandles 全局数据保存句柄
|
||||||
|
globalDataSaveHandles []func() error
|
||||||
|
)
|
||||||
|
|
||||||
|
// SaveAll 保存所有数据
|
||||||
|
// - errorHandle 错误处理中如果返回 false 将重试,否则跳过当前保存下一个
|
||||||
|
func SaveAll(errorHandle func(err error) bool, retryInterval time.Duration) {
|
||||||
|
var err error
|
||||||
|
for _, handle := range globalDataSaveHandles {
|
||||||
|
for {
|
||||||
|
if err = handle(); err != nil {
|
||||||
|
if !errorHandle(err) {
|
||||||
|
time.Sleep(retryInterval)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
break
|
||||||
|
}
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,23 @@
|
||||||
|
package storages
|
||||||
|
|
||||||
|
import "encoding/json"
|
||||||
|
|
||||||
|
// FileStorageEncoder 全局数据文件存储编码器
|
||||||
|
type FileStorageEncoder[T any] func(data T) ([]byte, error)
|
||||||
|
|
||||||
|
// FileStorageDecoder 全局数据文件存储解码器
|
||||||
|
type FileStorageDecoder[T any] func(bytes []byte, data T) error
|
||||||
|
|
||||||
|
// FileStorageJSONEncoder JSON 编码器
|
||||||
|
func FileStorageJSONEncoder[T any]() FileStorageEncoder[T] {
|
||||||
|
return func(data T) ([]byte, error) {
|
||||||
|
return json.Marshal(data)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// FileStorageJSONDecoder JSON 解码器
|
||||||
|
func FileStorageJSONDecoder[T any]() FileStorageDecoder[T] {
|
||||||
|
return func(bytes []byte, data T) error {
|
||||||
|
return json.Unmarshal(bytes, data)
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1 @@
|
||||||
|
{"CreateAt":"2023-07-19T14:49:35.7235348+08:00","TotalCount":10}
|
|
@ -0,0 +1 @@
|
||||||
|
{"ID":"INDEX_001","Value":10}
|
|
@ -0,0 +1,63 @@
|
||||||
|
package storages
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"github.com/kercylan98/minotaur/utils/file"
|
||||||
|
"path/filepath"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
// GlobalDataFileDefaultSuffix 是 GlobalDataFileStorage 的文件默认后缀
|
||||||
|
GlobalDataFileDefaultSuffix = "stock"
|
||||||
|
)
|
||||||
|
|
||||||
|
// NewGlobalDataFileStorage 创建一个 GlobalDataFileStorage,dir 是文件存储目录,generate 是生成数据的函数,options 是可选参数
|
||||||
|
// - 生成的文件名为 ${name}.${suffix},可以通过 WithGlobalDataFileStorageSuffix 修改后缀
|
||||||
|
// - 默认使用 JSON 格式存储,可以通过 WithGlobalDataFileStorageEncoder 和 WithGlobalDataFileStorageDecoder 修改编码和解码函数
|
||||||
|
// - 内置了 JSON 编码和解码函数,可以通过 FileStorageJSONEncoder 和 FileStorageJSONDecoder 获取
|
||||||
|
func NewGlobalDataFileStorage[T any](dir string, generate func(name string) T, options ...GlobalDataFileStorageOption[T]) *GlobalDataFileStorage[T] {
|
||||||
|
abs, err := filepath.Abs(dir)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
storage := &GlobalDataFileStorage[T]{
|
||||||
|
dir: abs,
|
||||||
|
suffix: GlobalDataFileDefaultSuffix,
|
||||||
|
generate: generate,
|
||||||
|
encoder: FileStorageJSONEncoder[T](),
|
||||||
|
decoder: FileStorageJSONDecoder[T](),
|
||||||
|
}
|
||||||
|
for _, option := range options {
|
||||||
|
option(storage)
|
||||||
|
}
|
||||||
|
return storage
|
||||||
|
}
|
||||||
|
|
||||||
|
// GlobalDataFileStorage 用于存储全局数据的文件存储器
|
||||||
|
type GlobalDataFileStorage[T any] struct {
|
||||||
|
dir string
|
||||||
|
suffix string
|
||||||
|
generate func(name string) T
|
||||||
|
encoder FileStorageEncoder[T]
|
||||||
|
decoder FileStorageDecoder[T]
|
||||||
|
}
|
||||||
|
|
||||||
|
// Load 从文件中加载数据,如果文件不存在则使用 generate 生成数据
|
||||||
|
func (slf *GlobalDataFileStorage[T]) Load(name string) T {
|
||||||
|
bytes, err := file.ReadOnce(filepath.Join(slf.dir, fmt.Sprintf("%s.%s", name, slf.suffix)))
|
||||||
|
if err != nil {
|
||||||
|
return slf.generate(name)
|
||||||
|
}
|
||||||
|
var data = slf.generate(name)
|
||||||
|
_ = slf.decoder(bytes, data)
|
||||||
|
return data
|
||||||
|
}
|
||||||
|
|
||||||
|
// Save 将数据保存到文件中
|
||||||
|
func (slf *GlobalDataFileStorage[T]) Save(name string, data T) error {
|
||||||
|
bytes, err := slf.encoder(data)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return file.WriterFile(filepath.Join(slf.dir, fmt.Sprintf("%s.%s", name, slf.suffix)), bytes)
|
||||||
|
}
|
|
@ -0,0 +1,19 @@
|
||||||
|
package storages_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"github.com/kercylan98/minotaur/utils/storage/storages"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func ExampleNewGlobalDataFileStorage() {
|
||||||
|
storage := storages.NewGlobalDataFileStorage[*GlobalData](".", func(name string) *GlobalData {
|
||||||
|
return &GlobalData{
|
||||||
|
CreateAt: time.Now(),
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
fmt.Println(storage != nil)
|
||||||
|
// Output:
|
||||||
|
// true
|
||||||
|
}
|
|
@ -0,0 +1,28 @@
|
||||||
|
package storages
|
||||||
|
|
||||||
|
// GlobalDataFileStorageOption 全局数据文件存储选项
|
||||||
|
type GlobalDataFileStorageOption[T any] func(storage *GlobalDataFileStorage[T])
|
||||||
|
|
||||||
|
// WithGlobalDataFileStorageEncoder 设置编码器
|
||||||
|
// - 默认为 FileStorageJSONEncoder 编码器
|
||||||
|
func WithGlobalDataFileStorageEncoder[T any](encoder FileStorageEncoder[T]) GlobalDataFileStorageOption[T] {
|
||||||
|
return func(storage *GlobalDataFileStorage[T]) {
|
||||||
|
storage.encoder = encoder
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// WithGlobalDataFileStorageDecoder 设置解码器
|
||||||
|
// - 默认为 FileStorageJSONDecoder 解码器
|
||||||
|
func WithGlobalDataFileStorageDecoder[T any](decoder FileStorageDecoder[T]) GlobalDataFileStorageOption[T] {
|
||||||
|
return func(storage *GlobalDataFileStorage[T]) {
|
||||||
|
storage.decoder = decoder
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// WithGlobalDataFileStorageSuffix 设置文件后缀
|
||||||
|
// - 默认为 GlobalDataFileDefaultSuffix
|
||||||
|
func WithGlobalDataFileStorageSuffix[T any](suffix string) GlobalDataFileStorageOption[T] {
|
||||||
|
return func(storage *GlobalDataFileStorage[T]) {
|
||||||
|
storage.suffix = suffix
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,42 @@
|
||||||
|
package storages_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/kercylan98/minotaur/utils/storage"
|
||||||
|
"github.com/kercylan98/minotaur/utils/storage/storages"
|
||||||
|
. "github.com/smartystreets/goconvey/convey"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type GlobalData struct {
|
||||||
|
CreateAt time.Time
|
||||||
|
TotalCount int
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGlobalDataFileStorage_Save(t *testing.T) {
|
||||||
|
Convey("TestGlobalDataFileStorage_Save", t, func() {
|
||||||
|
data := storage.NewGlobalData[*GlobalData]("global_data_file_test", storages.NewGlobalDataFileStorage[*GlobalData]("./example-data", func(name string) *GlobalData {
|
||||||
|
return &GlobalData{
|
||||||
|
CreateAt: time.Now(),
|
||||||
|
}
|
||||||
|
}))
|
||||||
|
data.Handle(func(name string, data *GlobalData) {
|
||||||
|
data.TotalCount = 10
|
||||||
|
})
|
||||||
|
if err := data.SaveData(); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
So(data.GetData().TotalCount, ShouldEqual, 10)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGlobalDataFileStorage_Load(t *testing.T) {
|
||||||
|
Convey("TestGlobalDataFileStorage_Load", t, func() {
|
||||||
|
data := storage.NewGlobalData[*GlobalData]("global_data_file_test", storages.NewGlobalDataFileStorage[*GlobalData]("./example-data", func(name string) *GlobalData {
|
||||||
|
return &GlobalData{
|
||||||
|
CreateAt: time.Now(),
|
||||||
|
}
|
||||||
|
}))
|
||||||
|
So(data.GetData().TotalCount, ShouldEqual, 10)
|
||||||
|
})
|
||||||
|
}
|
|
@ -0,0 +1,117 @@
|
||||||
|
package storages
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"github.com/kercylan98/minotaur/utils/file"
|
||||||
|
"github.com/kercylan98/minotaur/utils/generic"
|
||||||
|
"github.com/kercylan98/minotaur/utils/storage"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
// IndexDataFileDefaultSuffix 是 IndexDataFileStorage 的文件默认后缀
|
||||||
|
IndexDataFileDefaultSuffix = "stock"
|
||||||
|
|
||||||
|
indexNameFormat = "%s.%v.%s"
|
||||||
|
)
|
||||||
|
|
||||||
|
// NewIndexDataFileStorage 创建索引数据文件存储器
|
||||||
|
func NewIndexDataFileStorage[I generic.Ordered, T storage.IndexDataItem[I]](dir string, generate func(name string, index I) T, generateZero func(name string) T, options ...IndexDataFileStorageOption[I, T]) *IndexDataFileStorage[I, T] {
|
||||||
|
s := &IndexDataFileStorage[I, T]{
|
||||||
|
dir: dir,
|
||||||
|
suffix: IndexDataFileDefaultSuffix,
|
||||||
|
generate: generate,
|
||||||
|
generateZero: generateZero,
|
||||||
|
encoder: FileStorageJSONEncoder[T](),
|
||||||
|
decoder: FileStorageJSONDecoder[T](),
|
||||||
|
}
|
||||||
|
for _, option := range options {
|
||||||
|
option(s)
|
||||||
|
}
|
||||||
|
return s
|
||||||
|
}
|
||||||
|
|
||||||
|
// IndexDataFileStorage 索引数据文件存储器
|
||||||
|
type IndexDataFileStorage[I generic.Ordered, T storage.IndexDataItem[I]] struct {
|
||||||
|
dir string
|
||||||
|
suffix string
|
||||||
|
generate func(name string, index I) T
|
||||||
|
generateZero func(name string) T
|
||||||
|
encoder FileStorageEncoder[T]
|
||||||
|
decoder FileStorageDecoder[T]
|
||||||
|
}
|
||||||
|
|
||||||
|
func (slf *IndexDataFileStorage[I, T]) Load(name string, index I) T {
|
||||||
|
bytes, err := file.ReadOnce(filepath.Join(slf.dir, fmt.Sprintf(indexNameFormat, name, index, slf.suffix)))
|
||||||
|
if err != nil {
|
||||||
|
return slf.generate(name, index)
|
||||||
|
}
|
||||||
|
var data = slf.generate(name, index)
|
||||||
|
_ = slf.decoder(bytes, data)
|
||||||
|
return data
|
||||||
|
}
|
||||||
|
|
||||||
|
func (slf *IndexDataFileStorage[I, T]) LoadAll(name string) map[I]T {
|
||||||
|
var result = make(map[I]T)
|
||||||
|
files, err := os.ReadDir(slf.dir)
|
||||||
|
if err != nil {
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
for _, entry := range files {
|
||||||
|
if entry.IsDir() || !strings.HasPrefix(entry.Name(), name) || !strings.HasSuffix(entry.Name(), slf.suffix) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
bytes, err := file.ReadOnce(filepath.Join(slf.dir, entry.Name()))
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
data := slf.generateZero(name)
|
||||||
|
if err := slf.decoder(bytes, data); err == nil {
|
||||||
|
result[data.GetIndex()] = data
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
func (slf *IndexDataFileStorage[I, T]) Save(name string, index I, data T) error {
|
||||||
|
bytes, err := slf.encoder(data)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return file.WriterFile(filepath.Join(slf.dir, fmt.Sprintf(indexNameFormat, name, index, slf.suffix)), bytes)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (slf *IndexDataFileStorage[I, T]) SaveAll(name string, data map[I]T, errHandle func(err error) bool, retryInterval time.Duration) {
|
||||||
|
for index, data := range data {
|
||||||
|
for {
|
||||||
|
if err := slf.Save(name, index, data); err != nil {
|
||||||
|
if !errHandle(err) {
|
||||||
|
time.Sleep(retryInterval)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
break
|
||||||
|
}
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (slf *IndexDataFileStorage[I, T]) Delete(name string, index I) {
|
||||||
|
_ = os.Remove(filepath.Join(slf.dir, fmt.Sprintf(indexNameFormat, name, index, slf.suffix)))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (slf *IndexDataFileStorage[I, T]) DeleteAll(name string) {
|
||||||
|
files, err := os.ReadDir(slf.dir)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
for _, entry := range files {
|
||||||
|
if entry.IsDir() || !strings.HasPrefix(entry.Name(), name) || !strings.HasSuffix(entry.Name(), slf.suffix) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
_ = os.Remove(filepath.Join(slf.dir, entry.Name()))
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,18 @@
|
||||||
|
package storages_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"github.com/kercylan98/minotaur/utils/storage/storages"
|
||||||
|
)
|
||||||
|
|
||||||
|
func ExampleNewIndexDataFileStorage() {
|
||||||
|
storage := storages.NewIndexDataFileStorage[string, *IndexData[string]]("./example-data", func(name string, index string) *IndexData[string] {
|
||||||
|
return &IndexData[string]{ID: index}
|
||||||
|
}, func(name string) *IndexData[string] {
|
||||||
|
return new(IndexData[string])
|
||||||
|
})
|
||||||
|
|
||||||
|
fmt.Println(storage != nil)
|
||||||
|
// Output:
|
||||||
|
// true
|
||||||
|
}
|
|
@ -0,0 +1,33 @@
|
||||||
|
package storages
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/kercylan98/minotaur/utils/generic"
|
||||||
|
"github.com/kercylan98/minotaur/utils/storage"
|
||||||
|
)
|
||||||
|
|
||||||
|
// IndexDataFileStorageOption 索引数据文件存储器选项
|
||||||
|
type IndexDataFileStorageOption[I generic.Ordered, T storage.IndexDataItem[I]] func(storage *IndexDataFileStorage[I, T])
|
||||||
|
|
||||||
|
// WithIndexDataFileStorageEncoder 设置编码器
|
||||||
|
// - 默认为 FileStorageJSONEncoder 编码器
|
||||||
|
func WithIndexDataFileStorageEncoder[I generic.Ordered, T storage.IndexDataItem[I]](encoder FileStorageEncoder[T]) IndexDataFileStorageOption[I, T] {
|
||||||
|
return func(storage *IndexDataFileStorage[I, T]) {
|
||||||
|
storage.encoder = encoder
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// WithIndexDataFileStorageDecoder 设置解码器
|
||||||
|
// - 默认为 FileStorageJSONDecoder 解码器
|
||||||
|
func WithIndexDataFileStorageDecoder[I generic.Ordered, T storage.IndexDataItem[I]](decoder FileStorageDecoder[T]) IndexDataFileStorageOption[I, T] {
|
||||||
|
return func(storage *IndexDataFileStorage[I, T]) {
|
||||||
|
storage.decoder = decoder
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// WithIndexDataFileStorageSuffix 设置文件后缀
|
||||||
|
// - 默认为 IndexDataFileDefaultSuffix
|
||||||
|
func WithIndexDataFileStorageSuffix[I generic.Ordered, T storage.IndexDataItem[I]](suffix string) IndexDataFileStorageOption[I, T] {
|
||||||
|
return func(storage *IndexDataFileStorage[I, T]) {
|
||||||
|
storage.suffix = suffix
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,45 @@
|
||||||
|
package storages_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/kercylan98/minotaur/utils/storage"
|
||||||
|
"github.com/kercylan98/minotaur/utils/storage/storages"
|
||||||
|
. "github.com/smartystreets/goconvey/convey"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
type IndexData[I string] struct {
|
||||||
|
ID I
|
||||||
|
Value int
|
||||||
|
}
|
||||||
|
|
||||||
|
func (slf *IndexData[I]) GetIndex() I {
|
||||||
|
return slf.ID
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestIndexDataFileStorage_Save(t *testing.T) {
|
||||||
|
Convey("TestIndexDataFileStorage_Save", t, func() {
|
||||||
|
data := storage.NewIndexData[string, *IndexData[string]]("index_data_file_test", storages.NewIndexDataFileStorage[string, *IndexData[string]]("./example-data", func(name string, index string) *IndexData[string] {
|
||||||
|
return &IndexData[string]{ID: index}
|
||||||
|
}, func(name string) *IndexData[string] {
|
||||||
|
return new(IndexData[string])
|
||||||
|
}))
|
||||||
|
data.Handle("INDEX_001", func(name string, index string, data *IndexData[string]) {
|
||||||
|
data.Value = 10
|
||||||
|
})
|
||||||
|
if err := data.SaveData("INDEX_001"); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
So(data.GetData("INDEX_001").Value, ShouldEqual, 10)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestIndexDataFileStorage_Load(t *testing.T) {
|
||||||
|
Convey("TestIndexDataFileStorage_Load", t, func() {
|
||||||
|
data := storage.NewIndexData[string, *IndexData[string]]("index_data_file_test", storages.NewIndexDataFileStorage[string, *IndexData[string]]("./example-data", func(name string, index string) *IndexData[string] {
|
||||||
|
return &IndexData[string]{ID: index}
|
||||||
|
}, func(name string) *IndexData[string] {
|
||||||
|
return new(IndexData[string])
|
||||||
|
}))
|
||||||
|
So(data.GetData("INDEX_001").Value, ShouldEqual, 10)
|
||||||
|
})
|
||||||
|
}
|
|
@ -0,0 +1,22 @@
|
||||||
|
package super
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
)
|
||||||
|
|
||||||
|
var errorMapper = make(map[error]int)
|
||||||
|
|
||||||
|
// RegError 通过错误码注册错误,返回错误的引用
|
||||||
|
func RegError(code int, message string) error {
|
||||||
|
if code == 0 {
|
||||||
|
panic("error code can not be 0")
|
||||||
|
}
|
||||||
|
err := errors.New(message)
|
||||||
|
errorMapper[err] = code
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetErrorCode 通过错误引用获取错误码,如果错误不存在则返回 0
|
||||||
|
func GetErrorCode(err error) int {
|
||||||
|
return errorMapper[err]
|
||||||
|
}
|
|
@ -17,3 +17,23 @@ func CalcNextSecWithTime(t time.Time, sec int) int {
|
||||||
next := now + int64(sec) - now%int64(sec)
|
next := now + int64(sec) - now%int64(sec)
|
||||||
return int(next - now)
|
return int(next - now)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// CalcNextTimeWithRefer 根据参考时间计算下一个整点时间
|
||||||
|
// - 假设当 now 为 14:15:16 , 参考时间为 10 分钟, 则返回 14:20:00
|
||||||
|
// - 假设当 now 为 14:15:16 , 参考时间为 20 分钟, 则返回 14:20:00
|
||||||
|
//
|
||||||
|
// 当 refer 小于 1 分钟时,将会返回当前时间
|
||||||
|
func CalcNextTimeWithRefer(now time.Time, refer time.Duration) time.Time {
|
||||||
|
referInSeconds := int(refer.Minutes()) * 60
|
||||||
|
if referInSeconds <= 0 {
|
||||||
|
return now
|
||||||
|
}
|
||||||
|
|
||||||
|
minutes := now.Minute()
|
||||||
|
seconds := now.Second()
|
||||||
|
|
||||||
|
remainder := referInSeconds - (minutes*60+seconds)%referInSeconds
|
||||||
|
nextTime := now.Add(time.Duration(remainder) * time.Second)
|
||||||
|
nextTime = time.Date(nextTime.Year(), nextTime.Month(), nextTime.Day(), nextTime.Hour(), nextTime.Minute(), 0, 0, nextTime.Location())
|
||||||
|
return nextTime
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in New Issue