mirror of
https://github.com/facebook/sapling.git
synced 2024-10-05 14:28:17 +03:00
070fc3a461
Summary: As title, this diff fully separate GlobNode and GlobTree, GlobTree only does Glob on tree and do not depend on inode. GlobNode and GlobTree both use shared utilities to minimize code duplication Reviewed By: kmancini Differential Revision: D49998533 fbshipit-source-id: 6323cacd58aa68678ee348620956a967e719b7aa
112 lines
3.6 KiB
C++
112 lines
3.6 KiB
C++
/*
|
|
* Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
*
|
|
* This software may be used and distributed according to the terms of the
|
|
* GNU General Public License version 2.
|
|
*/
|
|
|
|
#include "eden/fs/inodes/GlobNode.h"
|
|
#include <eden/fs/inodes/InodePtrFwd.h>
|
|
#include <iomanip>
|
|
#include <iostream>
|
|
#include "eden/fs/inodes/TreeInode.h"
|
|
#include "eden/fs/telemetry/TaskTrace.h"
|
|
|
|
using folly::StringPiece;
|
|
|
|
namespace facebook::eden {
|
|
|
|
namespace {
|
|
|
|
// Policy objects to help avoid duplicating the core globbing logic.
|
|
// We can walk over two different kinds of trees; either TreeInodes
|
|
// or raw Trees from the storage layer. While they have similar
|
|
// properties, accessing them is a little different. These policy
|
|
// objects are thin shims that make access more uniform.
|
|
|
|
/** TreeInodePtrRoot wraps a TreeInodePtr for globbing.
|
|
* TreeInodes require that a lock be held while its entries
|
|
* are iterated.
|
|
* We only need to prefetch children of TreeInodes that are
|
|
* not materialized.
|
|
*/
|
|
struct TreeInodePtrRoot {
|
|
TreeInodePtr root;
|
|
|
|
explicit TreeInodePtrRoot(TreeInodePtr root) : root(std::move(root)) {}
|
|
|
|
/** Return an object that holds a lock over the children */
|
|
folly::Synchronized<TreeInodeState>::RLockedPtr lockContents() {
|
|
return root->getContents().rlock();
|
|
}
|
|
|
|
/** Given the return value from lockContents and a name,
|
|
* return a pointer to the child with that name, or nullptr
|
|
* if there is no match */
|
|
template <typename CONTENTS>
|
|
typename DirContents::const_pointer FOLLY_NULLABLE
|
|
lookupEntry(CONTENTS& contents, PathComponentPiece name) {
|
|
auto it = contents->entries.find(name);
|
|
if (it != contents->entries.end()) {
|
|
return &*it;
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
/** Return an object that can be used in a generic for()
|
|
* constructor to iterate over the contents. You must supply
|
|
* the CONTENTS object you obtained via lockContents().
|
|
* The returned iterator yields ENTRY elements that can be
|
|
* used with the entryXXX methods below. */
|
|
const DirContents& iterate(
|
|
const folly::Synchronized<TreeInodeState>::RLockedPtr& contents) const {
|
|
return contents->entries;
|
|
}
|
|
|
|
/** Arrange to load a child TreeInode */
|
|
ImmediateFuture<TreeInodePtr> getOrLoadChildTree(
|
|
PathComponentPiece name,
|
|
const ObjectFetchContextPtr& context) {
|
|
return root->getOrLoadChildTree(name, context);
|
|
}
|
|
/** Returns true if we should call getOrLoadChildTree() for the given
|
|
* ENTRY. We only do this if the child is already materialized */
|
|
bool entryShouldLoadChildTree(const DirEntry* entry) {
|
|
return entry->isMaterialized();
|
|
}
|
|
|
|
/** Returns true if the given entry is a tree */
|
|
bool entryIsTree(const DirEntry* entry) {
|
|
return entry->isDirectory();
|
|
}
|
|
|
|
/** Returns true if we should prefetch the blob content for the entry.
|
|
* We only do this if the child is not already materialized */
|
|
bool entryShouldPrefetch(const DirEntry* entry) {
|
|
return !entry->isMaterialized() && !entryIsTree(entry);
|
|
}
|
|
};
|
|
} // namespace
|
|
|
|
ImmediateFuture<folly::Unit> GlobNode::evaluate(
|
|
std::shared_ptr<ObjectStore> store,
|
|
const ObjectFetchContextPtr& context,
|
|
RelativePathPiece rootPath,
|
|
TreeInodePtr root,
|
|
PrefetchList* fileBlobsToPrefetch,
|
|
ResultList& globResult,
|
|
const RootId& originRootId) const {
|
|
return evaluateImpl<TreeInodePtrRoot, TreeInodePtr>(
|
|
store.get(),
|
|
context,
|
|
rootPath,
|
|
TreeInodePtrRoot(std::move(root)),
|
|
fileBlobsToPrefetch,
|
|
globResult,
|
|
originRootId)
|
|
// Make sure the store stays alive for the duration of globbing.
|
|
.ensure([store] {});
|
|
}
|
|
|
|
} // namespace facebook::eden
|