You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
118 lines
3.3 KiB
C++
118 lines
3.3 KiB
C++
#include "webcc/http_client_session.h"
|
|
|
|
#include "webcc/url.h"
|
|
|
|
namespace webcc {
|
|
|
|
HttpClientSession::HttpClientSession() {
|
|
InitHeaders();
|
|
}
|
|
|
|
HttpResponsePtr HttpClientSession::Request(HttpRequestArgs&& args) {
|
|
assert(args.parameters_.size() % 2 == 0);
|
|
assert(args.headers_.size() % 2 == 0);
|
|
|
|
HttpRequest request{ args.method_, args.url_ };
|
|
|
|
for (std::size_t i = 1; i < args.parameters_.size(); i += 2) {
|
|
request.AddParameter(args.parameters_[i - 1], args.parameters_[i]);
|
|
}
|
|
|
|
if (!args.data_.empty()) {
|
|
request.SetContent(std::move(args.data_), true);
|
|
|
|
// TODO: Request-level charset.
|
|
if (args.json_) {
|
|
request.SetContentType(http::media_types::kApplicationJson, charset_);
|
|
} else if (!content_type_.empty()) {
|
|
request.SetContentType(content_type_, charset_);
|
|
}
|
|
}
|
|
|
|
// Apply the session-level headers.
|
|
for (const HttpHeader& h : headers_.data()) {
|
|
request.SetHeader(h.first, h.second);
|
|
}
|
|
|
|
// Apply the request-level headers.
|
|
// This will overwrite the session-level headers.
|
|
for (std::size_t i = 1; i < args.headers_.size(); i += 2) {
|
|
request.SetHeader(std::move(args.headers_[i - 1]),
|
|
std::move(args.headers_[i]));
|
|
}
|
|
|
|
request.Prepare();
|
|
|
|
// Determine SSL verify flag.
|
|
bool ssl_verify = true;
|
|
if (args.ssl_verify_) {
|
|
ssl_verify = args.ssl_verify_.value();
|
|
} else if (ssl_verify_) {
|
|
ssl_verify = ssl_verify_.value();
|
|
}
|
|
|
|
bool reuse = false;
|
|
const HttpClientPool::Key key{ request.url() };
|
|
|
|
HttpClientPtr client = pool_.Get(key);
|
|
if (!client) {
|
|
client.reset(new HttpClient{ 0, ssl_verify });
|
|
reuse = false;
|
|
} else {
|
|
// TODO: Apply args.ssl_verify even if reuse a client.
|
|
reuse = false;
|
|
LOG_VERB("Reuse an existing connection.");
|
|
}
|
|
|
|
if (!client->Request(request, args.buffer_size_, !reuse)) {
|
|
throw Exception(client->error(), client->timed_out());
|
|
}
|
|
|
|
// Update pool.
|
|
if (reuse) {
|
|
if (client->closed()) {
|
|
pool_.Remove(key);
|
|
}
|
|
} else {
|
|
if (!client->closed()) {
|
|
pool_.Add(key, client);
|
|
}
|
|
}
|
|
|
|
return client->response();
|
|
}
|
|
|
|
HttpResponsePtr HttpClientSession::Get(const std::string& url,
|
|
std::vector<std::string>&& parameters,
|
|
std::vector<std::string>&& headers,
|
|
HttpRequestArgs&& args) {
|
|
return Request(args.method(http::kGet)
|
|
.url(url)
|
|
.parameters(std::move(parameters))
|
|
.headers(std::move(headers)));
|
|
}
|
|
|
|
HttpResponsePtr HttpClientSession::Post(const std::string& url,
|
|
std::string&& data, bool json,
|
|
std::vector<std::string>&& headers,
|
|
HttpRequestArgs&& args) {
|
|
return Request(args.method(http::kPost)
|
|
.url(url)
|
|
.data(std::move(data))
|
|
.json(json)
|
|
.headers(std::move(headers)));
|
|
}
|
|
|
|
void HttpClientSession::InitHeaders() {
|
|
headers_.Add(http::headers::kUserAgent, http::UserAgent());
|
|
|
|
// TODO: Support gzip, deflate
|
|
headers_.Add(http::headers::kAcceptEncoding, "identity");
|
|
|
|
headers_.Add(http::headers::kAccept, "*/*");
|
|
|
|
headers_.Add(http::headers::kConnection, "Keep-Alive");
|
|
}
|
|
|
|
} // namespace webcc
|