78279350aa
Summary: Adding a list of blob db counters. Also remove WaStats() which doesn't expose the stats and can be substitute by (BLOB_DB_BYTES_WRITTEN / BLOB_DB_BLOB_FILE_BYTES_WRITTEN). Closes https://github.com/facebook/rocksdb/pull/3193 Differential Revision: D6394216 Pulled By: yiwu-arbug fbshipit-source-id: 017508c8ff3fcd7ea7403c64d0f9834b24816803
125 lines
3.1 KiB
C++
125 lines
3.1 KiB
C++
// Copyright (c) 2011-present, Facebook, Inc. All rights reserved.
|
|
// This source code is licensed under both the GPLv2 (found in the
|
|
// COPYING file in the root directory) and Apache 2.0 License
|
|
// (found in the LICENSE.Apache file in the root directory).
|
|
|
|
#pragma once
|
|
#ifndef ROCKSDB_LITE
|
|
|
|
#include "monitoring/statistics.h"
|
|
#include "rocksdb/iterator.h"
|
|
#include "util/stop_watch.h"
|
|
#include "utilities/blob_db/blob_db_impl.h"
|
|
|
|
namespace rocksdb {
|
|
namespace blob_db {
|
|
|
|
using rocksdb::ManagedSnapshot;
|
|
|
|
class BlobDBIterator : public Iterator {
|
|
public:
|
|
BlobDBIterator(ManagedSnapshot* snapshot, ArenaWrappedDBIter* iter,
|
|
BlobDBImpl* blob_db, Env* env, Statistics* statistics)
|
|
: snapshot_(snapshot),
|
|
iter_(iter),
|
|
blob_db_(blob_db),
|
|
env_(env),
|
|
statistics_(statistics) {}
|
|
|
|
virtual ~BlobDBIterator() = default;
|
|
|
|
bool Valid() const override {
|
|
if (!iter_->Valid()) {
|
|
return false;
|
|
}
|
|
return status_.ok();
|
|
}
|
|
|
|
Status status() const override {
|
|
if (!iter_->status().ok()) {
|
|
return iter_->status();
|
|
}
|
|
return status_;
|
|
}
|
|
|
|
void SeekToFirst() override {
|
|
StopWatch seek_sw(env_, statistics_, BLOB_DB_SEEK_MICROS);
|
|
RecordTick(statistics_, BLOB_DB_NUM_SEEK);
|
|
iter_->SeekToFirst();
|
|
UpdateBlobValue();
|
|
}
|
|
|
|
void SeekToLast() override {
|
|
StopWatch seek_sw(env_, statistics_, BLOB_DB_SEEK_MICROS);
|
|
RecordTick(statistics_, BLOB_DB_NUM_SEEK);
|
|
iter_->SeekToLast();
|
|
UpdateBlobValue();
|
|
}
|
|
|
|
void Seek(const Slice& target) override {
|
|
StopWatch seek_sw(env_, statistics_, BLOB_DB_SEEK_MICROS);
|
|
RecordTick(statistics_, BLOB_DB_NUM_SEEK);
|
|
iter_->Seek(target);
|
|
UpdateBlobValue();
|
|
}
|
|
|
|
void SeekForPrev(const Slice& target) override {
|
|
StopWatch seek_sw(env_, statistics_, BLOB_DB_SEEK_MICROS);
|
|
RecordTick(statistics_, BLOB_DB_NUM_SEEK);
|
|
iter_->SeekForPrev(target);
|
|
UpdateBlobValue();
|
|
}
|
|
|
|
void Next() override {
|
|
assert(Valid());
|
|
StopWatch next_sw(env_, statistics_, BLOB_DB_NEXT_MICROS);
|
|
RecordTick(statistics_, BLOB_DB_NUM_NEXT);
|
|
iter_->Next();
|
|
UpdateBlobValue();
|
|
}
|
|
|
|
void Prev() override {
|
|
assert(Valid());
|
|
StopWatch prev_sw(env_, statistics_, BLOB_DB_PREV_MICROS);
|
|
RecordTick(statistics_, BLOB_DB_NUM_PREV);
|
|
iter_->Prev();
|
|
UpdateBlobValue();
|
|
}
|
|
|
|
Slice key() const override {
|
|
assert(Valid());
|
|
return iter_->key();
|
|
}
|
|
|
|
Slice value() const override {
|
|
assert(Valid());
|
|
if (!iter_->IsBlob()) {
|
|
return iter_->value();
|
|
}
|
|
return value_;
|
|
}
|
|
|
|
// Iterator::Refresh() not supported.
|
|
|
|
private:
|
|
void UpdateBlobValue() {
|
|
TEST_SYNC_POINT("BlobDBIterator::UpdateBlobValue:Start:1");
|
|
TEST_SYNC_POINT("BlobDBIterator::UpdateBlobValue:Start:2");
|
|
value_.Reset();
|
|
if (iter_->Valid() && iter_->IsBlob()) {
|
|
status_ = blob_db_->GetBlobValue(iter_->key(), iter_->value(), &value_);
|
|
}
|
|
}
|
|
|
|
std::unique_ptr<ManagedSnapshot> snapshot_;
|
|
std::unique_ptr<ArenaWrappedDBIter> iter_;
|
|
BlobDBImpl* blob_db_;
|
|
Env* env_;
|
|
Statistics* statistics_;
|
|
Status status_;
|
|
PinnableSlice value_;
|
|
};
|
|
} // namespace blob_db
|
|
} // namespace rocksdb
|
|
#endif // !ROCKSDB_LITE
|