flush range on first only

paritytech · Apr 15, 2024 · 25b1c1b · 25b1c1b
1 parent 24ba453
commit 25b1c1b
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 8 deletions.
diff --git a/src/index.rs b/src/index.rs
@@ -671,12 +671,16 @@ impl IndexTable {
 	}
 
 	pub fn flush(&self) -> Result<()> {
-		// Flush everything except stats.
-		let mut start = META_SIZE;
 		let maps = self.map.read();
+		let mut first = true;
 		for map in maps.iter() {
-			try_io!(map.flush_range(start, map.len() - start)); // TODO not flush range when start 0
-			start = 0;
+			if first {
+				first = false;
+				// Flush everything except stats.
+				try_io!(map.flush_range(META_SIZE, map.len() - META_SIZE));
+			} else {
+				try_io!(map.flush());
+			}
 		}
 		Ok(())
 	}

diff --git a/src/ref_count.rs b/src/ref_count.rs
@@ -497,12 +497,16 @@ impl RefCountTable {
 	}
 
 	pub fn flush(&self) -> Result<()> {
-		// Flush everything except stats.
-		let mut start = META_SIZE;
 		let maps = self.map.read();
+		let mut first = true;
 		for map in maps.iter() {
-			try_io!(map.flush_range(start, map.len() - start)); // TODO not flush range when start 0
-			start = 0;
+			if first {
+				first = false;
+				// Flush everything except stats.
+				try_io!(map.flush_range(META_SIZE, map.len() - META_SIZE));
+			} else {
+				try_io!(map.flush());
+			}
 		}
 		Ok(())
 	}