46 lines
722 B
Go
46 lines
722 B
Go
|
package temporalrepo
|
||
|
|
|
||
|
|
type Memory struct {
|
||
|
|
cache map[int32]map[string]bool
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewMemory() *Memory {
|
||
|
|
return &Memory{
|
||
|
|
cache: map[int32]map[string]bool{},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (repo *Memory) SetNodesAt(names [][]byte, heights []int32) error {
|
||
|
|
|
||
|
|
for i, height := range heights {
|
||
|
|
c, ok := repo.cache[height]
|
||
|
|
if !ok {
|
||
|
|
c = map[string]bool{}
|
||
|
|
repo.cache[height] = c
|
||
|
|
}
|
||
|
|
name := string(names[i])
|
||
|
|
c[name] = true
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (repo *Memory) NodesAt(height int32) ([][]byte, error) {
|
||
|
|
|
||
|
|
var names [][]byte
|
||
|
|
|
||
|
|
for name := range repo.cache[height] {
|
||
|
|
names = append(names, []byte(name))
|
||
|
|
}
|
||
|
|
|
||
|
|
return names, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (repo *Memory) Close() error {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (repo *Memory) Flush() error {
|
||
|
|
return nil
|
||
|
|
}
|