69 lines
1.9 KiB
C
69 lines
1.9 KiB
C
|
#ifndef STORAGE_LEVELDB_UTIL_STOP_WATCH_H_
|
||
|
|
#define STORAGE_LEVELDB_UTIL_STOP_WATCH_H_
|
||
|
|
|
||
|
|
#include "leveldb/env.h"
|
||
|
|
#include "leveldb/statistics.h"
|
||
|
|
#include <iostream>
|
||
|
|
namespace leveldb {
|
||
|
|
|
||
|
|
class StopWatch {
|
||
|
|
public:
|
||
|
|
virtual uint64_t ElapsedMicros() = 0;
|
||
|
|
virtual ~StopWatch() {}
|
||
|
|
};
|
||
|
|
|
||
|
|
class DoNothingStopWatch : public StopWatch {
|
||
|
|
public:
|
||
|
|
virtual uint64_t ElapsedMicros() {
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
// Auto-scoped.
|
||
|
|
// Records the statistic into the corresponding histogram.
|
||
|
|
class ScopedRecordingStopWatch : public StopWatch {
|
||
|
|
public:
|
||
|
|
ScopedRecordingStopWatch(Env * const env,
|
||
|
|
Statistics * const statistics,
|
||
|
|
const Histograms histogram_name) :
|
||
|
|
env_(env),
|
||
|
|
start_time_(env->NowMicros()),
|
||
|
|
statistics_(statistics),
|
||
|
|
histogram_name_(histogram_name) {}
|
||
|
|
|
||
|
|
virtual uint64_t ElapsedMicros() {
|
||
|
|
return env_->NowMicros() - start_time_;
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual ~ScopedRecordingStopWatch() {
|
||
|
|
uint64_t elapsed_time = env_->NowMicros() - start_time_;
|
||
|
|
statistics_->measureTime(histogram_name_, elapsed_time);
|
||
|
|
}
|
||
|
|
|
||
|
|
private:
|
||
|
|
Env* const env_;
|
||
|
|
const uint64_t start_time_;
|
||
|
|
Statistics* const statistics_;
|
||
|
|
const Histograms histogram_name_;
|
||
|
|
|
||
|
|
};
|
||
|
|
|
||
|
|
namespace stats {
|
||
|
|
// Helper method
|
||
|
|
std::unique_ptr<StopWatch> StartStopWatch(Env * const env,
|
||
|
|
Statistics * const statistics,
|
||
|
|
Histograms histogram_name) {
|
||
|
|
assert(env);
|
||
|
|
if (statistics != nullptr) {
|
||
|
|
return std::unique_ptr<StopWatch>(new ScopedRecordingStopWatch(
|
||
|
|
env,
|
||
|
|
statistics,
|
||
|
|
histogram_name));
|
||
|
|
} else {
|
||
|
|
return std::unique_ptr<StopWatch>(new DoNothingStopWatch());
|
||
|
|
}
|
||
|
|
};
|
||
|
|
} // namespace stats
|
||
|
|
} // namespace leveldb
|
||
|
|
#endif // STORAGE_LEVELDB_UTIL_STOP_WATCH_H_
|