fzf/src/cache.go

54 lines
1.1 KiB
Go
Raw Normal View History

2015-01-01 19:49:30 +00:00
package fzf
import "sync"
// queryCache associates strings to lists of items
type queryCache map[string][]*Item
2015-01-11 18:01:24 +00:00
// ChunkCache associates Chunk and query string to lists of items
2015-01-01 19:49:30 +00:00
type ChunkCache struct {
mutex sync.Mutex
cache map[*Chunk]*queryCache
2015-01-01 19:49:30 +00:00
}
2015-01-11 18:01:24 +00:00
// NewChunkCache returns a new ChunkCache
2015-01-01 19:49:30 +00:00
func NewChunkCache() ChunkCache {
return ChunkCache{sync.Mutex{}, make(map[*Chunk]*queryCache)}
2015-01-01 19:49:30 +00:00
}
2015-01-11 18:01:24 +00:00
// Add adds the list to the cache
2015-01-01 19:49:30 +00:00
func (cc *ChunkCache) Add(chunk *Chunk, key string, list []*Item) {
if len(key) == 0 || !chunk.IsFull() || len(list) > queryCacheMax {
2015-01-01 19:49:30 +00:00
return
}
cc.mutex.Lock()
defer cc.mutex.Unlock()
qc, ok := cc.cache[chunk]
if !ok {
cc.cache[chunk] = &queryCache{}
2015-01-01 19:49:30 +00:00
qc = cc.cache[chunk]
}
(*qc)[key] = list
}
2015-01-11 18:01:24 +00:00
// Find is called to lookup ChunkCache
2015-01-01 19:49:30 +00:00
func (cc *ChunkCache) Find(chunk *Chunk, key string) ([]*Item, bool) {
if len(key) == 0 || !chunk.IsFull() {
return nil, false
}
cc.mutex.Lock()
defer cc.mutex.Unlock()
qc, ok := cc.cache[chunk]
if ok {
list, ok := (*qc)[key]
if ok {
return list, true
}
}
return nil, false
}