38 lines
602 B
Go
38 lines
602 B
Go
|
package ccache
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Bucket struct {
|
||
|
|
sync.RWMutex
|
||
|
|
lookup map[string]*Item
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Bucket) Get(key string) *Item {
|
||
|
|
b.RLock()
|
||
|
|
defer b.RUnlock()
|
||
|
|
return b.lookup[key]
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Bucket) Set(key string, value Value) *Item {
|
||
|
|
b.Lock()
|
||
|
|
defer b.Unlock()
|
||
|
|
if existing, exists := b.lookup[key]; exists {
|
||
|
|
existing.Lock()
|
||
|
|
existing.value = value
|
||
|
|
existing.Unlock()
|
||
|
|
return existing
|
||
|
|
}
|
||
|
|
item := &Item{key: key, value: value}
|
||
|
|
b.lookup[key] = item
|
||
|
|
return item
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
func (b *Bucket) Remove(key string) {
|
||
|
|
b.Lock()
|
||
|
|
defer b.Unlock()
|
||
|
|
delete(b.lookup, key)
|
||
|
|
}
|