66 lines
1.3 KiB
Go
66 lines
1.3 KiB
Go
package splitstore
|
|
|
|
import (
|
|
"math/rand"
|
|
|
|
"golang.org/x/xerrors"
|
|
|
|
bbloom "github.com/ipfs/bbloom"
|
|
cid "github.com/ipfs/go-cid"
|
|
)
|
|
|
|
type BloomLiveSetEnv struct{}
|
|
|
|
var _ LiveSetEnv = (*BloomLiveSetEnv)(nil)
|
|
|
|
type BloomLiveSet struct {
|
|
salt []byte
|
|
bf *bbloom.Bloom
|
|
}
|
|
|
|
var _ LiveSet = (*BloomLiveSet)(nil)
|
|
|
|
func NewBloomLiveSetEnv() (*BloomLiveSetEnv, error) {
|
|
return &BloomLiveSetEnv{}, nil
|
|
}
|
|
|
|
func (e *BloomLiveSetEnv) NewLiveSet(name string) (LiveSet, error) {
|
|
salt := make([]byte, 4)
|
|
_, err := rand.Read(salt)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("error reading salt: %w", err)
|
|
}
|
|
|
|
bf, err := bbloom.New(float64(10_000_000), float64(0.01))
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("error creating bloom filter: %w", err)
|
|
}
|
|
|
|
return &BloomLiveSet{salt: salt, bf: bf}, nil
|
|
}
|
|
|
|
func (e *BloomLiveSetEnv) Close() error {
|
|
return nil
|
|
}
|
|
|
|
func (s *BloomLiveSet) saltedKey(cid cid.Cid) []byte {
|
|
hash := cid.Hash()
|
|
key := make([]byte, len(s.salt)+len(hash))
|
|
n := copy(key, s.salt)
|
|
copy(key[n:], hash)
|
|
return key
|
|
}
|
|
|
|
func (s *BloomLiveSet) Mark(cid cid.Cid) error {
|
|
s.bf.Add(s.saltedKey(cid))
|
|
return nil
|
|
}
|
|
|
|
func (s *BloomLiveSet) Has(cid cid.Cid) (bool, error) {
|
|
return s.bf.Has(s.saltedKey(cid)), nil
|
|
}
|
|
|
|
func (s *BloomLiveSet) Close() error {
|
|
return nil
|
|
}
|