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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
/**
* Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
* SPDX-License-Identifier: Apache-2.0.
*/
#include <aws/core/http/standard/StandardHttpRequest.h>
#include <aws/core/utils/logging/LogMacros.h>
#include <aws/core/utils/StringUtils.h>
#include <iostream>
#include <algorithm>
#include <cassert>
using namespace Aws::Http;
using namespace Aws::Http::Standard;
using namespace Aws::Utils;
static const char* STANDARD_HTTP_REQUEST_LOG_TAG = "StandardHttpRequest";
static bool IsDefaultPort(const URI& uri)
{
switch(uri.GetPort())
{
case 80:
return uri.GetScheme() == Scheme::HTTP;
case 443:
return uri.GetScheme() == Scheme::HTTPS;
default:
return false;
}
}
StandardHttpRequest::StandardHttpRequest(const URI& uri, HttpMethod method) :
HttpRequest(uri, method),
bodyStream(nullptr),
m_responseStreamFactory()
{
if(IsDefaultPort(uri))
{
StandardHttpRequest::SetHeaderValue(HOST_HEADER, uri.GetAuthority());
}
else
{
Aws::StringStream host;
host << uri.GetAuthority() << ":" << uri.GetPort();
StandardHttpRequest::SetHeaderValue(HOST_HEADER, host.str());
}
}
HeaderValueCollection StandardHttpRequest::GetHeaders() const
{
HeaderValueCollection headers;
for (HeaderValueCollection::const_iterator iter = headerMap.begin(); iter != headerMap.end(); ++iter)
{
headers.emplace(HeaderValuePair(iter->first, iter->second));
}
return headers;
}
const Aws::String& StandardHttpRequest::GetHeaderValue(const char* headerName) const
{
auto iter = headerMap.find(StringUtils::ToLower(headerName));
assert (iter != headerMap.end());
if (iter == headerMap.end()) {
AWS_LOGSTREAM_ERROR(STANDARD_HTTP_REQUEST_LOG_TAG, "Requested a header value for a missing header key: " << headerName);
static const Aws::String EMPTY_STRING = "";
return EMPTY_STRING;
}
return iter->second;
}
void StandardHttpRequest::SetHeaderValue(const char* headerName, const Aws::String& headerValue)
{
headerMap[StringUtils::ToLower(headerName)] = StringUtils::Trim(headerValue.c_str());
}
void StandardHttpRequest::SetHeaderValue(const Aws::String& headerName, const Aws::String& headerValue)
{
headerMap[StringUtils::ToLower(headerName.c_str())] = StringUtils::Trim(headerValue.c_str());
}
void StandardHttpRequest::DeleteHeader(const char* headerName)
{
headerMap.erase(StringUtils::ToLower(headerName));
}
bool StandardHttpRequest::HasHeader(const char* headerName) const
{
return headerMap.find(StringUtils::ToLower(headerName)) != headerMap.end();
}
int64_t StandardHttpRequest::GetSize() const
{
int64_t size = 0;
std::for_each(headerMap.cbegin(), headerMap.cend(), [&](const HeaderValueCollection::value_type& kvPair){ size += kvPair.first.length(); size += kvPair.second.length(); });
return size;
}
const Aws::IOStreamFactory& StandardHttpRequest::GetResponseStreamFactory() const
{
return m_responseStreamFactory;
}
void StandardHttpRequest::SetResponseStreamFactory(const Aws::IOStreamFactory& factory)
{
m_responseStreamFactory = factory;
}
|
