添加存储

bloom/mem_store.go

@@ -0,0 +1,37 @@
+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
+}