mirror of
https://github.com/ipfs/kubo.git
synced 2026-02-25 20:37:53 +08:00
3b6216b239
This has required changing the order of some parameters and adding HashOnRead to the Blockstore interface (which I have in turn added to all the wrapper implementations). License: MIT Signed-off-by: Hector Sanjuan <hector@protocol.ai>
69 lines
1.6 KiB
Go
69 lines
1.6 KiB
Go
// Package set defines the BlockSet interface which provides
|
|
// abstraction for sets of Cids.
|
|
// It provides a default implementation using cid.Set.
|
|
package set
|
|
|
|
import (
|
|
logging "gx/ipfs/QmSpJByNKFX1sCsHBEp3R73FL4NF6FnQTEGyNAXHm2GS52/go-log"
|
|
cid "gx/ipfs/QmV5gPoRsjN1Gid3LMdNZTyfCtP2DsvqEbMAmz82RmmiGk/go-cid"
|
|
|
|
"github.com/ipfs/go-ipfs/blocks/bloom"
|
|
)
|
|
|
|
var log = logging.Logger("blockset")
|
|
|
|
// BlockSet represents a mutable set of blocks CIDs.
|
|
type BlockSet interface {
|
|
AddBlock(*cid.Cid)
|
|
RemoveBlock(*cid.Cid)
|
|
HasKey(*cid.Cid) bool
|
|
// GetBloomFilter creates and returns a bloom filter to which
|
|
// all the CIDs in the set have been added.
|
|
GetBloomFilter() bloom.Filter
|
|
GetKeys() []*cid.Cid
|
|
}
|
|
|
|
// SimpleSetFromKeys returns a default implementation of BlockSet
|
|
// using cid.Set. The given keys are added to the set.
|
|
func SimpleSetFromKeys(keys []*cid.Cid) BlockSet {
|
|
sbs := &simpleBlockSet{blocks: cid.NewSet()}
|
|
for _, k := range keys {
|
|
sbs.AddBlock(k)
|
|
}
|
|
return sbs
|
|
}
|
|
|
|
// NewSimpleBlockSet returns a new empty default implementation
|
|
// of BlockSet using cid.Set.
|
|
func NewSimpleBlockSet() BlockSet {
|
|
return &simpleBlockSet{blocks: cid.NewSet()}
|
|
}
|
|
|
|
type simpleBlockSet struct {
|
|
blocks *cid.Set
|
|
}
|
|
|
|
func (b *simpleBlockSet) AddBlock(k *cid.Cid) {
|
|
b.blocks.Add(k)
|
|
}
|
|
|
|
func (b *simpleBlockSet) RemoveBlock(k *cid.Cid) {
|
|
b.blocks.Remove(k)
|
|
}
|
|
|
|
func (b *simpleBlockSet) HasKey(k *cid.Cid) bool {
|
|
return b.blocks.Has(k)
|
|
}
|
|
|
|
func (b *simpleBlockSet) GetBloomFilter() bloom.Filter {
|
|
f := bloom.BasicFilter()
|
|
for _, k := range b.blocks.Keys() {
|
|
f.Add(k.Bytes())
|
|
}
|
|
return f
|
|
}
|
|
|
|
func (b *simpleBlockSet) GetKeys() []*cid.Cid {
|
|
return b.blocks.Keys()
|
|
}
|