NetworkStreamReceiver.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "NetworkResponse.hpp"
#include <map>
#include <string>
 
AST_NAMESPACE_BEGIN
 
class NetworkResponse;
 
class AST_UTIL_API NetworkStreamReceiver
{
public:
    NetworkStreamReceiver() = default;
    virtual ~NetworkStreamReceiver() = default;
 
    virtual void onHeaders(int statusCode,
                           const std::map<std::string, std::string>& headers)
    {
    }
 
    virtual errc_t onData(const char* data, size_t size) = 0;
 
    virtual void onError(errc_t error) {}
 
    virtual void onComplete() {}
};
 
 
class AST_UTIL_API CollectingStreamReceiver : public NetworkStreamReceiver
{
public:
    explicit CollectingStreamReceiver(NetworkResponse& response)
        : response_(response) {}
 
    void onHeaders(int statusCode,
                   const std::map<std::string, std::string>& headers) override
    {
        response_.setStatusCode(statusCode);
        response_.setHeaders(headers);
    }
 
    errc_t onData(const char* data, size_t size) override
    {
        // 防止无限制响应体耗尽内存（上限 100 MB）
        if (body_.size() + size > kMaxBodySize)
            return eErrorOutOfRange;
        body_.append(data, size);
        return 0;
    }
 
    void onComplete() override
    {
        response_.setBody(body_);
    }
 
    void onError(errc_t /*error*/) override
    {
        // 保留已累积的部分体数据（用于诊断）
        if (!body_.empty())
            response_.setBody(body_);
    }
 
private:
    static constexpr size_t kMaxBodySize = 100 * 1024 * 1024;  // 100 MB
    NetworkResponse& response_;
    std::string body_;
};
 
AST_NAMESPACE_END