Merge pull request #50 from codecrafters-io/fix-header-handling

use actual kv pair count
codecrafters-io · Nov 21, 2023 · 39562fb · 39562fb
2 parents d941063 + 64b36d4
commit 39562fb
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/internal/rdb_file_creator.go b/internal/rdb_file_creator.go
@@ -70,7 +70,15 @@ func (r *RDBFileCreator) Write(keyValuePairs []KeyValuePair) error {
 		}
 	}
 
-	if err := enc.WriteDBHeader(0, 5, 1); err != nil {
+	keysWithTtlCount := 0
+
+	for _, keyValuePair := range keyValuePairs {
+		if keyValuePair.expiryTs > 0 {
+			keysWithTtlCount++
+		}
+	}
+
+	if err := enc.WriteDBHeader(0, uint64(len(keyValuePairs)), uint64(keysWithTtlCount)); err != nil {
 		return err
 	}