replace get_mut_unchecked with make_mut for SortKeyBatchRanges

Signed-off-by: zenghua <[email protected]>
lakesoul-io · Nov 1, 2023 · 467d4a2 · 467d4a2
1 parent 0ae33b7
commit 467d4a2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/rust/lakesoul-io/src/sorted_merge/combiner.rs b/rust/lakesoul-io/src/sorted_merge/combiner.rs
@@ -200,7 +200,7 @@ impl MinHeapSortKeyBatchRangeCombiner {
     }
 
     fn get_mut_current_sort_key_range(&mut self) -> &mut SortKeyBatchRanges {
-        Arc::get_mut(&mut self.current_sort_key_range).unwrap()
+        Arc::make_mut(&mut self.current_sort_key_range)
     }
 }
 

diff --git a/rust/lakesoul-io/src/sorted_merge/sort_key_range.rs b/rust/lakesoul-io/src/sorted_merge/sort_key_range.rs
@@ -188,7 +188,7 @@ impl Clone for SortKeyArrayRange {
 }
 
 // Multiple ranges with same sorted primary key from variant source record_batch. These ranges will be merged into ONE row of target record_batch finnally.
-#[derive(Debug)]
+#[derive(Debug, Clone)]
 pub struct SortKeyBatchRanges {
     // vector with length=column_num that holds a Vector of SortKeyArrayRange to be merged for each column
     pub(crate) sort_key_array_ranges: Vec<SmallVec<[SortKeyArrayRange; 4]>>,
-Original file line number
+Diff line change
@@ Expand Up / @@ -200,7 +200,7 @@ impl MinHeapSortKeyBatchRangeCombiner { @@
         }
         fn get_mut_current_sort_key_range(&mut self) -> &mut SortKeyBatchRanges {
-            Arc::get_mut(&mut self.current_sort_key_range).unwrap()
+            Arc::make_mut(&mut self.current_sort_key_range)
         }
     }
@@ Expand Down @@