mirror of
https://github.com/ipfs/kubo.git
synced 2026-02-22 10:57:42 +08:00
bf22aeec0a
- Modified Godeps/Godeps.json by hand - [TEST] Updated welcome docs hash to sharness - [TEST] Updated contact doc - [TEST] disabled breaking test (t0080-repo refs local)
65 lines
1.2 KiB
Go
65 lines
1.2 KiB
Go
// package set contains various different types of 'BlockSet's
|
|
package set
|
|
|
|
import (
|
|
"github.com/ipfs/go-ipfs/blocks/bloom"
|
|
"github.com/ipfs/go-ipfs/util"
|
|
)
|
|
|
|
var log = util.Logger("blockset")
|
|
|
|
// BlockSet represents a mutable set of keyed blocks
|
|
type BlockSet interface {
|
|
AddBlock(util.Key)
|
|
RemoveBlock(util.Key)
|
|
HasKey(util.Key) bool
|
|
GetBloomFilter() bloom.Filter
|
|
|
|
GetKeys() []util.Key
|
|
}
|
|
|
|
func SimpleSetFromKeys(keys []util.Key) BlockSet {
|
|
sbs := &simpleBlockSet{blocks: make(map[util.Key]struct{})}
|
|
for _, k := range keys {
|
|
sbs.blocks[k] = struct{}{}
|
|
}
|
|
return sbs
|
|
}
|
|
|
|
func NewSimpleBlockSet() BlockSet {
|
|
return &simpleBlockSet{blocks: make(map[util.Key]struct{})}
|
|
}
|
|
|
|
type simpleBlockSet struct {
|
|
blocks map[util.Key]struct{}
|
|
}
|
|
|
|
func (b *simpleBlockSet) AddBlock(k util.Key) {
|
|
b.blocks[k] = struct{}{}
|
|
}
|
|
|
|
func (b *simpleBlockSet) RemoveBlock(k util.Key) {
|
|
delete(b.blocks, k)
|
|
}
|
|
|
|
func (b *simpleBlockSet) HasKey(k util.Key) bool {
|
|
_, has := b.blocks[k]
|
|
return has
|
|
}
|
|
|
|
func (b *simpleBlockSet) GetBloomFilter() bloom.Filter {
|
|
f := bloom.BasicFilter()
|
|
for k := range b.blocks {
|
|
f.Add([]byte(k))
|
|
}
|
|
return f
|
|
}
|
|
|
|
func (b *simpleBlockSet) GetKeys() []util.Key {
|
|
var out []util.Key
|
|
for k := range b.blocks {
|
|
out = append(out, k)
|
|
}
|
|
return out
|
|
}
|