From d95a8d022bbdf6ae0dc8ca434d9b3673189551f8 Mon Sep 17 00:00:00 2001
From: Li0k <yuli@singularity-data.com>
Date: Thu, 16 Feb 2023 13:57:15 +0800
Subject: [PATCH] fix(storage): fix comments

---
 .../picker/space_reclaim_compaction_picker.rs       | 13 +++++++++----
 .../picker/ttl_reclaim_compaction_picker.rs         | 13 +++++++++----
 2 files changed, 18 insertions(+), 8 deletions(-)

diff --git a/src/meta/src/hummock/compaction/picker/space_reclaim_compaction_picker.rs b/src/meta/src/hummock/compaction/picker/space_reclaim_compaction_picker.rs
index 406a7e7017be6..4e2f1c30b2261 100644
--- a/src/meta/src/hummock/compaction/picker/space_reclaim_compaction_picker.rs
+++ b/src/meta/src/hummock/compaction/picker/space_reclaim_compaction_picker.rs
@@ -108,26 +108,27 @@ impl SpaceReclaimCompactionPicker {
             state.init(start_key, end_key)
         }
 
-        let start_key = state.last_select_end_key.clone();
         let mut select_file_size = 0;
 
         let matched_sst = reclaimed_level
             .table_infos
             .iter()
             .filter(|sst| match &sst.key_range {
-                Some(key_range) => key_range.left >= start_key,
+                Some(key_range) => key_range.left >= state.last_select_end_key,
 
                 None => false,
             });
 
         for sst in matched_sst {
-            state.last_select_end_key = sst.key_range.as_ref().unwrap().right.to_vec();
-
             if level_handler.is_pending_compact(&sst.id) || self.filter(sst) {
                 if !select_input_ssts.is_empty() {
                     // Our goal is to pick as many complete layers of data as possible and keep the
                     // picked files contiguous to avoid overlapping key_ranges, so the strategy is
                     // to pick as many contiguous files as possible (at least one)
+                    state.last_select_end_key.clear();
+                    state
+                        .last_select_end_key
+                        .extend_from_slice(sst.key_range.as_ref().unwrap().right.as_ref());
                     break;
                 }
 
@@ -137,6 +138,10 @@ impl SpaceReclaimCompactionPicker {
             select_input_ssts.push(sst.clone());
             select_file_size += sst.file_size;
             if select_file_size > self.max_space_reclaim_bytes {
+                state.last_select_end_key.clear();
+                state
+                    .last_select_end_key
+                    .extend_from_slice(sst.key_range.as_ref().unwrap().right.as_ref());
                 break;
             }
         }
diff --git a/src/meta/src/hummock/compaction/picker/ttl_reclaim_compaction_picker.rs b/src/meta/src/hummock/compaction/picker/ttl_reclaim_compaction_picker.rs
index 66b6b2a7befc8..35660e1b9356c 100644
--- a/src/meta/src/hummock/compaction/picker/ttl_reclaim_compaction_picker.rs
+++ b/src/meta/src/hummock/compaction/picker/ttl_reclaim_compaction_picker.rs
@@ -142,25 +142,26 @@ impl TtlReclaimCompactionPicker {
         }
 
         let current_epoch_time = Epoch::now().0;
-        let start_key = state.last_select_end_key.clone();
         let mut select_file_size = 0;
         let matched_sst = reclaimed_level
             .table_infos
             .iter()
             .filter(|sst| match &sst.key_range {
-                Some(key_range) => key_range.left >= start_key,
+                Some(key_range) => key_range.left >= state.last_select_end_key,
 
                 None => false,
             });
 
         for sst in matched_sst {
-            state.last_select_end_key = sst.key_range.as_ref().unwrap().right.to_vec();
-
             if level_handler.is_pending_compact(&sst.id) || self.filter(sst, current_epoch_time) {
                 if !select_input_ssts.is_empty() {
                     // Our goal is to pick as many complete layers of data as possible and keep the
                     // picked files contiguous to avoid overlapping key_ranges, so the strategy is
                     // to pick as many contiguous files as possible (at least one)
+                    state.last_select_end_key.clear();
+                    state
+                        .last_select_end_key
+                        .extend_from_slice(sst.key_range.as_ref().unwrap().right.as_ref());
                     break;
                 }
 
@@ -170,6 +171,10 @@ impl TtlReclaimCompactionPicker {
             select_input_ssts.push(sst.clone());
             select_file_size += sst.file_size;
             if select_file_size > self.max_ttl_reclaim_bytes {
+                state.last_select_end_key.clear();
+                state
+                    .last_select_end_key
+                    .extend_from_slice(sst.key_range.as_ref().unwrap().right.as_ref());
                 break;
             }
         }