mirror of
https://github.com/featurebasedb/featurebase.git
synced 2026-09-07 00:55:55 +00:00
48aef0c8a4
```bash for file in `cat diffys`; do printf '%s\n%s\n' "// Copyright 2021 Molecula Corp. All rights reserved." "$(cat $file)" >$file; done ```
53 lines
1 KiB
Go
53 lines
1 KiB
Go
// Copyright 2021 Molecula Corp. All rights reserved.
|
|
// package ctl contains all pilosa subcommands other than 'server'. These are
|
|
// generally administration, testing, and debugging tools.
|
|
|
|
package client
|
|
|
|
import (
|
|
"sync"
|
|
|
|
pnet "github.com/molecula/featurebase/v2/net"
|
|
)
|
|
|
|
type shardNodes struct {
|
|
data map[string]map[uint64][]*pnet.URI
|
|
mu *sync.RWMutex
|
|
}
|
|
|
|
func newShardNodes() shardNodes {
|
|
return shardNodes{
|
|
data: make(map[string]map[uint64][]*pnet.URI),
|
|
mu: &sync.RWMutex{},
|
|
}
|
|
}
|
|
|
|
func (s shardNodes) Get(index string, shard uint64) ([]*pnet.URI, bool) {
|
|
s.mu.RLock()
|
|
defer s.mu.RUnlock()
|
|
if idx, ok := s.data[index]; ok {
|
|
if uris, ok := idx[shard]; ok {
|
|
return uris, true
|
|
}
|
|
}
|
|
return nil, false
|
|
}
|
|
|
|
func (s shardNodes) Put(index string, shard uint64, uris []*pnet.URI) {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
idx, ok := s.data[index]
|
|
if !ok {
|
|
idx = make(map[uint64][]*pnet.URI)
|
|
}
|
|
idx[shard] = uris
|
|
s.data[index] = idx
|
|
}
|
|
|
|
func (s shardNodes) Invalidate() {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
for k := range s.data {
|
|
delete(s.data, k)
|
|
}
|
|
}
|