refactor: remove double checks of memtable size (#4117)

* refactor: remove unnecessary unwrap * Update src/mito2/src/region/version.rs Co-authored-by: dennis zhuang <[email protected]> --------- Co-authored-by: Kermit <[email protected]> Co-authored-by: dennis zhuang <[email protected]>
GreptimeTeam · Jun 7, 2024 · d8b51cf · d8b51cf
1 parent e142ca4
commit d8b51cf
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/src/mito2/src/region/version.rs b/src/mito2/src/region/version.rs
@@ -80,14 +80,11 @@ impl VersionControl {
     /// Freezes the mutable memtable if it is not empty.
     pub(crate) fn freeze_mutable(&self) -> Result<()> {
         let version = self.current().version;
-        if version.memtables.mutable.is_empty() {
+
+        let Some(new_memtables) = version.memtables.freeze_mutable(&version.metadata)? else {
             return Ok(());
-        }
-        // Safety: Immutable memtable is None.
-        let new_memtables = version
-            .memtables
-            .freeze_mutable(&version.metadata)?
-            .unwrap();
+        };
+
         // Create a new version with memtable switched.
         let new_version = Arc::new(
             VersionBuilder::from_version(version)