rocksdb/util/thread_status_util_debug.cc

// Copyright (c) 2013, 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 <atomic>

#include "rocksdb/env.h"
#include "util/thread_status_updater.h"
#include "util/thread_status_util.h"

namespace rocksdb {

#ifndef NDEBUG
// the delay for debugging purpose.
static std::atomic<int> operations_delay[ThreadStatus::NUM_OP_TYPES];
static std::atomic<int> states_delay[ThreadStatus::NUM_STATE_TYPES];

void ThreadStatusUtil::TEST_SetStateDelay(
    const ThreadStatus::StateType state, int micro) {
  states_delay[state].store(micro, std::memory_order_relaxed);
}

void ThreadStatusUtil::TEST_StateDelay(
    const ThreadStatus::StateType state) {
  Env::Default()->SleepForMicroseconds(
      states_delay[state].load(std::memory_order_relaxed));
}

void ThreadStatusUtil::TEST_SetOperationDelay(
    const ThreadStatus::OperationType operation, int micro) {
  operations_delay[operation].store(micro, std::memory_order_relaxed);
}

void ThreadStatusUtil::TEST_OperationDelay(
    const ThreadStatus::OperationType operation) {
  Env::Default()->SleepForMicroseconds(
      operations_delay[operation].load(std::memory_order_relaxed));
}
#endif  // !NDEBUG

}  // namespace rocksdb