2016-02-10 00:12:00 +01:00
|
|
|
// Copyright (c) 2011-present, Facebook, Inc. All rights reserved.
|
2017-07-16 01:03:42 +02:00
|
|
|
// 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).
|
2015-06-02 11:07:58 +02:00
|
|
|
//
|
|
|
|
#pragma once
|
2017-04-06 04:02:00 +02:00
|
|
|
#include "monitoring/perf_level_imp.h"
|
2015-06-02 11:07:58 +02:00
|
|
|
#include "rocksdb/env.h"
|
|
|
|
#include "util/stop_watch.h"
|
|
|
|
|
|
|
|
namespace rocksdb {
|
|
|
|
|
|
|
|
class PerfStepTimer {
|
|
|
|
public:
|
2018-05-17 21:46:23 +02:00
|
|
|
explicit PerfStepTimer(uint64_t* metric, bool for_mutex = false,
|
|
|
|
Statistics* statistics = nullptr,
|
|
|
|
uint32_t ticker_type = 0)
|
|
|
|
: perf_counter_enabled_(
|
|
|
|
perf_level >= PerfLevel::kEnableTime ||
|
|
|
|
(!for_mutex && perf_level >= kEnableTimeExceptForMutex)),
|
|
|
|
env_((perf_counter_enabled_ || statistics != nullptr) ? Env::Default()
|
|
|
|
: nullptr),
|
2016-01-21 22:55:56 +01:00
|
|
|
start_(0),
|
2018-05-17 21:46:23 +02:00
|
|
|
metric_(metric),
|
|
|
|
statistics_(statistics),
|
|
|
|
ticker_type_(ticker_type) {}
|
2015-06-02 11:07:58 +02:00
|
|
|
|
|
|
|
~PerfStepTimer() {
|
|
|
|
Stop();
|
|
|
|
}
|
|
|
|
|
|
|
|
void Start() {
|
2018-05-17 21:46:23 +02:00
|
|
|
if (perf_counter_enabled_ || statistics_ != nullptr) {
|
2015-06-02 11:07:58 +02:00
|
|
|
start_ = env_->NowNanos();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void Measure() {
|
|
|
|
if (start_) {
|
|
|
|
uint64_t now = env_->NowNanos();
|
|
|
|
*metric_ += now - start_;
|
|
|
|
start_ = now;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void Stop() {
|
|
|
|
if (start_) {
|
2018-05-17 21:46:23 +02:00
|
|
|
uint64_t duration = env_->NowNanos() - start_;
|
|
|
|
if (perf_counter_enabled_) {
|
|
|
|
*metric_ += duration;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (statistics_ != nullptr) {
|
|
|
|
RecordTick(statistics_, ticker_type_, duration);
|
|
|
|
}
|
2015-06-02 11:07:58 +02:00
|
|
|
start_ = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
private:
|
2018-05-17 21:46:23 +02:00
|
|
|
const bool perf_counter_enabled_;
|
2015-06-02 11:07:58 +02:00
|
|
|
Env* const env_;
|
|
|
|
uint64_t start_;
|
|
|
|
uint64_t* metric_;
|
2018-05-17 21:46:23 +02:00
|
|
|
Statistics* statistics_;
|
|
|
|
uint32_t ticker_type_;
|
2015-06-02 11:07:58 +02:00
|
|
|
};
|
|
|
|
|
|
|
|
} // namespace rocksdb
|