call_context.hpp

Go to the documentation of this file.

#pragma once
 
/// @file userver/ugrpc/server/call_context.hpp
/// @brief @copybrief ugrpc::server::CallContext
 
#include <grpcpp/server_context.h>
 
#include <userver/tracing/span.hpp>
#include <userver/ugrpc/server/storage_context.hpp>
#include <userver/utils/any_storage.hpp>
 
USERVER_NAMESPACE_BEGIN
 
namespace ugrpc::server {
 
class CallAnyBase;
 
/// @brief gRPC call context
class CallContext {
public:
    /// @cond
    explicit CallContext(CallAnyBase& call);
    /// @endcond
 
    /// @returns the `ServerContext` used for this RPC
    grpc::ServerContext& GetServerContext();
 
    /// @brief Name of the RPC in the format `full.path.ServiceName/MethodName`
    std::string_view GetCallName() const;
 
    /// @brief Get name of gRPC service
    std::string_view GetServiceName() const;
 
    /// @brief Get name of called gRPC method
    std::string_view GetMethodName() const;
 
    /// @brief Get the span of the current RPC
    tracing::Span& GetSpan();
 
    /// @brief Returns call context for storing per-call custom data
    ///
    /// The context can be used to pass data from server middleware to client
    /// handler or from one middleware to another one.
    ///
    /// ## Example usage:
    ///
    /// In authentication middleware:
    ///
    /// @code
    /// if (password_is_correct) {
    ///   // Username is authenticated, set it in per-call storage context
    ///   ctx.GetCall().GetStorageContext().Emplace(kAuthUsername, username);
    /// }
    /// @endcode
    ///
    /// In client handler:
    ///
    /// @code
    /// const auto& username = context.GetStorageContext().Get(kAuthUsername);
    /// auto msg = fmt::format("Hello, {}!", username);
    /// @endcode
    utils::AnyStorage<StorageContext>& GetStorageContext();
 
protected:
    /// @cond
    const CallAnyBase& GetCall() const;
 
    CallAnyBase& GetCall();
    /// @endcond
 
private:
    CallAnyBase& call_;
};
 
/// @brief generic gRPC call context
class GenericCallContext : public CallContext {
public:
    /// @cond
    using CallContext::CallContext;
    /// @endcond
 
    /// @brief Set a custom call name for metric labels
    void SetMetricsCallName(std::string_view call_name);
};
 
}  // namespace ugrpc::server
 
USERVER_NAMESPACE_END