fix: check mem != nil before calling Mem() (tikv#1236)

Signed-off-by: ekexium <[email protected]>
pingyu · Mar 18, 2024 · a9128e8 · a9128e8
1 parent 73c0712
commit a9128e8
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/internal/unionstore/pipelined_memdb.go b/internal/unionstore/pipelined_memdb.go
@@ -411,7 +411,14 @@ func (p *PipelinedMemDB) onMemChange() {
 
 // Mem returns the memory usage of MemBuffer.
 func (p *PipelinedMemDB) Mem() uint64 {
-	return p.memDB.Mem() + p.flushingMemDB.Mem()
+	var mem uint64
+	if p.memDB != nil {
+		mem += p.memDB.Mem()
+	}
+	if p.flushingMemDB != nil {
+		mem += p.flushingMemDB.Mem()
+	}
+	return mem
 }
 
 type errIterator struct {