d4d338de33
Summary: This diff adds timeout_hint_us to WriteOptions. If it's non-zero, then 1) writes associated with this options MAY be aborted when it has been waiting for longer than the specified time. If an abortion happens, associated writes will return Status::TimeOut. 2) the stall time of the associated write caused by flush or compaction will be limited by timeout_hint_us. The default value of timeout_hint_us is 0 (i.e., OFF.) The statistics of timeout writes will be recorded in WRITE_TIMEDOUT. Test Plan: export ROCKSDB_TESTS=WriteTimeoutAndDelayTest make db_test ./db_test Reviewers: igor, ljin, haobo, sdong Reviewed By: sdong Subscribers: dhruba, leveldb Differential Revision: https://reviews.facebook.net/D18837
68 lines
1.6 KiB
C++
68 lines
1.6 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.
|
|
//
|
|
#pragma once
|
|
#include "rocksdb/env.h"
|
|
#include "util/statistics.h"
|
|
|
|
namespace rocksdb {
|
|
// Auto-scoped.
|
|
// Records the statistic into the corresponding histogram.
|
|
class StopWatch {
|
|
public:
|
|
explicit StopWatch(
|
|
Env * const env,
|
|
Statistics* statistics = nullptr,
|
|
const Histograms histogram_name = DB_GET,
|
|
bool auto_start = true) :
|
|
env_(env),
|
|
start_time_((!auto_start && !statistics) ? 0 : env->NowMicros()),
|
|
statistics_(statistics),
|
|
histogram_name_(histogram_name) {}
|
|
|
|
|
|
|
|
uint64_t ElapsedMicros() const {
|
|
return env_->NowMicros() - start_time_;
|
|
}
|
|
|
|
~StopWatch() { MeasureTime(statistics_, histogram_name_, ElapsedMicros()); }
|
|
|
|
private:
|
|
Env* const env_;
|
|
const uint64_t start_time_;
|
|
Statistics* statistics_;
|
|
const Histograms histogram_name_;
|
|
|
|
};
|
|
|
|
// a nano second precision stopwatch
|
|
class StopWatchNano {
|
|
public:
|
|
explicit StopWatchNano(Env* const env, bool auto_start = false)
|
|
: env_(env), start_(0) {
|
|
if (auto_start) {
|
|
Start();
|
|
}
|
|
}
|
|
|
|
void Start() { start_ = env_->NowNanos(); }
|
|
|
|
uint64_t ElapsedNanos(bool reset = false) {
|
|
auto now = env_->NowNanos();
|
|
auto elapsed = now - start_;
|
|
if (reset) {
|
|
start_ = now;
|
|
}
|
|
return elapsed;
|
|
}
|
|
|
|
private:
|
|
Env* const env_;
|
|
uint64_t start_;
|
|
};
|
|
|
|
} // namespace rocksdb
|