49 lines
1.2 KiB
Go
49 lines
1.2 KiB
Go
|
package cache
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"sync"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// SliceCache结构体代表基于切片的缓存，内部使用切片存储元素，并通过读写锁保证并发安全
|
|||
|
|
type SliceCache struct {
|
|||
|
|
data []interface{}
|
|||
|
|
rwmu sync.RWMutex
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewSliceCache创建并返回一个新的基于切片的缓存实例
|
|||
|
|
func NewSliceCache() *SliceCache {
|
|||
|
|
return &SliceCache{
|
|||
|
|
data: make([]interface{}, 0),
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Add方法用于向切片缓存中添加元素，写操作加写锁保证并发安全
|
|||
|
|
func (sc *SliceCache) Add(element interface{}) {
|
|||
|
|
sc.rwmu.Lock()
|
|||
|
|
sc.data = append(sc.data, element)
|
|||
|
|
sc.rwmu.Unlock()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetAll方法用于获取切片缓存中的所有元素，读操作加读锁允许并发读
|
|||
|
|
func (sc *SliceCache) GetAll() []interface{} {
|
|||
|
|
sc.rwmu.RLock()
|
|||
|
|
result := make([]interface{}, len(sc.data))
|
|||
|
|
copy(result, sc.data)
|
|||
|
|
sc.rwmu.RUnlock()
|
|||
|
|
return result
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Remove方法用于从切片缓存中删除指定元素，写操作加写锁保证并发安全
|
|||
|
|
func (sc *SliceCache) Remove(element interface{}) bool {
|
|||
|
|
sc.rwmu.Lock()
|
|||
|
|
for i, e := range sc.data {
|
|||
|
|
if e == element {
|
|||
|
|
sc.data = append(sc.data[:i], sc.data[i+1:]...)
|
|||
|
|
sc.rwmu.Unlock()
|
|||
|
|
return true
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
sc.rwmu.Unlock()
|
|||
|
|
return false
|
|||
|
|
}
|