mirror of
https://github.com/facebook/sapling.git
synced 2024-10-05 22:37:37 +03:00
53b9dc5d3f
Summary:
Original commit changeset: abc1785f194a
Original Phabricator Diff: D40349610 (1eed4f0063)
Reviewed By: xavierd
Differential Revision: D40731896
fbshipit-source-id: 632c08fd3e44e8d2a126fdf688f41803354641a4
136 lines
4.2 KiB
C++
136 lines
4.2 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.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <folly/futures/Future.h>
|
|
#include <atomic>
|
|
#include <utility>
|
|
|
|
#include "eden/fs/fuse/FuseChannel.h"
|
|
#include "eden/fs/inodes/RequestContext.h"
|
|
#include "eden/fs/store/ImportPriority.h"
|
|
#include "eden/fs/store/ObjectFetchContext.h"
|
|
#include "eden/fs/telemetry/EdenStats.h"
|
|
#include "eden/fs/telemetry/RequestMetricsScope.h"
|
|
#include "eden/fs/utils/FsChannelTypes.h"
|
|
|
|
namespace facebook::eden {
|
|
|
|
/**
|
|
* Each FUSE request has a corresponding FuseRequestContext object that is
|
|
* allocated at request start and deallocated when it finishes.
|
|
*
|
|
* Unless a member function indicates otherwise, FuseRequestContext may be used
|
|
* from multiple threads, but only by one thread at a time.
|
|
*/
|
|
class FuseRequestContext : public RequestContext {
|
|
public:
|
|
FuseRequestContext(const FuseRequestContext&) = delete;
|
|
FuseRequestContext& operator=(const FuseRequestContext&) = delete;
|
|
FuseRequestContext(FuseRequestContext&&) = delete;
|
|
FuseRequestContext& operator=(FuseRequestContext&&) = delete;
|
|
explicit FuseRequestContext(
|
|
FuseChannel* channel,
|
|
const fuse_in_header& fuseHeader);
|
|
|
|
// Override of `ObjectFetchContext`
|
|
std::optional<pid_t> getClientPid() const override {
|
|
return static_cast<pid_t>(fuseHeader_.pid);
|
|
}
|
|
|
|
// Override of `ObjectFetchContext`
|
|
std::optional<std::string_view> getCauseDetail() const override {
|
|
return fuseOpcodeName(fuseHeader_.opcode);
|
|
}
|
|
|
|
/**
|
|
* After sendReply or replyError, this returns the error code we returned to
|
|
* the kernel, negated.
|
|
*
|
|
* After sendReplyWithInode, this returns the inode number that the kernel
|
|
* will reference until it sends FORGET.
|
|
*/
|
|
const std::optional<int64_t>& getResult() const {
|
|
return result_;
|
|
}
|
|
|
|
/**
|
|
* Returns the underlying fuse request, throwing an error if it has
|
|
* already been released
|
|
*/
|
|
const fuse_in_header& getReq() const;
|
|
|
|
/**
|
|
* Append error handling clauses to a future chain. These clauses result in
|
|
* reporting a fuse request error back to the kernel.
|
|
*/
|
|
folly::Future<folly::Unit> catchErrors(
|
|
folly::Future<folly::Unit>&& fut,
|
|
Notifier* FOLLY_NULLABLE notifier) {
|
|
return std::move(fut).thenTryInline([this, notifier](
|
|
folly::Try<folly::Unit>&& try_) {
|
|
if (try_.hasException()) {
|
|
if (auto* err = try_.tryGetExceptionObject<folly::FutureTimeout>()) {
|
|
timeoutErrorHandler(*err, notifier);
|
|
} else if (
|
|
auto* err = try_.tryGetExceptionObject<std::system_error>()) {
|
|
systemErrorHandler(*err, notifier);
|
|
} else if (auto* err = try_.tryGetExceptionObject<std::exception>()) {
|
|
genericErrorHandler(*err, notifier);
|
|
} else {
|
|
genericErrorHandler(
|
|
std::runtime_error{"unknown exception type"}, notifier);
|
|
}
|
|
}
|
|
});
|
|
}
|
|
|
|
void systemErrorHandler(
|
|
const std::system_error& err,
|
|
Notifier* FOLLY_NULLABLE notifier);
|
|
void genericErrorHandler(
|
|
const std::exception& err,
|
|
Notifier* FOLLY_NULLABLE notifier);
|
|
void timeoutErrorHandler(
|
|
const folly::FutureTimeout& err,
|
|
Notifier* FOLLY_NULLABLE notifier);
|
|
|
|
template <typename... T>
|
|
void sendReply(T&&... payload) {
|
|
channel_->sendReply(stealReqWithResult(0), std::forward<T>(payload)...);
|
|
}
|
|
|
|
/**
|
|
* Same as sendReply, but is called when the kernel will take a reference to
|
|
* the returned inode. The returned inode value will be logged to make trace
|
|
* logs more useful.
|
|
*/
|
|
template <typename T>
|
|
void sendReplyWithInode(uint64_t nodeid, T&& reply) {
|
|
channel_->sendReply(stealReqWithResult(nodeid), std::forward<T>(reply));
|
|
}
|
|
|
|
// Reply with a negative errno value or 0 for success
|
|
void replyError(int err);
|
|
|
|
// Don't send a reply, just release req_
|
|
void replyNone();
|
|
|
|
private:
|
|
// Returns the header and sets result_ to indicate
|
|
// that the request has been released.
|
|
fuse_in_header stealReqWithResult(int64_t result);
|
|
|
|
FuseChannel* channel_;
|
|
const fuse_in_header fuseHeader_;
|
|
|
|
std::optional<int64_t> result_;
|
|
};
|
|
|
|
} // namespace facebook::eden
|