task_inherited_data.hpp

Go to the documentation of this file.

#pragma once
 
/// @file userver/server/request/task_inherited_data.hpp
/// @brief @copybrief server::request::TaskInheritedData
 
#include <atomic>
#include <chrono>
#include <optional>
#include <string_view>
 
#include <userver/engine/deadline.hpp>
#include <userver/engine/task/inherited_variable.hpp>
#include <userver/server/request/impl/absolute_deadline.hpp>
 
USERVER_NAMESPACE_BEGIN
 
/// Server request related types and functions
namespace server::request {
 
/// @brief Signals when an operation has detected deadline expiration.
class DeadlineSignal final {
public:
    DeadlineSignal() noexcept;
    DeadlineSignal(const DeadlineSignal&) noexcept;
    DeadlineSignal& operator=(const DeadlineSignal&) noexcept;
 
    void SetExpired() noexcept;
    bool IsExpired() const noexcept;
 
private:
    std::atomic<bool> value_{false};
};
 
/// @brief Per-request data that should be available inside handlers
struct TaskInheritedData final {
    /// The static path of the handler
    std::string_view path;
 
    /// The method of the request
    std::string_view method;
 
    /// The time when the request started being handled
    std::chrono::steady_clock::time_point start_time{};
 
    /// The time when there is no use handling the request anymore
    engine::Deadline deadline;
 
    /// Signals when an operation has detected deadline expiration
    mutable DeadlineSignal deadline_signal{};
 
    /// Original absolute deadline from `X-Request-Deadline` (Unix epoch microseconds).
    /// Intended solely for propagation to downstream services; use @ref deadline for the task deadline.
    std::optional<TaskInheritedOriginalDeadline> original_deadline = std::nullopt;
};
 
/// @see TaskInheritedData for details on the contents.
extern engine::TaskInheritedVariable<TaskInheritedData> kTaskInheritedData;
 
/// @brief Returns TaskInheritedData::deadline, or an unreachable
/// engine::Deadline if none was set.
/// @note Use this method to set the deadline for tasks.
engine::Deadline GetTaskInheritedDeadline() noexcept;
 
/// @brief Returns TaskInheritedData::original_deadline, or std::nullopt if the header was absent or invalid.
/// @warning Do not use this method to set the deadline for tasks. It is intended for deadline propagation only.
/// Use @ref GetTaskInheritedDeadline() to set the deadline for tasks.
std::optional<TaskInheritedOriginalDeadline> GetTaskInheritedOriginalDeadline() noexcept;
 
/// @brief Marks that the current TaskInheritedData::deadline has expired.
void MarkTaskInheritedDeadlineExpired() noexcept;
 
/// @brief Stops deadline propagation within its scope
///
/// By default, handler deadline is honored in requests created directly
/// from the handler task, as well as from its child tasks. However, some
/// requests need to be completed regardless of whether the initial request
/// timed out, because they are needed for something other than forming the
/// upstream response.
///
/// Deadline propagation is automatically blocked in tasks launched using:
/// @see concurrent::BackgroundTaskStorage::AsyncDetach
/// @see utils::AsyncBackground
///
/// @see concurrent::BackgroundTaskStorage::AsyncDetach does it by default.
class [[nodiscard]] DeadlinePropagationBlocker final {
public:
    DeadlinePropagationBlocker();
 
    DeadlinePropagationBlocker(DeadlinePropagationBlocker&&) = delete;
    DeadlinePropagationBlocker& operator=(DeadlinePropagationBlocker&&) = delete;
    ~DeadlinePropagationBlocker();
 
private:
    TaskInheritedData old_value_;
};
 
}  // namespace server::request
 
USERVER_NAMESPACE_END