andreacavalli/rocksdb
0
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Changed the CompactionJobStats::output_key_prefix type from char[] to string.

Browse Source 
Summary:
Keys in RocksDB can be arbitrary byte strings.  However, in the current
CompactionJobStats, smallest_output_key_prefix and largest_output_key_prefix
are of type char[] without having a length, which is insufficient to handle
non-null terminated strings.

This patch change their type to std::string.

Test Plan: compaction_job_stats_test

Reviewers: igor, rven, IslamAbdelRahman, kradhakrishnan, sdong

Reviewed By: sdong

Subscribers: dhruba, leveldb

Differential Revision: https://reviews.facebook.net/D39537
This commit is contained in:
Yueh-Hsuan Chiang 2015-06-04 12:31:12 -07:00
parent 0b3172d071
commit bb808eaddb
3 changed files with 25 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
db/compaction_job.cc
View File

@ -1252,11 +1252,10 @@ void CompactionJob::CleanupCompaction(const Status& status) {
#ifndef ROCKSDB_LITE
namespace {
void CopyPrefix(
char* dst, size_t dst_length, const Slice& src) {
assert(dst_length > 0);
size_t length = src.size() > dst_length - 1 ? dst_length - 1 : src.size();
memcpy(dst, src.data(), length);
dst[length] = 0;
const Slice& src, size_t prefix_length, std::string* dst) {
assert(prefix_length > 0);
size_t length = src.size() > prefix_length ? prefix_length : src.size();
dst->assign(src.data(), length);
}
} // namespace
@ -1285,13 +1284,13 @@ void CompactionJob::UpdateCompactionJobStats(
if (compact_->outputs.size() > 0U) {
CopyPrefix(
compaction_job_stats_->smallest_output_key_prefix,
sizeof(compaction_job_stats_->smallest_output_key_prefix),
compact_->outputs[0].smallest.user_key().ToString());
compact_->outputs[0].smallest.user_key(),
CompactionJobStats::kMaxPrefixLength,
&compaction_job_stats_->smallest_output_key_prefix);
CopyPrefix(
compaction_job_stats_->largest_output_key_prefix,
sizeof(compaction_job_stats_->largest_output_key_prefix),
compact_->current_output()->largest.user_key().ToString());
compact_->current_output()->largest.user_key(),
CompactionJobStats::kMaxPrefixLength,
&compaction_job_stats_->largest_output_key_prefix);
}
}
#endif // !ROCKSDB_LITE

21
db/compaction_job_stats_test.cc
View File

@ -457,11 +457,10 @@ uint64_t EstimatedFileSize(
namespace {
void CopyPrefix(
char* dst, size_t dst_length, const Slice& src) {
assert(dst_length > 0);
size_t length = src.size() > dst_length - 1 ? dst_length - 1 : src.size();
memcpy(dst, src.data(), length);
dst[length] = 0;
const Slice& src, size_t prefix_length, std::string* dst) {
assert(prefix_length > 0);
size_t length = src.size() > prefix_length ? prefix_length : src.size();
dst->assign(src.data(), length);
}
} // namespace
@ -499,12 +498,12 @@ CompactionJobStats NewManualCompactionJobStats(
stats.num_records_replaced = num_records_replaced;
CopyPrefix(stats.smallest_output_key_prefix,
sizeof(stats.smallest_output_key_prefix),
smallest_key);
CopyPrefix(stats.largest_output_key_prefix,
sizeof(stats.largest_output_key_prefix),
largest_key);
CopyPrefix(smallest_key,
CompactionJobStats::kMaxPrefixLength,
&stats.smallest_output_key_prefix);
CopyPrefix(largest_key,
CompactionJobStats::kMaxPrefixLength,
&stats.largest_output_key_prefix);
return stats;
}

7
include/rocksdb/compaction_job_stats.h
View File

@ -6,6 +6,7 @@
#pragma once
#include <stddef.h>
#include <stdint.h>
#include <string>
namespace rocksdb {
struct CompactionJobStats {
@ -45,7 +46,9 @@ struct CompactionJobStats {
// 0-terminated strings storing the first 8 bytes of the smallest and
// largest key in the output.
char smallest_output_key_prefix[9];
char largest_output_key_prefix[9];
static const size_t kMaxPrefixLength = 8;
std::string smallest_output_key_prefix;
std::string largest_output_key_prefix;
};
} // namespace rocksdb