z: Add TotalSize method on bloom filter (#197)

This PR adds a TotalSize function which returns the total size of the bloom filter.
dgraph-io · Sep 30, 2020 · e1609c8 · e1609c8
1 parent 646c5f3
commit e1609c8
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/z/bbloom.go b/z/bbloom.go
@@ -131,6 +131,13 @@ func (bl *Bloom) AddIfNotHas(hash uint64) bool {
 	return true
 }
 
+// TotalSize returns the total size of the bloom filter.
+func (bl *Bloom) TotalSize() int {
+	// The bl struct has 5 members and each one is 8 byte. The bitset is a
+	// uint64 byte slice.
+	return len(bl.bitset)*8 + 5*8
+}
+
 // Size makes Bloom filter with as bitset of size sz.
 func (bl *Bloom) Size(sz uint64) {
 	bl.bitset = make([]uint64, sz>>6)