NetworkStreamReceiver.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "NetworkResponse.hpp"
#include "AstUtil/Logger.hpp"
#include <map>
#include <string>
#include <functional>
#include <cstdint>
 
AST_NAMESPACE_BEGIN
 
class NetworkResponse;
 
class AST_UTIL_API NetworkStreamReceiver
{
public:
    NetworkStreamReceiver() = default;
    virtual ~NetworkStreamReceiver() = default;
 
    virtual errc_t onHeaders(int statusCode, const std::map<std::string, std::string>& headers) { return eNoError; }
 
    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), body_() {}
 
    errc_t onHeaders(int statusCode, const std::map<std::string, std::string>& headers) override;
    errc_t onData(const char* data, size_t size) override;
    void onComplete() override;
    void onError(errc_t /*error*/) override;
private:
    NetworkResponse& response_;
    std::string body_;
};
 
using DownloadProgressCallback = std::function<bool(uint64_t downloaded, uint64_t total)>;
 
 
 
class AST_UTIL_API FileDownloadReceiver : public NetworkStreamReceiver
{
public:
    FileDownloadReceiver(const std::string& filepath, const DownloadProgressCallback& progress);
 
    ~FileDownloadReceiver() override;
 
    FileDownloadReceiver(const FileDownloadReceiver& other) = delete;
    FileDownloadReceiver& operator=(const FileDownloadReceiver& other) = delete;
 
    errc_t onHeaders(int statusCode, const std::map<std::string, std::string>& headers) override;
    errc_t onData(const char* data, size_t size) override;
    void onError(errc_t /*error*/) override{}
 
    uint64_t downloaded() const { return downloaded_; }
    uint64_t total() const { return total_; }
 
    errc_t finish();
 
    void discard();
 
private:
    void close();
 
 
    std::string filepath_;
    std::string tempPath_;
    DownloadProgressCallback progress_;
    FILE* fp_ = nullptr;
    uint64_t total_ = 0;
    uint64_t downloaded_ = 0;
};
 
AST_NAMESPACE_END