2016-10-14 20:39:59 +03:00
|
|
|
/*
|
2019-06-20 02:58:25 +03:00
|
|
|
* Copyright (c) Facebook, Inc. and its affiliates.
|
2016-10-14 20:39:59 +03:00
|
|
|
*
|
2019-06-20 02:58:25 +03:00
|
|
|
* This software may be used and distributed according to the terms of the
|
|
|
|
* GNU General Public License version 2.
|
2016-10-14 20:39:59 +03:00
|
|
|
*/
|
2019-10-11 15:26:59 +03:00
|
|
|
|
2016-10-14 20:39:59 +03:00
|
|
|
#pragma once
|
|
|
|
|
|
|
|
#include <unordered_map>
|
|
|
|
#include "eden/fs/store/IObjectStore.h"
|
|
|
|
|
|
|
|
#include "eden/fs/model/Blob.h"
|
|
|
|
#include "eden/fs/model/Hash.h"
|
|
|
|
#include "eden/fs/model/Tree.h"
|
2016-12-14 05:11:05 +03:00
|
|
|
#include "eden/fs/store/BlobMetadata.h"
|
2020-03-17 12:29:57 +03:00
|
|
|
#include "eden/fs/store/ImportPriority.h"
|
2020-06-23 19:52:08 +03:00
|
|
|
#include "eden/fs/store/ObjectFetchContext.h"
|
2016-10-14 20:39:59 +03:00
|
|
|
|
|
|
|
namespace facebook {
|
|
|
|
namespace eden {
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Fake implementation of IObjectStore that allows the data to be injected
|
|
|
|
* directly. This is designed to be used for unit tests.
|
|
|
|
*/
|
|
|
|
class FakeObjectStore : public IObjectStore {
|
|
|
|
public:
|
|
|
|
FakeObjectStore();
|
2017-05-24 02:08:03 +03:00
|
|
|
~FakeObjectStore() override;
|
2016-10-14 20:39:59 +03:00
|
|
|
|
|
|
|
void addTree(Tree&& tree);
|
|
|
|
void addBlob(Blob&& blob);
|
|
|
|
void setTreeForCommit(const Hash& commitID, Tree&& tree);
|
|
|
|
|
2017-09-30 02:43:01 +03:00
|
|
|
folly::Future<std::shared_ptr<const Tree>> getTree(
|
2020-02-06 00:13:11 +03:00
|
|
|
const Hash& id,
|
|
|
|
ObjectFetchContext& context =
|
|
|
|
ObjectFetchContext::getNullContext()) const override;
|
2017-09-30 02:43:01 +03:00
|
|
|
folly::Future<std::shared_ptr<const Blob>> getBlob(
|
2020-02-06 00:13:11 +03:00
|
|
|
const Hash& id,
|
2020-07-02 21:58:46 +03:00
|
|
|
ObjectFetchContext& context =
|
|
|
|
ObjectFetchContext::getNullContext()) const override;
|
2017-09-30 02:43:01 +03:00
|
|
|
folly::Future<std::shared_ptr<const Tree>> getTreeForCommit(
|
2020-02-06 00:13:11 +03:00
|
|
|
const Hash& commitID,
|
|
|
|
ObjectFetchContext& context =
|
|
|
|
ObjectFetchContext::getNullContext()) const override;
|
2019-12-10 03:23:34 +03:00
|
|
|
folly::Future<std::shared_ptr<const Tree>> getTreeForManifest(
|
|
|
|
const Hash& commitID,
|
2020-02-06 00:13:11 +03:00
|
|
|
const Hash& manifestID,
|
|
|
|
ObjectFetchContext& context =
|
|
|
|
ObjectFetchContext::getNullContext()) const override;
|
2018-05-25 23:47:49 +03:00
|
|
|
folly::Future<folly::Unit> prefetchBlobs(
|
2020-02-06 00:13:11 +03:00
|
|
|
const std::vector<Hash>& ids,
|
|
|
|
ObjectFetchContext& context =
|
|
|
|
ObjectFetchContext::getNullContext()) const override;
|
2016-12-14 05:11:05 +03:00
|
|
|
|
2018-11-26 21:44:26 +03:00
|
|
|
size_t getAccessCount(const Hash& hash) const;
|
|
|
|
|
2016-10-14 20:39:59 +03:00
|
|
|
private:
|
|
|
|
std::unordered_map<Hash, Tree> trees_;
|
|
|
|
std::unordered_map<Hash, Blob> blobs_;
|
|
|
|
std::unordered_map<Hash, Tree> commits_;
|
2018-11-26 21:44:26 +03:00
|
|
|
mutable std::unordered_map<Hash, size_t> accessCounts_;
|
2016-10-14 20:39:59 +03:00
|
|
|
};
|
2017-11-04 01:58:04 +03:00
|
|
|
} // namespace eden
|
|
|
|
} // namespace facebook
|