56 lines
1.5 KiB
C++
56 lines
1.5 KiB
C++
|
// Copyright (c) 2013, Facebook, Inc. All rights reserved.
|
||
|
|
// This source code is licensed under the BSD-style license found in the
|
||
|
|
// LICENSE file in the root directory of this source tree. An additional grant
|
||
|
|
// of patent rights can be found in the PATENTS file in the same directory.
|
||
|
|
|
||
|
|
#include "db/table_stats_collector.h"
|
||
|
|
|
||
|
|
#include "db/dbformat.h"
|
||
|
|
#include "util/coding.h"
|
||
|
|
|
||
|
|
namespace rocksdb {
|
||
|
|
|
||
|
|
Status InternalKeyStatsCollector::Add(const Slice& key, const Slice& value) {
|
||
|
|
ParsedInternalKey ikey;
|
||
|
|
if (!ParseInternalKey(key, &ikey)) {
|
||
|
|
return Status::InvalidArgument("Invalid internal key");
|
||
|
|
}
|
||
|
|
|
||
|
|
if (ikey.type == ValueType::kTypeDeletion) {
|
||
|
|
++deleted_keys_;
|
||
|
|
}
|
||
|
|
|
||
|
|
return Status::OK();
|
||
|
|
}
|
||
|
|
|
||
|
|
Status InternalKeyStatsCollector::Finish(
|
||
|
|
TableStats::UserCollectedStats* stats) {
|
||
|
|
assert(stats);
|
||
|
|
assert(stats->find(InternalKeyTableStatsNames::kDeletedKeys) == stats->end());
|
||
|
|
std::string val;
|
||
|
|
|
||
|
|
PutVarint64(&val, deleted_keys_);
|
||
|
|
stats->insert(std::make_pair(InternalKeyTableStatsNames::kDeletedKeys, val));
|
||
|
|
|
||
|
|
return Status::OK();
|
||
|
|
}
|
||
|
|
|
||
|
|
Status UserKeyTableStatsCollector::Add(const Slice& key, const Slice& value) {
|
||
|
|
ParsedInternalKey ikey;
|
||
|
|
if (!ParseInternalKey(key, &ikey)) {
|
||
|
|
return Status::InvalidArgument("Invalid internal key");
|
||
|
|
}
|
||
|
|
|
||
|
|
return collector_->Add(ikey.user_key, value);
|
||
|
|
}
|
||
|
|
|
||
|
|
Status UserKeyTableStatsCollector::Finish(
|
||
|
|
TableStats::UserCollectedStats* stats) {
|
||
|
|
return collector_->Finish(stats);
|
||
|
|
}
|
||
|
|
|
||
|
|
const std::string InternalKeyTableStatsNames::kDeletedKeys
|
||
|
|
= "rocksdb.deleted.keys";
|
||
|
|
|
||
|
|
} // namespace rocksdb
|