2016-02-10 00:12:00 +01:00
|
|
|
// Copyright (c) 2011-present, Facebook, Inc. All rights reserved.
|
2015-02-05 06:39:45 +01:00
|
|
|
// 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/instrumented_mutex.h"
|
2015-10-13 19:41:48 +02:00
|
|
|
#include "util/perf_context_imp.h"
|
2016-10-18 07:22:48 +02:00
|
|
|
#include "util/sync_point.h"
|
2015-02-05 06:39:45 +01:00
|
|
|
#include "util/thread_status_util.h"
|
|
|
|
|
|
|
|
namespace rocksdb {
|
2016-01-26 02:07:37 +01:00
|
|
|
namespace {
|
|
|
|
bool ShouldReportToStats(Env* env, Statistics* stats) {
|
|
|
|
return env != nullptr && stats != nullptr &&
|
2016-09-02 04:57:55 +02:00
|
|
|
stats->stats_level_ > kExceptTimeForMutex;
|
2016-01-26 02:07:37 +01:00
|
|
|
}
|
|
|
|
} // namespace
|
|
|
|
|
2015-02-05 06:39:45 +01:00
|
|
|
void InstrumentedMutex::Lock() {
|
2016-01-21 22:55:56 +01:00
|
|
|
PERF_CONDITIONAL_TIMER_FOR_MUTEX_GUARD(db_mutex_lock_nanos,
|
|
|
|
stats_code_ == DB_MUTEX_WAIT_MICROS);
|
2015-02-05 06:39:45 +01:00
|
|
|
uint64_t wait_time_micros = 0;
|
2016-01-26 02:07:37 +01:00
|
|
|
if (ShouldReportToStats(env_, stats_)) {
|
2015-02-05 06:39:45 +01:00
|
|
|
{
|
|
|
|
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() {
|
2016-01-21 22:55:56 +01:00
|
|
|
PERF_CONDITIONAL_TIMER_FOR_MUTEX_GUARD(db_condition_wait_nanos,
|
|
|
|
stats_code_ == DB_MUTEX_WAIT_MICROS);
|
2015-02-05 06:39:45 +01:00
|
|
|
uint64_t wait_time_micros = 0;
|
2016-01-26 02:07:37 +01:00
|
|
|
if (ShouldReportToStats(env_, stats_)) {
|
2015-02-05 06:39:45 +01:00
|
|
|
{
|
|
|
|
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) {
|
2016-01-21 22:55:56 +01:00
|
|
|
PERF_CONDITIONAL_TIMER_FOR_MUTEX_GUARD(db_condition_wait_nanos,
|
|
|
|
stats_code_ == DB_MUTEX_WAIT_MICROS);
|
2015-02-05 06:39:45 +01:00
|
|
|
uint64_t wait_time_micros = 0;
|
|
|
|
bool result = false;
|
2016-01-26 02:07:37 +01:00
|
|
|
if (ShouldReportToStats(env_, stats_)) {
|
2015-02-05 06:39:45 +01:00
|
|
|
{
|
|
|
|
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
|
2016-10-18 07:22:48 +02:00
|
|
|
|
|
|
|
TEST_SYNC_POINT_CALLBACK("InstrumentedCondVar::TimedWaitInternal",
|
|
|
|
&abs_time_us);
|
|
|
|
|
2015-02-05 06:39:45 +01:00
|
|
|
return cond_.TimedWait(abs_time_us);
|
|
|
|
}
|
|
|
|
|
|
|
|
} // namespace rocksdb
|