Sarthak | Memtable now uses InMemoryMap instead of Node

SarthakMakhija · Apr 10, 2023 · 7bfb203 · 7bfb203
1 parent 7399782
commit 7bfb203
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/storage/memory/MemTable.go b/storage/memory/MemTable.go
@@ -7,7 +7,8 @@ import (
 )
 
 type MemTable struct {
-	head           *Node
+	//head           *Node
+	inMemoryMap    *InMemoryMap
 	size           uint64
 	totalKeys      int
 	keyComparator  comparator.KeyComparator
@@ -16,15 +17,16 @@ type MemTable struct {
 
 func NewMemTable(maxLevel int, keyComparator comparator.KeyComparator) *MemTable {
 	return &MemTable{
-		head:           NewNode(model.NilSlice(), model.NilSlice(), maxLevel),
+		//head:           NewNode(model.NilSlice(), model.NilSlice(), maxLevel),
+		inMemoryMap:    NewInMemoryMap(),
 		size:           0,
 		keyComparator:  keyComparator,
 		levelGenerator: utils.NewLevelGenerator(maxLevel),
 	}
 }
 
 func (memTable *MemTable) Put(key, value model.Slice) bool {
-	if ok := memTable.head.Put(key, value, memTable.keyComparator, memTable.levelGenerator); ok {
+	if ok := memTable.inMemoryMap.Put(key, value); ok {
 		memTable.size = memTable.size + uint64(key.Size()) + uint64(value.Size())
 		memTable.totalKeys = memTable.totalKeys + 1
 		return ok
@@ -33,15 +35,15 @@ func (memTable *MemTable) Put(key, value model.Slice) bool {
 }
 
 func (memTable *MemTable) Get(key model.Slice) model.GetResult {
-	return memTable.head.Get(key, memTable.keyComparator)
+	return memTable.inMemoryMap.Get(key)
 }
 
 func (memTable *MemTable) MultiGet(keys []model.Slice) (model.MultiGetResult, []model.Slice) {
-	return memTable.head.MultiGet(keys, memTable.keyComparator)
+	return memTable.inMemoryMap.MultiGet(keys)
 }
 
 func (memTable *MemTable) AllKeyValues() []model.KeyValuePair {
-	return memTable.head.AllKeyValues()
+	return memTable.inMemoryMap.AllKeyValues(memTable.keyComparator)
 }
 
 func (memTable *MemTable) TotalSize() uint64 {