mirror of https://github.com/facebook/rocksdb.git
77 lines
2.0 KiB
C++
77 lines
2.0 KiB
C++
// Copyright (c) 2015, 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.
|
|
|
|
#include "util/perf_context_imp.h"
|
|
#include "util/instrumented_mutex.h"
|
|
#include "util/thread_status_util.h"
|
|
|
|
namespace rocksdb {
|
|
void InstrumentedMutex::Lock() {
|
|
PERF_TIMER_GUARD(db_mutex_lock_nanos);
|
|
uint64_t wait_time_micros = 0;
|
|
if (env_ != nullptr && stats_ != nullptr) {
|
|
{
|
|
StopWatch sw(env_, nullptr, 0, &wait_time_micros);
|
|
LockInternal();
|
|
}
|
|
RecordTick(stats_, stats_code_, wait_time_micros);
|
|
} else {
|
|
LockInternal();
|
|
}
|
|
}
|
|
|
|
void InstrumentedMutex::LockInternal() {
|
|
#ifndef NDEBUG
|
|
ThreadStatusUtil::TEST_StateDelay(ThreadStatus::STATE_MUTEX_WAIT);
|
|
#endif
|
|
mutex_.Lock();
|
|
}
|
|
|
|
void InstrumentedCondVar::Wait() {
|
|
PERF_TIMER_GUARD(db_condition_wait_nanos);
|
|
uint64_t wait_time_micros = 0;
|
|
if (env_ != nullptr && stats_ != nullptr) {
|
|
{
|
|
StopWatch sw(env_, nullptr, 0, &wait_time_micros);
|
|
WaitInternal();
|
|
}
|
|
RecordTick(stats_, stats_code_, wait_time_micros);
|
|
} else {
|
|
WaitInternal();
|
|
}
|
|
}
|
|
|
|
void InstrumentedCondVar::WaitInternal() {
|
|
#ifndef NDEBUG
|
|
ThreadStatusUtil::TEST_StateDelay(ThreadStatus::STATE_MUTEX_WAIT);
|
|
#endif
|
|
cond_.Wait();
|
|
}
|
|
|
|
bool InstrumentedCondVar::TimedWait(uint64_t abs_time_us) {
|
|
PERF_TIMER_GUARD(db_condition_wait_nanos);
|
|
uint64_t wait_time_micros = 0;
|
|
bool result = false;
|
|
if (env_ != nullptr && stats_ != nullptr) {
|
|
{
|
|
StopWatch sw(env_, nullptr, 0, &wait_time_micros);
|
|
result = TimedWaitInternal(abs_time_us);
|
|
}
|
|
RecordTick(stats_, stats_code_, wait_time_micros);
|
|
} else {
|
|
result = TimedWaitInternal(abs_time_us);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
bool InstrumentedCondVar::TimedWaitInternal(uint64_t abs_time_us) {
|
|
#ifndef NDEBUG
|
|
ThreadStatusUtil::TEST_StateDelay(ThreadStatus::STATE_MUTEX_WAIT);
|
|
#endif
|
|
return cond_.TimedWait(abs_time_us);
|
|
}
|
|
|
|
} // namespace rocksdb
|