You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
53 lines
1.3 KiB
53 lines
1.3 KiB
// Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
|
|
//
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
|
// If a copy of the MIT was not distributed with this file,
|
|
// You can obtain one at https://github.com/gogf/gf.
|
|
|
|
package gcache
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/gogf/gf/v2/container/gset"
|
|
)
|
|
|
|
type adapterMemoryExpireSets struct {
|
|
mu sync.RWMutex // expireSetMu ensures the concurrent safety of expireSets map.
|
|
expireSets map[int64]*gset.Set // expireSets is the expiring timestamp to its key set mapping, which is used for quick indexing and deleting.
|
|
}
|
|
|
|
func newAdapterMemoryExpireSets() *adapterMemoryExpireSets {
|
|
return &adapterMemoryExpireSets{
|
|
expireSets: make(map[int64]*gset.Set),
|
|
}
|
|
}
|
|
|
|
func (d *adapterMemoryExpireSets) Get(key int64) (result *gset.Set) {
|
|
d.mu.RLock()
|
|
result = d.expireSets[key]
|
|
d.mu.RUnlock()
|
|
return
|
|
}
|
|
|
|
func (d *adapterMemoryExpireSets) GetOrNew(key int64) (result *gset.Set) {
|
|
if result = d.Get(key); result != nil {
|
|
return
|
|
}
|
|
d.mu.Lock()
|
|
if es, ok := d.expireSets[key]; ok {
|
|
result = es
|
|
} else {
|
|
result = gset.New(true)
|
|
d.expireSets[key] = result
|
|
}
|
|
d.mu.Unlock()
|
|
return
|
|
}
|
|
|
|
func (d *adapterMemoryExpireSets) Delete(key int64) {
|
|
d.mu.Lock()
|
|
delete(d.expireSets, key)
|
|
d.mu.Unlock()
|
|
}
|