forked from continuwuation/rocksdb
4951494a27
Summary: Continuing work from https://github.com/facebook/rocksdb/issues/13965. Here I'm migrating the "next with shift" kind of bit field and for that I've added an API for atomic additive transformations that can be combined into a single atomic update for multiple fields. (I implemented more features than needed, just in case they are needed someday and to demonstrate what is possible.) Pull Request resolved: https://github.com/facebook/rocksdb/pull/14027 Test Plan: BitFields unit test updated/added, existing HCC tests Reviewed By: xingbowang Differential Revision: D83895094 Pulled By: pdillinger fbshipit-source-id: e4487f34f5607b20f94b85a645ca654e6401e35d
86 lines
2.6 KiB
C++
86 lines
2.6 KiB
C++
// 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).
|
|
|
|
#include "test_util/sync_point.h"
|
|
|
|
#include <fcntl.h>
|
|
|
|
#include "test_util/sync_point_impl.h"
|
|
|
|
std::vector<std::string> rocksdb_kill_exclude_prefixes;
|
|
|
|
#ifndef NDEBUG
|
|
namespace ROCKSDB_NAMESPACE {
|
|
|
|
SyncPoint* SyncPoint::GetInstance() {
|
|
static SyncPoint sync_point;
|
|
return &sync_point;
|
|
}
|
|
|
|
SyncPoint::SyncPoint() : impl_(new Data) {}
|
|
|
|
SyncPoint::~SyncPoint() { delete impl_; }
|
|
|
|
void SyncPoint::LoadDependency(const std::vector<SyncPointPair>& dependencies) {
|
|
impl_->LoadDependency(dependencies);
|
|
}
|
|
|
|
void SyncPoint::LoadDependencyAndMarkers(
|
|
const std::vector<SyncPointPair>& dependencies,
|
|
const std::vector<SyncPointPair>& markers) {
|
|
impl_->LoadDependencyAndMarkers(dependencies, markers);
|
|
}
|
|
|
|
void SyncPoint::SetCallBack(const std::string& point,
|
|
const std::function<void(void*)>& callback) {
|
|
impl_->SetCallBack(point, callback);
|
|
}
|
|
|
|
void SyncPoint::ClearCallBack(const std::string& point) {
|
|
impl_->ClearCallBack(point);
|
|
}
|
|
|
|
void SyncPoint::ClearAllCallBacks() { impl_->ClearAllCallBacks(); }
|
|
|
|
void SyncPoint::EnableProcessing() { impl_->EnableProcessing(); }
|
|
|
|
void SyncPoint::DisableProcessing() { impl_->DisableProcessing(); }
|
|
|
|
void SyncPoint::ClearTrace() { impl_->ClearTrace(); }
|
|
|
|
void SyncPoint::Process(const Slice& point, void* cb_arg) {
|
|
impl_->Process(point, cb_arg);
|
|
}
|
|
|
|
} // namespace ROCKSDB_NAMESPACE
|
|
#endif // NDEBUG
|
|
|
|
namespace ROCKSDB_NAMESPACE {
|
|
void SetupSyncPointsToMockDirectIO() {
|
|
#if !defined(NDEBUG) && !defined(OS_MACOSX) && !defined(OS_WIN) && \
|
|
!defined(OS_SOLARIS) && !defined(OS_AIX) && !defined(OS_OPENBSD)
|
|
ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->SetCallBack(
|
|
"NewWritableFile:O_DIRECT", [&](void* arg) {
|
|
int* val = static_cast<int*>(arg);
|
|
*val &= ~O_DIRECT;
|
|
});
|
|
ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->SetCallBack(
|
|
"NewRandomAccessFile:O_DIRECT", [&](void* arg) {
|
|
int* val = static_cast<int*>(arg);
|
|
*val &= ~O_DIRECT;
|
|
});
|
|
ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->SetCallBack(
|
|
"NewSequentialFile:O_DIRECT", [&](void* arg) {
|
|
int* val = static_cast<int*>(arg);
|
|
*val &= ~O_DIRECT;
|
|
});
|
|
ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->EnableProcessing();
|
|
#endif
|
|
}
|
|
|
|
#ifndef NDEBUG
|
|
std::atomic<int> g_throw_on_testable_assertion_failure{0};
|
|
#endif // NDEBUG
|
|
} // namespace ROCKSDB_NAMESPACE
|