request_base.hpp

#pragma once
 
#include <chrono>
#include <memory>
#include <string>
#include <vector>
 
// TODO: use fwd declarations
#include <userver/logging/fwd.hpp>
#include <userver/server/request/response_base.hpp>
 
USERVER_NAMESPACE_BEGIN
 
namespace engine::io {
class Socket;
class Sockaddr;
}  // namespace engine::io
 
namespace server::request {
 
class RequestBase {
public:
    RequestBase();
    virtual ~RequestBase();
 
    virtual bool IsFinal() const = 0;
    virtual bool IsUpgradeWebsocket() const = 0;
    virtual void DoUpgrade(std::unique_ptr<engine::io::RwBase>&& socket, engine::io::Sockaddr&& peer_name) const = 0;
 
    virtual ResponseBase& GetResponse() const = 0;
 
    virtual void WriteAccessLogs(
        const logging::LoggerPtr& logger_access,
        const logging::LoggerPtr& logger_access_tskv,
        const std::string& remote_address
    ) const = 0;
 
    virtual const std::string& GetRequestPath() const = 0;
 
    void SetTaskCreateTime();
    void SetTaskStartTime();
    void SetResponseNotifyTime();
    void SetResponseNotifyTime(std::chrono::steady_clock::time_point now);
    void SetStartSendResponseTime();
    void SetFinishSendResponseTime();
 
    virtual void SetMatchedPathLength(size_t length) = 0;
 
    std::chrono::steady_clock::time_point StartTime() const { return start_time_; }
 
    virtual void MarkAsInternalServerError() const = 0;
 
    virtual void AccountResponseTime() = 0;
 
protected:
    // NOLINTNEXTLINE(misc-non-private-member-variables-in-classes)
    std::chrono::steady_clock::time_point start_time_;
    // NOLINTNEXTLINE(misc-non-private-member-variables-in-classes)
    std::chrono::steady_clock::time_point task_create_time_;
    // NOLINTNEXTLINE(misc-non-private-member-variables-in-classes)
    std::chrono::steady_clock::time_point task_start_time_;
    // NOLINTNEXTLINE(misc-non-private-member-variables-in-classes)
    std::chrono::steady_clock::time_point response_notify_time_;
    // NOLINTNEXTLINE(misc-non-private-member-variables-in-classes)
    std::chrono::steady_clock::time_point start_send_response_time_;
    // NOLINTNEXTLINE(misc-non-private-member-variables-in-classes)
    std::chrono::steady_clock::time_point finish_send_response_time_;
};
 
}  // namespace server::request
 
USERVER_NAMESPACE_END