mirror of
https://github.com/facebook/sapling.git
synced 2024-10-10 16:57:49 +03:00
9c0effaff3
Summary: The definition of RLockedPtr should use a non-const Subclass type, and not a const one. Reviewed By: ot Differential Revision: D15356827 fbshipit-source-id: b8ad41e263f0e15ffa25b0698aa85eab8ca2ccb8
110 lines
3.4 KiB
C++
110 lines
3.4 KiB
C++
/*
|
|
* Copyright (c) Facebook, Inc. and its affiliates.
|
|
*
|
|
* This software may be used and distributed according to the terms of the
|
|
* GNU General Public License version 2.
|
|
*/
|
|
#pragma once
|
|
#include <folly/CppAttributes.h>
|
|
#include <folly/Synchronized.h>
|
|
|
|
#include "eden/fs/rocksdb/RocksHandles.h"
|
|
#include "eden/fs/store/LocalStore.h"
|
|
#include "eden/fs/utils/UnboundedQueueExecutor.h"
|
|
|
|
namespace facebook {
|
|
namespace eden {
|
|
|
|
class FaultInjector;
|
|
|
|
/** An implementation of LocalStore that uses RocksDB for the underlying
|
|
* storage.
|
|
*/
|
|
class RocksDbLocalStore : public LocalStore {
|
|
public:
|
|
/**
|
|
* The given FaultInjector must be valid during the lifetime of this
|
|
* RocksDbLocalStore object.
|
|
*/
|
|
explicit RocksDbLocalStore(
|
|
AbsolutePathPiece pathToRocksDb,
|
|
FaultInjector* FOLLY_NONNULL faultInjector,
|
|
RocksDBOpenMode mode = RocksDBOpenMode::ReadWrite);
|
|
~RocksDbLocalStore();
|
|
void close() override;
|
|
void clearKeySpace(KeySpace keySpace) override;
|
|
void compactKeySpace(KeySpace keySpace) override;
|
|
StoreResult get(LocalStore::KeySpace keySpace, folly::ByteRange key)
|
|
const override;
|
|
FOLLY_NODISCARD folly::Future<StoreResult> getFuture(
|
|
KeySpace keySpace,
|
|
folly::ByteRange key) const override;
|
|
FOLLY_NODISCARD folly::Future<std::vector<StoreResult>> getBatch(
|
|
KeySpace keySpace,
|
|
const std::vector<folly::ByteRange>& keys) const override;
|
|
bool hasKey(LocalStore::KeySpace keySpace, folly::ByteRange key)
|
|
const override;
|
|
void put(
|
|
LocalStore::KeySpace keySpace,
|
|
folly::ByteRange key,
|
|
folly::ByteRange value) override;
|
|
std::unique_ptr<WriteBatch> beginWrite(size_t bufSize = 0) override;
|
|
|
|
// Call RocksDB's RepairDB() function on the DB at the specified location
|
|
static void repairDB(AbsolutePathPiece path);
|
|
|
|
// Get the approximate number of bytes stored on disk for the
|
|
// specified key space.
|
|
uint64_t getApproximateSize(KeySpace keySpace) const;
|
|
|
|
void periodicManagementTask(const EdenConfig& config) override;
|
|
|
|
private:
|
|
/**
|
|
* Get a pointer to the RocksHandles object in order to perform an I/O
|
|
* operation.
|
|
*
|
|
* Note that even though this acquires a read-lock, write operations to the
|
|
* DB may still be performed. The lock exists to prevent the DB from being
|
|
* closed while the I/O operation is in progress.
|
|
*/
|
|
folly::Synchronized<RocksHandles>::ConstRLockedPtr getHandles() const {
|
|
auto handles = dbHandles_.rlock();
|
|
if (!handles->db) {
|
|
throwStoreClosedError();
|
|
}
|
|
return handles;
|
|
}
|
|
[[noreturn]] void throwStoreClosedError() const;
|
|
std::shared_ptr<RocksDbLocalStore> getSharedFromThis() {
|
|
return std::static_pointer_cast<RocksDbLocalStore>(shared_from_this());
|
|
}
|
|
std::shared_ptr<const RocksDbLocalStore> getSharedFromThis() const {
|
|
return std::static_pointer_cast<const RocksDbLocalStore>(
|
|
shared_from_this());
|
|
}
|
|
|
|
struct AutoGCState {
|
|
bool inProgress_{false};
|
|
std::chrono::steady_clock::time_point startTime_;
|
|
};
|
|
|
|
/**
|
|
* Publish fb303 counters.
|
|
* Returns the approximate size of all ephemeral column families.
|
|
*/
|
|
size_t publishStats();
|
|
|
|
void triggerAutoGC();
|
|
void autoGCFinished(bool successful);
|
|
|
|
const std::string statsPrefix_{"local_store."};
|
|
FaultInjector& faultInjector_;
|
|
mutable UnboundedQueueExecutor ioPool_;
|
|
folly::Synchronized<AutoGCState> autoGCState_;
|
|
folly::Synchronized<RocksHandles> dbHandles_;
|
|
};
|
|
|
|
} // namespace eden
|
|
} // namespace facebook
|