hanfuit
645b64fe21
服务端（Go）： - GM 塞背包/带属性装备发放：在线角色推送 op14 增量物品行，背包实时刷新，无需回角色选择界面 - GM 改金币/改等级：去掉在线拒绝，落库后推送 op14 钱包/op37 等级状态，在线实时刷新 - 等级端点统一切换到新版 GMSetCharacterLevel（自带在线刷新） - 无限/锁死疲劳模式 + 燃烧疲劳：每房间消耗 2 点、打怪与通关经验双倍 - 燃烧疲劳开关时给所有在线角色推送系统提示（op12 mode 0） - 深渊史诗掉率：GM 全局倍率 + 城镇增益药剂窗口倍率，真实接入掉落结算 启动器（Python/JS）： - 标题栏自绘、窗口 1204x717、禁止拖拽改大小、仅最小化/关闭 - GM 工具：疲劳与燃烧、深渊史诗掉率、拍卖行上架（单一定价对齐） - 拍卖行上架前端对齐服务端单一价格语义，修复 400 报错
97 lines
2.2 KiB
Go
97 lines
2.2 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
"sync"
|
|
)
|
|
|
|
// MemoryStore 是线程安全的内存记录存储。
|
|
type MemoryStore[T any] struct {
|
|
mu sync.RWMutex
|
|
records map[string]T
|
|
keyFn KeyFunc[T]
|
|
cloneFn CloneFunc[T]
|
|
}
|
|
|
|
// NewMemoryStore 创建内存记录存储。
|
|
func NewMemoryStore[T any](keyFn KeyFunc[T], cloneFn CloneFunc[T]) *MemoryStore[T] {
|
|
if cloneFn == nil {
|
|
cloneFn = IdentityClone[T]
|
|
}
|
|
return &MemoryStore[T]{
|
|
records: make(map[string]T),
|
|
keyFn: keyFn,
|
|
cloneFn: cloneFn,
|
|
}
|
|
}
|
|
|
|
// Load 从内存按主键读取记录。
|
|
func (s *MemoryStore[T]) Load(_ context.Context, key string) (T, bool, error) {
|
|
s.mu.RLock()
|
|
record, ok := s.records[key]
|
|
s.mu.RUnlock()
|
|
if !ok {
|
|
var zero T
|
|
return zero, false, nil
|
|
}
|
|
return s.cloneFn(record), true, nil
|
|
}
|
|
|
|
// Check 检查上下文是否已取消。
|
|
func (s *MemoryStore[T]) Check(ctx context.Context) error {
|
|
ctx = contextOrBackground(ctx)
|
|
return ctx.Err()
|
|
}
|
|
|
|
// Save 保存记录副本到内存。
|
|
func (s *MemoryStore[T]) Save(_ context.Context, record T) error {
|
|
key, err := RecordKey(s.keyFn, record)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
s.mu.Lock()
|
|
s.records[key] = s.cloneFn(record)
|
|
s.mu.Unlock()
|
|
return nil
|
|
}
|
|
|
|
// Delete removes one record. It is primarily used by in-memory transactional
|
|
// tests to restore an aggregate that did not exist before a failed commit.
|
|
func (s *MemoryStore[T]) Delete(ctx context.Context, key string) error {
|
|
ctx = contextOrBackground(ctx)
|
|
if err := ctx.Err(); err != nil {
|
|
return err
|
|
}
|
|
key = strings.TrimSpace(key)
|
|
if key == "" {
|
|
return ErrRecordKeyRequired
|
|
}
|
|
s.mu.Lock()
|
|
delete(s.records, key)
|
|
s.mu.Unlock()
|
|
return nil
|
|
}
|
|
|
|
// Snapshot 返回全部记录副本，可按 less 排序。
|
|
func (s *MemoryStore[T]) Snapshot(less func(T, T) bool) []T {
|
|
s.mu.RLock()
|
|
out := make([]T, 0, len(s.records))
|
|
for _, record := range s.records {
|
|
out = append(out, s.cloneFn(record))
|
|
}
|
|
s.mu.RUnlock()
|
|
if less != nil {
|
|
sortRecords(out, less)
|
|
}
|
|
return out
|
|
}
|
|
|
|
func sortRecords[T any](records []T, less func(T, T) bool) {
|
|
for i := 1; i < len(records); i++ {
|
|
for j := i; j > 0 && less(records[j], records[j-1]); j-- {
|
|
records[j], records[j-1] = records[j-1], records[j]
|
|
}
|
|
}
|
|
}
|