mirror of
https://github.com/charlienet/go-mixed.git
synced 2025-07-18 00:22:41 +08:00
38 lines
556 B
Go
38 lines
556 B
Go
package bloom
|
|
|
|
import "github.com/bits-and-blooms/bitset"
|
|
|
|
type memStore struct {
|
|
size uint
|
|
set *bitset.BitSet // 内存位图
|
|
}
|
|
|
|
func newMemStore(size uint) *memStore {
|
|
return &memStore{
|
|
size: size,
|
|
set: bitset.New(size),
|
|
}
|
|
}
|
|
|
|
func (s *memStore) Clear() {
|
|
s.set.ClearAll()
|
|
}
|
|
|
|
func (s *memStore) Set(offsets ...uint) error {
|
|
for _, p := range offsets {
|
|
s.set.Set(p)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *memStore) Test(offsets ...uint) (bool, error) {
|
|
for _, p := range offsets {
|
|
if !s.set.Test(p) {
|
|
return false, nil
|
|
}
|
|
}
|
|
|
|
return true, nil
|
|
}
|