library/cpp/http/simple/http_client_options.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
79
80
81
82
83
84
85
86
87
88
89
90

#pragma once

#include <util/datetime/base.h>
#include <util/string/cast.h>
#include <library/cpp/string_utils/url/url.h>

class TSimpleHttpClientOptions {
    using TSelf = TSimpleHttpClientOptions;

public:
    TSimpleHttpClientOptions() = default;

    explicit TSimpleHttpClientOptions(TStringBuf url) {
        TStringBuf scheme, host;
        GetSchemeHostAndPort(url, scheme, host, Port_);
        Host_ = url.Head(scheme.size() + host.size());
    }

    TSelf& Host(TStringBuf host) {
        Host_ = host;
        return *this;
    }

    const TString& Host() const noexcept {
        return Host_;
    }

    TSelf& Port(ui16 port) {
        Port_ = port;
        return *this;
    }

    ui16 Port() const noexcept {
        return Port_;
    }

    TSelf& SocketTimeout(TDuration timeout) {
        SocketTimeout_ = timeout;
        return *this;
    }

    TDuration SocketTimeout() const noexcept {
        return SocketTimeout_;
    }

    TSelf& ConnectTimeout(TDuration timeout) {
        ConnectTimeout_ = timeout;
        return *this;
    }

    TDuration ConnectTimeout() const noexcept {
        return ConnectTimeout_;
    }

    TSelf& MaxRedirectCount(int count) {
        MaxRedirectCount_ = count;
        return *this;
    }

    ui16 MaxRedirectCount() const noexcept {
        return MaxRedirectCount_;
    }

    TSelf& UseKeepAlive(bool useKeepAlive) {
        UseKeepAlive_ = useKeepAlive;
        return *this;
    }

    bool UseKeepAlive() const noexcept {
        return UseKeepAlive_;
    }

    TSelf& UseConnectionPool(bool useConnectionPool) {
        UseConnectionPool_ = useConnectionPool;
        return *this;
    }

    bool UseConnectionPool() const noexcept {
        return UseConnectionPool_;
    }

private:
    TString Host_;
    ui16 Port_;
    TDuration SocketTimeout_ = TDuration::Seconds(5);
    TDuration ConnectTimeout_ = TDuration::Seconds(30);
    int MaxRedirectCount_ = INT_MAX;
    bool UseKeepAlive_ = true;
    bool UseConnectionPool_ = false;
};