Add SstPartitioner::Context::target_output_file_size

chejinge · May 7, 2023 · e1bb138 · e1bb138
1 parent e508e6e
commit e1bb138
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/db/compaction/compaction.cc b/db/compaction/compaction.cc
@@ -723,6 +723,7 @@ std::unique_ptr<SstPartitioner> Compaction::CreateSstPartitioner() const {
   context.output_level = output_level_;
   context.smallest_user_key = smallest_user_key_;
   context.largest_user_key = largest_user_key_;
+  context.target_output_file_size = target_output_file_size_;
   return immutable_options_.sst_partitioner_factory->CreatePartitioner(context);
 }
 

diff --git a/db/db_impl/db_impl_compaction_flush.cc b/db/db_impl/db_impl_compaction_flush.cc
@@ -1100,6 +1100,7 @@ Status DBImpl::CompactRangeInternal(const CompactRangeOptions& options,
         // Small lies about compaction range
         context.smallest_user_key = *begin;
         context.largest_user_key = *end;
+        context.target_output_file_size = 0;
         partitioner = partitioner_factory->CreatePartitioner(context);
       }
 

diff --git a/include/rocksdb/sst_partitioner.h b/include/rocksdb/sst_partitioner.h
@@ -75,6 +75,7 @@ class SstPartitioner {
     Slice smallest_user_key;
     // Largest key for compaction
     Slice largest_user_key;
+    size_t target_output_file_size;
   };
 };