fix(fight): 修复技能效果添加逻辑并优化效果管理
- 修改 `AddEffect` 方法,使用 `EffectID` 包装技能效果,并避免重复添加 - 将 `Effects` 类型从 `[]Effect` 改为 `*utils.OrderedMap[int, Effect]` 以提升查找和管理效率 - 移除 `Effect` 接口中的 `ID()` 方法,改由 `EffectID` 结构体维护 - 增加 `GetSkillEffect` 和 `GetDamageEffect` 方法返回带 ID 的效果结构 - 更新 `CancelTurn` 和 `Exec` 方法以适配新的数据结构 - 初始化 `Effects` 为 `OrderedMap` 实例,确保容器正确创建
This commit is contained in:
206
common/utils/emap.go
Normal file
206
common/utils/emap.go
Normal file
@@ -0,0 +1,206 @@
|
||||
package utils
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"hash/maphash"
|
||||
"sync"
|
||||
)
|
||||
|
||||
// 泛型有序Map的实现,结合了SliceMap的并发安全和orderedMap的有序性及高效查找
|
||||
|
||||
// MapEntry 表示Map中的一个键值对条目
|
||||
type MapEntry[K comparable, V any] struct {
|
||||
key K
|
||||
value V
|
||||
|
||||
// 用于维护插入顺序的双向链表指针
|
||||
prev, next *MapEntry[K, V]
|
||||
// 用于哈希表碰撞处理的链表指针
|
||||
hashNext *MapEntry[K, V]
|
||||
// 标记是否被删除
|
||||
deleted bool
|
||||
}
|
||||
|
||||
// OrderedMap 是一个泛型的有序Map实现
|
||||
type OrderedMap[K comparable, V any] struct {
|
||||
mu sync.RWMutex // 并发安全锁
|
||||
hash *maphash.Hash // 哈希计算器
|
||||
hashTable map[uint64]*MapEntry[K, V] // 哈希表,用于快速查找
|
||||
first *MapEntry[K, V] // 双向链表头,维护插入顺序
|
||||
last *MapEntry[K, V] // 双向链表尾,维护插入顺序
|
||||
size int // 当前元素数量(排除已删除的)
|
||||
}
|
||||
|
||||
// NewOrderedMap 创建一个新的泛型有序Map
|
||||
func NewOrderedMap[K comparable, V any]() *OrderedMap[K, V] {
|
||||
return &OrderedMap[K, V]{
|
||||
hash: &maphash.Hash{},
|
||||
hashTable: make(map[uint64]*MapEntry[K, V]),
|
||||
}
|
||||
}
|
||||
|
||||
// 计算键的哈希值
|
||||
func (m *OrderedMap[K, V]) hashKey(key K) uint64 {
|
||||
// 这里简化实现,实际使用中可能需要根据K的类型做特殊处理
|
||||
// 对于基本类型可以直接写入,复杂类型可能需要序列化
|
||||
m.hash.Reset()
|
||||
// 注意:实际使用时需要根据K的具体类型实现正确的哈希计算
|
||||
// 这里只是示例,可能需要调整
|
||||
m.hash.WriteString(fmt.Sprintf("%v", key))
|
||||
return m.hash.Sum64()
|
||||
}
|
||||
|
||||
// 查找键对应的条目
|
||||
func (m *OrderedMap[K, V]) lookup(key K) (hash uint64, entry, prevEntry *MapEntry[K, V]) {
|
||||
hash = m.hashKey(key)
|
||||
for entry = m.hashTable[hash]; entry != nil; prevEntry, entry = entry, entry.hashNext {
|
||||
if !entry.deleted && entry.key == key {
|
||||
break
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// Store 存储键值对,如果键已存在则更新值
|
||||
func (m *OrderedMap[K, V]) Store(key K, value V) {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
|
||||
hash, entry, prevEntry := m.lookup(key)
|
||||
|
||||
if entry != nil {
|
||||
// 键已存在,更新值并确保标记为未删除
|
||||
entry.value = value
|
||||
if entry.deleted {
|
||||
entry.deleted = false
|
||||
m.size++
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// 创建新条目
|
||||
entry = &MapEntry[K, V]{
|
||||
key: key,
|
||||
value: value,
|
||||
}
|
||||
|
||||
// 添加到哈希表
|
||||
if prevEntry == nil {
|
||||
m.hashTable[hash] = entry
|
||||
} else {
|
||||
prevEntry.hashNext = entry
|
||||
}
|
||||
|
||||
// 添加到双向链表末尾以保持插入顺序
|
||||
if m.last != nil {
|
||||
entry.prev = m.last
|
||||
m.last.next = entry
|
||||
} else {
|
||||
// 第一个元素
|
||||
m.first = entry
|
||||
}
|
||||
m.last = entry
|
||||
m.size++
|
||||
}
|
||||
|
||||
// Load 查找并返回键对应的值,如果不存在则返回零值和false
|
||||
func (m *OrderedMap[K, V]) Load(key K) (value V, exists bool) {
|
||||
m.mu.RLock()
|
||||
defer m.mu.RUnlock()
|
||||
|
||||
_, entry, _ := m.lookup(key)
|
||||
if entry != nil && !entry.deleted {
|
||||
return entry.value, true
|
||||
}
|
||||
return value, false
|
||||
}
|
||||
|
||||
// LoadOrStore 查找并返回键对应的值,如果不存在则存储并返回给定值
|
||||
func (m *OrderedMap[K, V]) LoadOrStore(key K, value V) (actual V, loaded bool) {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
|
||||
_, entry, _ := m.lookup(key)
|
||||
if entry != nil && !entry.deleted {
|
||||
return entry.value, true
|
||||
}
|
||||
|
||||
// 执行存储操作
|
||||
hash := m.hashKey(key)
|
||||
entry = &MapEntry[K, V]{
|
||||
key: key,
|
||||
value: value,
|
||||
}
|
||||
|
||||
// 添加到哈希表
|
||||
if m.hashTable[hash] == nil {
|
||||
m.hashTable[hash] = entry
|
||||
} else {
|
||||
// 处理哈希冲突
|
||||
last := m.hashTable[hash]
|
||||
for last.hashNext != nil {
|
||||
last = last.hashNext
|
||||
}
|
||||
last.hashNext = entry
|
||||
}
|
||||
|
||||
// 添加到双向链表
|
||||
if m.last != nil {
|
||||
entry.prev = m.last
|
||||
m.last.next = entry
|
||||
} else {
|
||||
m.first = entry
|
||||
}
|
||||
m.last = entry
|
||||
m.size++
|
||||
|
||||
return value, false
|
||||
}
|
||||
|
||||
// Delete 标记删除指定的键
|
||||
func (m *OrderedMap[K, V]) Delete(key K) {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
|
||||
_, entry, _ := m.lookup(key)
|
||||
if entry != nil && !entry.deleted {
|
||||
entry.deleted = true
|
||||
m.size--
|
||||
}
|
||||
}
|
||||
|
||||
// Range 按插入顺序迭代Map中的所有键值对
|
||||
// 如果f返回false,则停止迭代
|
||||
func (m *OrderedMap[K, V]) Range(f func(key K, value V) bool) {
|
||||
m.mu.RLock()
|
||||
defer m.mu.RUnlock()
|
||||
|
||||
current := m.first
|
||||
for current != nil {
|
||||
if !current.deleted {
|
||||
if !f(current.key, current.value) {
|
||||
return
|
||||
}
|
||||
}
|
||||
current = current.next
|
||||
}
|
||||
}
|
||||
|
||||
// Len 返回当前Map中的元素数量(排除已删除的)
|
||||
func (m *OrderedMap[K, V]) Len() int {
|
||||
m.mu.RLock()
|
||||
defer m.mu.RUnlock()
|
||||
return m.size
|
||||
}
|
||||
|
||||
// Clear 清空Map中的所有元素
|
||||
func (m *OrderedMap[K, V]) Clear() {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
|
||||
// 重置所有指针和计数器
|
||||
m.first = nil
|
||||
m.last = nil
|
||||
m.hashTable = make(map[uint64]*MapEntry[K, V])
|
||||
m.size = 0
|
||||
}
|
||||
Reference in New Issue
Block a user