sapling/eden/fs/testharness/TestDispatcher.cpp
Xavier Deguillard 9c73da76f9 fuse: use ImmediateFuture in the Fuse dispatcher
Summary:
Similarly to the NFS change, this moves all the folly::Future handling in the
FuseChannel itself instead of inside the dispatcher. This should allow the
inode code to be converted to using ImmediateFuture instead of folly::Future.

Reviewed By: genevievehelsel

Differential Revision: D28372252

fbshipit-source-id: 7aae29d4a32500513c0062dfa42b2c7a3be038db
2021-05-17 14:07:29 -07:00

71 lines
2.1 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.
*/
#ifndef _WIN32
#include "eden/fs/testharness/TestDispatcher.h"
#include <folly/Conv.h>
#include <folly/logging/xlog.h>
using std::string;
namespace facebook {
namespace eden {
ImmediateFuture<fuse_entry_out> TestDispatcher::lookup(
uint64_t requestID,
InodeNumber parent,
PathComponentPiece name,
ObjectFetchContext& /*context*/) {
XLOG(DBG5) << "received lookup " << requestID << ": parent=" << parent
<< ", name=" << name;
ImmediateFuture<fuse_entry_out> result{};
{
// Whenever we receive a lookup request just add it to the pendingLookups_
// The test harness can then respond to it later however it wants.
auto state = state_.lock();
auto emplaceResult =
state->pendingLookups.emplace(requestID, PendingLookup(parent, name));
// We expect the test code to generate unique request IDs,
// just like the kernel should.
XCHECK(emplaceResult.second) << "received duplicate request ID "
<< requestID << " from the test harness";
result = emplaceResult.first->second.promise.getSemiFuture();
}
requestReceived_.notify_all();
return result;
}
TestDispatcher::PendingLookup TestDispatcher::waitForLookup(
uint64_t requestId,
std::chrono::milliseconds timeout) {
auto state = state_.lock();
auto end_time = std::chrono::steady_clock::now() + timeout;
while (true) {
auto iter = state->pendingLookups.find(requestId);
if (iter != state->pendingLookups.end()) {
PendingLookup result(std::move(iter->second));
state->pendingLookups.erase(iter);
return result;
}
if (requestReceived_.wait_until(state.getUniqueLock(), end_time) ==
std::cv_status::timeout) {
throw std::runtime_error(folly::to<string>(
"timed out waiting for test dispatcher to receive lookup request ",
requestId));
}
}
}
} // namespace eden
} // namespace facebook
#endif