fork of https://github.com/rust-rocksdb/rust-rocksdb for nextgraph
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
97 lines
3.8 KiB
97 lines
3.8 KiB
1 year ago
|
// 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
|
||
|
|
#include "monitoring/perf_step_timer.h"
|
||
|
|
#include "rocksdb/perf_context.h"
|
||
|
|
#include "util/stop_watch.h"
|
||
|
|
|
||
|
|
namespace ROCKSDB_NAMESPACE {
|
||
|
|
#if defined(NPERF_CONTEXT)
|
||
|
|
extern PerfContext perf_context;
|
||
|
|
#else
|
||
|
|
#if defined(OS_SOLARIS)
|
||
|
|
extern thread_local PerfContext perf_context_;
|
||
|
|
#define perf_context (*get_perf_context())
|
||
|
|
#else
|
||
|
|
extern thread_local PerfContext perf_context;
|
||
|
|
#endif
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#if defined(NPERF_CONTEXT)
|
||
|
|
|
||
|
|
#define PERF_TIMER_STOP(metric)
|
||
|
|
#define PERF_TIMER_START(metric)
|
||
|
|
#define PERF_TIMER_GUARD(metric)
|
||
|
|
#define PERF_TIMER_GUARD_WITH_CLOCK(metric, clock)
|
||
|
|
#define PERF_CPU_TIMER_GUARD(metric, clock)
|
||
|
|
#define PERF_CONDITIONAL_TIMER_FOR_MUTEX_GUARD(metric, condition, stats, \
|
||
|
|
ticker_type)
|
||
|
|
#define PERF_TIMER_MEASURE(metric)
|
||
|
|
#define PERF_COUNTER_ADD(metric, value)
|
||
|
|
#define PERF_COUNTER_BY_LEVEL_ADD(metric, value, level)
|
||
|
|
|
||
|
|
#else
|
||
|
|
|
||
|
|
// Stop the timer and update the metric
|
||
|
|
#define PERF_TIMER_STOP(metric) perf_step_timer_##metric.Stop();
|
||
|
|
|
||
|
|
#define PERF_TIMER_START(metric) perf_step_timer_##metric.Start();
|
||
|
|
|
||
|
|
// Declare and set start time of the timer
|
||
|
|
#define PERF_TIMER_GUARD(metric) \
|
||
|
|
PerfStepTimer perf_step_timer_##metric(&(perf_context.metric)); \
|
||
|
|
perf_step_timer_##metric.Start();
|
||
|
|
|
||
|
|
// Declare and set start time of the timer
|
||
|
|
#define PERF_TIMER_GUARD_WITH_CLOCK(metric, clock) \
|
||
|
|
PerfStepTimer perf_step_timer_##metric(&(perf_context.metric), clock); \
|
||
|
|
perf_step_timer_##metric.Start();
|
||
|
|
|
||
|
|
// Declare and set start time of the timer
|
||
|
|
#define PERF_CPU_TIMER_GUARD(metric, clock) \
|
||
|
|
PerfStepTimer perf_step_timer_##metric( \
|
||
|
|
&(perf_context.metric), clock, true, \
|
||
|
|
PerfLevel::kEnableTimeAndCPUTimeExceptForMutex); \
|
||
|
|
perf_step_timer_##metric.Start();
|
||
|
|
|
||
|
|
#define PERF_CONDITIONAL_TIMER_FOR_MUTEX_GUARD(metric, condition, stats, \
|
||
|
|
ticker_type) \
|
||
|
|
PerfStepTimer perf_step_timer_##metric(&(perf_context.metric), nullptr, \
|
||
|
|
false, PerfLevel::kEnableTime, stats, \
|
||
|
|
ticker_type); \
|
||
|
|
if (condition) { \
|
||
|
|
perf_step_timer_##metric.Start(); \
|
||
|
|
}
|
||
|
|
|
||
|
|
// Update metric with time elapsed since last START. start time is reset
|
||
|
|
// to current timestamp.
|
||
|
|
#define PERF_TIMER_MEASURE(metric) perf_step_timer_##metric.Measure();
|
||
|
|
|
||
|
|
// Increase metric value
|
||
|
|
#define PERF_COUNTER_ADD(metric, value) \
|
||
|
|
if (perf_level >= PerfLevel::kEnableCount) { \
|
||
|
|
perf_context.metric += value; \
|
||
|
|
}
|
||
|
|
|
||
|
|
// Increase metric value
|
||
|
|
#define PERF_COUNTER_BY_LEVEL_ADD(metric, value, level) \
|
||
|
|
if (perf_level >= PerfLevel::kEnableCount && \
|
||
|
|
perf_context.per_level_perf_context_enabled && \
|
||
|
|
perf_context.level_to_perf_context) { \
|
||
|
|
if ((*(perf_context.level_to_perf_context)).find(level) != \
|
||
|
|
(*(perf_context.level_to_perf_context)).end()) { \
|
||
|
|
(*(perf_context.level_to_perf_context))[level].metric += value; \
|
||
|
|
} else { \
|
||
|
|
PerfContextByLevel empty_context; \
|
||
|
|
(*(perf_context.level_to_perf_context))[level] = empty_context; \
|
||
|
|
(*(perf_context.level_to_perf_context))[level].metric += value; \
|
||
|
|
} \
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif
|
||
|
|
|
||
|
|
} // namespace ROCKSDB_NAMESPACE
|