@ -46,6 +46,7 @@
# include "monitoring/iostats_context_imp.h"
# include "monitoring/iostats_context_imp.h"
# include "monitoring/perf_context_imp.h"
# include "monitoring/perf_context_imp.h"
# include "monitoring/thread_status_util.h"
# include "monitoring/thread_status_util.h"
# include "options/configurable_helper.h"
# include "options/options_helper.h"
# include "options/options_helper.h"
# include "port/port.h"
# include "port/port.h"
# include "rocksdb/db.h"
# include "rocksdb/db.h"
@ -54,6 +55,7 @@
# include "rocksdb/statistics.h"
# include "rocksdb/statistics.h"
# include "rocksdb/status.h"
# include "rocksdb/status.h"
# include "rocksdb/table.h"
# include "rocksdb/table.h"
# include "rocksdb/utilities/options_type.h"
# include "table/block_based/block.h"
# include "table/block_based/block.h"
# include "table/block_based/block_based_table_factory.h"
# include "table/block_based/block_based_table_factory.h"
# include "table/merging_iterator.h"
# include "table/merging_iterator.h"
@ -1939,6 +1941,7 @@ std::string CompactionJob::GetTableFileName(uint64_t file_number) {
file_number , compact_ - > compaction - > output_path_id ( ) ) ;
file_number , compact_ - > compaction - > output_path_id ( ) ) ;
}
}
# ifndef ROCKSDB_LITE
std : : string CompactionServiceCompactionJob : : GetTableFileName (
std : : string CompactionServiceCompactionJob : : GetTableFileName (
uint64_t file_number ) {
uint64_t file_number ) {
return MakeTableFileName ( output_path_ , file_number ) ;
return MakeTableFileName ( output_path_ , file_number ) ;
@ -1984,8 +1987,11 @@ Status CompactionServiceCompactionJob::Run() {
c - > column_family_data ( ) - > CalculateSSTWriteHint ( c - > output_level ( ) ) ;
c - > column_family_data ( ) - > CalculateSSTWriteHint ( c - > output_level ( ) ) ;
bottommost_level_ = c - > bottommost_level ( ) ;
bottommost_level_ = c - > bottommost_level ( ) ;
compact_ - > sub_compact_states . emplace_back ( c , compaction_input_ . begin ,
Slice begin = compaction_input_ . begin ;
compaction_input_ . end ,
Slice end = compaction_input_ . end ;
compact_ - > sub_compact_states . emplace_back (
c , compaction_input_ . has_begin ? & begin : nullptr ,
compaction_input_ . has_end ? & end : nullptr ,
compaction_input_ . approx_size ) ;
compaction_input_ . approx_size ) ;
log_buffer_ - > FlushBufferToLog ( ) ;
log_buffer_ - > FlushBufferToLog ( ) ;
@ -2062,4 +2068,412 @@ void CompactionServiceCompactionJob::CleanupCompaction() {
CompactionJob : : CleanupCompaction ( ) ;
CompactionJob : : CleanupCompaction ( ) ;
}
}
// Internal binary format for the input and result data
enum BinaryFormatVersion : uint32_t {
kOptionsString = 1 , // Use string format similar to Option string format
} ;
// offset_of is used to get the offset of a class data member
// ex: offset_of(&ColumnFamilyDescriptor::options)
// This call will return the offset of options in ColumnFamilyDescriptor class
//
// This is the same as offsetof() but allow us to work with non standard-layout
// classes and structures
// refs:
// http://en.cppreference.com/w/cpp/concept/StandardLayoutType
// https://gist.github.com/graphitemaster/494f21190bb2c63c5516
static ColumnFamilyDescriptor dummy_cfd ( " " , ColumnFamilyOptions ( ) ) ;
template < typename T1 >
int offset_of ( T1 ColumnFamilyDescriptor : : * member ) {
return int ( size_t ( & ( dummy_cfd . * member ) ) - size_t ( & dummy_cfd ) ) ;
}
static CompactionServiceInput dummy_cs_input ;
template < typename T1 >
int offset_of ( T1 CompactionServiceInput : : * member ) {
return int ( size_t ( & ( dummy_cs_input . * member ) ) - size_t ( & dummy_cs_input ) ) ;
}
static std : : unordered_map < std : : string , OptionTypeInfo > cfd_type_info = {
{ " name " ,
{ offset_of ( & ColumnFamilyDescriptor : : name ) , OptionType : : kEncodedString ,
OptionVerificationType : : kNormal , OptionTypeFlags : : kNone } } ,
{ " options " ,
{ offset_of ( & ColumnFamilyDescriptor : : options ) , OptionType : : kConfigurable ,
OptionVerificationType : : kNormal , OptionTypeFlags : : kNone ,
[ ] ( const ConfigOptions & opts , const std : : string & /*name*/ ,
const std : : string & value , char * addr ) {
auto cf_options = reinterpret_cast < ColumnFamilyOptions * > ( addr ) ;
return GetColumnFamilyOptionsFromString ( opts , ColumnFamilyOptions ( ) ,
value , cf_options ) ;
} ,
[ ] ( const ConfigOptions & opts , const std : : string & /*name*/ ,
const char * addr , std : : string * value ) {
const auto cf_options =
reinterpret_cast < const ColumnFamilyOptions * > ( addr ) ;
std : : string result ;
auto status =
GetStringFromColumnFamilyOptions ( opts , * cf_options , & result ) ;
* value = " { " + result + " } " ;
return status ;
} ,
[ ] ( const ConfigOptions & opts , const std : : string & name , const char * addr1 ,
const char * addr2 , std : : string * mismatch ) {
const auto this_one =
reinterpret_cast < const ColumnFamilyOptions * > ( addr1 ) ;
const auto that_one =
reinterpret_cast < const ColumnFamilyOptions * > ( addr2 ) ;
auto this_conf = CFOptionsAsConfigurable ( * this_one ) ;
auto that_conf = CFOptionsAsConfigurable ( * that_one ) ;
std : : string mismatch_opt ;
bool result =
this_conf - > AreEquivalent ( opts , that_conf . get ( ) , & mismatch_opt ) ;
if ( ! result ) {
* mismatch = name + " . " + mismatch_opt ;
}
return result ;
} } } ,
} ;
static std : : unordered_map < std : : string , OptionTypeInfo > cs_input_type_info = {
{ " column_family " ,
OptionTypeInfo : : Struct ( " column_family " , & cfd_type_info ,
offset_of ( & CompactionServiceInput : : column_family ) ,
OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone ) } ,
{ " db_options " ,
{ offset_of ( & CompactionServiceInput : : db_options ) , OptionType : : kConfigurable ,
OptionVerificationType : : kNormal , OptionTypeFlags : : kNone ,
[ ] ( const ConfigOptions & opts , const std : : string & /*name*/ ,
const std : : string & value , char * addr ) {
auto options = reinterpret_cast < DBOptions * > ( addr ) ;
return GetDBOptionsFromString ( opts , DBOptions ( ) , value , options ) ;
} ,
[ ] ( const ConfigOptions & opts , const std : : string & /*name*/ ,
const char * addr , std : : string * value ) {
const auto options = reinterpret_cast < const DBOptions * > ( addr ) ;
std : : string result ;
auto status = GetStringFromDBOptions ( opts , * options , & result ) ;
* value = " { " + result + " } " ;
return status ;
} ,
[ ] ( const ConfigOptions & opts , const std : : string & name , const char * addr1 ,
const char * addr2 , std : : string * mismatch ) {
const auto this_one = reinterpret_cast < const DBOptions * > ( addr1 ) ;
const auto that_one = reinterpret_cast < const DBOptions * > ( addr2 ) ;
auto this_conf = DBOptionsAsConfigurable ( * this_one ) ;
auto that_conf = DBOptionsAsConfigurable ( * that_one ) ;
std : : string mismatch_opt ;
bool result =
this_conf - > AreEquivalent ( opts , that_conf . get ( ) , & mismatch_opt ) ;
if ( ! result ) {
* mismatch = name + " . " + mismatch_opt ;
}
return result ;
} } } ,
{ " snapshots " , OptionTypeInfo : : Vector < uint64_t > (
offset_of ( & CompactionServiceInput : : snapshots ) ,
OptionVerificationType : : kNormal , OptionTypeFlags : : kNone ,
{ 0 , OptionType : : kUInt64T } ) } ,
{ " input_files " , OptionTypeInfo : : Vector < std : : string > (
offset_of ( & CompactionServiceInput : : input_files ) ,
OptionVerificationType : : kNormal , OptionTypeFlags : : kNone ,
{ 0 , OptionType : : kEncodedString } ) } ,
{ " output_level " ,
{ offset_of ( & CompactionServiceInput : : output_level ) , OptionType : : kInt ,
OptionVerificationType : : kNormal , OptionTypeFlags : : kNone } } ,
{ " has_begin " ,
{ offset_of ( & CompactionServiceInput : : has_begin ) , OptionType : : kBoolean ,
OptionVerificationType : : kNormal , OptionTypeFlags : : kNone } } ,
{ " begin " ,
{ offset_of ( & CompactionServiceInput : : begin ) , OptionType : : kEncodedString ,
OptionVerificationType : : kNormal , OptionTypeFlags : : kNone } } ,
{ " has_end " ,
{ offset_of ( & CompactionServiceInput : : has_end ) , OptionType : : kBoolean ,
OptionVerificationType : : kNormal , OptionTypeFlags : : kNone } } ,
{ " end " ,
{ offset_of ( & CompactionServiceInput : : end ) , OptionType : : kEncodedString ,
OptionVerificationType : : kNormal , OptionTypeFlags : : kNone } } ,
{ " approx_size " ,
{ offset_of ( & CompactionServiceInput : : approx_size ) , OptionType : : kUInt64T ,
OptionVerificationType : : kNormal , OptionTypeFlags : : kNone } } ,
} ;
static std : : unordered_map < std : : string , OptionTypeInfo >
cs_output_file_type_info = {
{ " file_name " ,
{ offsetof ( struct CompactionServiceOutputFile , file_name ) ,
OptionType : : kEncodedString , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " smallest_seqno " ,
{ offsetof ( struct CompactionServiceOutputFile , smallest_seqno ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " largest_seqno " ,
{ offsetof ( struct CompactionServiceOutputFile , largest_seqno ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " smallest_internal_key " ,
{ offsetof ( struct CompactionServiceOutputFile , smallest_internal_key ) ,
OptionType : : kEncodedString , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " largest_internal_key " ,
{ offsetof ( struct CompactionServiceOutputFile , largest_internal_key ) ,
OptionType : : kEncodedString , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " oldest_ancester_time " ,
{ offsetof ( struct CompactionServiceOutputFile , oldest_ancester_time ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " file_creation_time " ,
{ offsetof ( struct CompactionServiceOutputFile , file_creation_time ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " paranoid_hash " ,
{ offsetof ( struct CompactionServiceOutputFile , paranoid_hash ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " marked_for_compaction " ,
{ offsetof ( struct CompactionServiceOutputFile , marked_for_compaction ) ,
OptionType : : kBoolean , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
} ;
static std : : unordered_map < std : : string , OptionTypeInfo >
compaction_job_stats_type_info = {
{ " elapsed_micros " ,
{ offsetof ( struct CompactionJobStats , elapsed_micros ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " cpu_micros " ,
{ offsetof ( struct CompactionJobStats , cpu_micros ) , OptionType : : kUInt64T ,
OptionVerificationType : : kNormal , OptionTypeFlags : : kNone } } ,
{ " num_input_records " ,
{ offsetof ( struct CompactionJobStats , num_input_records ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " num_blobs_read " ,
{ offsetof ( struct CompactionJobStats , num_blobs_read ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " num_input_files " ,
{ offsetof ( struct CompactionJobStats , num_input_files ) ,
OptionType : : kSizeT , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " num_input_files_at_output_level " ,
{ offsetof ( struct CompactionJobStats , num_input_files_at_output_level ) ,
OptionType : : kSizeT , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " num_output_records " ,
{ offsetof ( struct CompactionJobStats , num_output_records ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " num_output_files " ,
{ offsetof ( struct CompactionJobStats , num_output_files ) ,
OptionType : : kSizeT , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " num_output_files_blob " ,
{ offsetof ( struct CompactionJobStats , num_output_files_blob ) ,
OptionType : : kSizeT , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " is_full_compaction " ,
{ offsetof ( struct CompactionJobStats , is_full_compaction ) ,
OptionType : : kBoolean , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " is_manual_compaction " ,
{ offsetof ( struct CompactionJobStats , is_manual_compaction ) ,
OptionType : : kBoolean , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " total_input_bytes " ,
{ offsetof ( struct CompactionJobStats , total_input_bytes ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " total_blob_bytes_read " ,
{ offsetof ( struct CompactionJobStats , total_blob_bytes_read ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " total_output_bytes " ,
{ offsetof ( struct CompactionJobStats , total_output_bytes ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " total_output_bytes_blob " ,
{ offsetof ( struct CompactionJobStats , total_output_bytes_blob ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " num_records_replaced " ,
{ offsetof ( struct CompactionJobStats , num_records_replaced ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " total_input_raw_key_bytes " ,
{ offsetof ( struct CompactionJobStats , total_input_raw_key_bytes ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " total_input_raw_value_bytes " ,
{ offsetof ( struct CompactionJobStats , total_input_raw_value_bytes ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " num_input_deletion_records " ,
{ offsetof ( struct CompactionJobStats , num_input_deletion_records ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " num_expired_deletion_records " ,
{ offsetof ( struct CompactionJobStats , num_expired_deletion_records ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " num_corrupt_keys " ,
{ offsetof ( struct CompactionJobStats , num_corrupt_keys ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " file_write_nanos " ,
{ offsetof ( struct CompactionJobStats , file_write_nanos ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " file_range_sync_nanos " ,
{ offsetof ( struct CompactionJobStats , file_range_sync_nanos ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " file_fsync_nanos " ,
{ offsetof ( struct CompactionJobStats , file_fsync_nanos ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " file_prepare_write_nanos " ,
{ offsetof ( struct CompactionJobStats , file_prepare_write_nanos ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " smallest_output_key_prefix " ,
{ offsetof ( struct CompactionJobStats , smallest_output_key_prefix ) ,
OptionType : : kEncodedString , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " largest_output_key_prefix " ,
{ offsetof ( struct CompactionJobStats , largest_output_key_prefix ) ,
OptionType : : kEncodedString , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " num_single_del_fallthru " ,
{ offsetof ( struct CompactionJobStats , num_single_del_fallthru ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " num_single_del_mismatch " ,
{ offsetof ( struct CompactionJobStats , num_single_del_mismatch ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
} ;
static std : : unordered_map < std : : string , OptionTypeInfo > cs_result_type_info = {
{ " output_files " ,
OptionTypeInfo : : Vector < CompactionServiceOutputFile > (
offsetof ( struct CompactionServiceResult , output_files ) ,
OptionVerificationType : : kNormal , OptionTypeFlags : : kNone ,
OptionTypeInfo : : Struct ( " output_files " , & cs_output_file_type_info , 0 ,
OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone ) ) } ,
{ " output_level " ,
{ offsetof ( struct CompactionServiceResult , output_level ) , OptionType : : kInt ,
OptionVerificationType : : kNormal , OptionTypeFlags : : kNone } } ,
{ " output_path " ,
{ offsetof ( struct CompactionServiceResult , output_path ) ,
OptionType : : kEncodedString , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " num_output_records " ,
{ offsetof ( struct CompactionServiceResult , num_output_records ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " total_bytes " ,
{ offsetof ( struct CompactionServiceResult , total_bytes ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " bytes_read " ,
{ offsetof ( struct CompactionServiceResult , bytes_read ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " bytes_written " ,
{ offsetof ( struct CompactionServiceResult , bytes_written ) ,
OptionType : : kUInt64T , OptionVerificationType : : kNormal ,
OptionTypeFlags : : kNone } } ,
{ " stats " , OptionTypeInfo : : Struct (
" stats " , & compaction_job_stats_type_info ,
offsetof ( struct CompactionServiceResult , stats ) ,
OptionVerificationType : : kNormal , OptionTypeFlags : : kNone ) } ,
} ;
Status CompactionServiceInput : : Read ( const std : : string & data_str ,
CompactionServiceInput * obj ) {
auto format_version = DecodeFixed32 ( data_str . data ( ) ) ;
if ( format_version = = kOptionsString ) {
ConfigOptions cf ;
cf . invoke_prepare_options = false ;
cf . ignore_unknown_options = true ;
return OptionTypeInfo : : ParseType (
cf , data_str . substr ( sizeof ( BinaryFormatVersion ) ) , cs_input_type_info ,
obj ) ;
} else {
return Status : : NotSupported (
" Compaction Service Input data version not supported: " +
ToString ( format_version ) ) ;
}
}
Status CompactionServiceInput : : Write ( std : : string * output ) {
char buf [ sizeof ( BinaryFormatVersion ) ] ;
EncodeFixed32 ( buf , kOptionsString ) ;
output - > append ( buf , sizeof ( BinaryFormatVersion ) ) ;
ConfigOptions cf ;
cf . invoke_prepare_options = false ;
return OptionTypeInfo : : SerializeType ( cf , cs_input_type_info , this , output ) ;
}
Status CompactionServiceResult : : Read ( const std : : string & data_str ,
CompactionServiceResult * obj ) {
auto format_version = DecodeFixed32 ( data_str . data ( ) ) ;
if ( format_version = = kOptionsString ) {
ConfigOptions cf ;
cf . invoke_prepare_options = false ;
cf . ignore_unknown_options = true ;
return OptionTypeInfo : : ParseType (
cf , data_str . substr ( sizeof ( BinaryFormatVersion ) ) , cs_result_type_info ,
obj ) ;
} else {
return Status : : NotSupported (
" Compaction Service Result data version not supported: " +
ToString ( format_version ) ) ;
}
}
Status CompactionServiceResult : : Write ( std : : string * output ) {
char buf [ sizeof ( BinaryFormatVersion ) ] ;
EncodeFixed32 ( buf , kOptionsString ) ;
output - > append ( buf , sizeof ( BinaryFormatVersion ) ) ;
ConfigOptions cf ;
cf . invoke_prepare_options = false ;
return OptionTypeInfo : : SerializeType ( cf , cs_result_type_info , this , output ) ;
}
# ifndef NDEBUG
bool CompactionServiceResult : : TEST_Equals ( CompactionServiceResult * other ) {
std : : string mismatch ;
return TEST_Equals ( other , & mismatch ) ;
}
bool CompactionServiceResult : : TEST_Equals ( CompactionServiceResult * other ,
std : : string * mismatch ) {
ConfigOptions cf ;
cf . invoke_prepare_options = false ;
return OptionTypeInfo : : TypesAreEqual ( cf , cs_result_type_info , this , other ,
mismatch ) ;
}
bool CompactionServiceInput : : TEST_Equals ( CompactionServiceInput * other ) {
std : : string mismatch ;
return TEST_Equals ( other , & mismatch ) ;
}
bool CompactionServiceInput : : TEST_Equals ( CompactionServiceInput * other ,
std : : string * mismatch ) {
ConfigOptions cf ;
cf . invoke_prepare_options = false ;
return OptionTypeInfo : : TypesAreEqual ( cf , cs_input_type_info , this , other ,
mismatch ) ;
}
# endif // NDEBUG
# endif // !ROCKSDB_LITE
} // namespace ROCKSDB_NAMESPACE
} // namespace ROCKSDB_NAMESPACE