library/cpp/messagebus/rain_check/http/client.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78

#pragma once

#include <library/cpp/messagebus/rain_check/core/task.h>

#include <library/cpp/http/misc/httpcodes.h>
 
#include <util/generic/maybe.h>
#include <util/generic/ptr.h>
#include <util/generic/string.h>
#include <util/system/defaults.h>
#include <util/system/yassert.h>

class THttpHeaders;

namespace NNeh {
    class IProtocol;
    struct TResponse;
}

namespace NRainCheck {
    class THttpCallback;
    class THttpClientService;

    class THttpFuture: public TSubtaskCompletion {
    public:
        enum EError {
            NoError = 0,

            CantResolveNameError = 1,
            BadHttpCodeError = 2,

            OtherError = 100
        };

    private:
        friend class THttpCallback;
        friend class THttpClientService;

    public:
        THttpFuture();
        ~THttpFuture() override;

        bool HasHttpCode() const;
        bool HasResponseBody() const;

        ui32 GetHttpCode() const;
        TString GetResponseBody() const;

        bool HasError() const;
        EError GetErrorCode() const;
        TString GetErrorDescription() const;

    private:
        void SetDoneAndSchedule(TAutoPtr<NNeh::TResponse> response);
        void SetFail(EError errorCode, const TStringBuf& errorDescription);

    private:
        TTaskRunnerBase* Task;
        TMaybe<HttpCodes> HttpCode;
        THolder<NNeh::TResponse> Response;
        EError ErrorCode;
        TString ErrorDescription;
    };

    class THttpClientService {
    public:
        THttpClientService();
        virtual ~THttpClientService();

        void Send(const TString& request, THttpFuture* future);
        void SendPost(TString addr, const TString& data, const THttpHeaders& headers, THttpFuture* future);

    private:
        NNeh::IProtocol* const GetProtocol;
        NNeh::IProtocol* const FullProtocol;
    };

}