mirror of
https://github.com/facebook/rocksdb.git
synced 2024-11-27 11:43:49 +00:00
0c4d1053df
Summary: TSAN complained that these are non-atomic reads and writes from different threads. Test Plan: TSAN no longer complains Reviewers: yhchiang, rven, sdong Reviewed By: sdong Subscribers: dhruba, leveldb Differential Revision: https://reviews.facebook.net/D32409
43 lines
1.4 KiB
C++
43 lines
1.4 KiB
C++
// 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
|