tdlight/tdnet/td/net/HttpQuery.cpp
levlam c022b1c5a1 Use td::vector in HttpQuery.
GitOrigin-RevId: ecef0bd11150712bdb45e59a33cd0706dda2097d
2020-06-13 23:36:10 +03:00

87 lines
2.3 KiB
C++

//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2020
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
#include "td/net/HttpQuery.h"
#include "td/utils/misc.h"
#include <algorithm>
namespace td {
Slice HttpQuery::get_header(Slice key) const {
auto it = std::find_if(headers_.begin(), headers_.end(),
[&key](const std::pair<MutableSlice, MutableSlice> &s) { return s.first == key; });
return it == headers_.end() ? Slice() : it->second;
}
MutableSlice HttpQuery::get_arg(Slice key) const {
auto it = std::find_if(args_.begin(), args_.end(),
[&key](const std::pair<MutableSlice, MutableSlice> &s) { return s.first == key; });
return it == args_.end() ? MutableSlice() : it->second;
}
td::vector<std::pair<string, string>> HttpQuery::get_args() const {
td::vector<std::pair<string, string>> res;
res.reserve(args_.size());
for (auto &it : args_) {
res.emplace_back(it.first.str(), it.second.str());
}
return res;
}
int HttpQuery::get_retry_after() const {
auto value = get_header("retry-after");
if (value.empty()) {
return 0;
}
auto r_retry_after = to_integer_safe<int>(value);
if (r_retry_after.is_error()) {
return 0;
}
return td::max(0, r_retry_after.ok());
}
StringBuilder &operator<<(StringBuilder &sb, const HttpQuery &q) {
switch (q.type_) {
case HttpQuery::Type::EMPTY:
sb << "EMPTY";
return sb;
case HttpQuery::Type::GET:
sb << "GET";
break;
case HttpQuery::Type::POST:
sb << "POST";
break;
case HttpQuery::Type::RESPONSE:
sb << "RESPONSE";
break;
}
if (q.type_ == HttpQuery::Type::RESPONSE) {
sb << ":" << q.code_ << ":" << q.reason_;
} else {
sb << ":" << q.url_path_;
for (auto &key_value : q.args_) {
sb << ":[" << key_value.first << ":" << key_value.second << "]";
}
}
if (q.keep_alive_) {
sb << ":keep-alive";
}
sb << "\n";
for (auto &key_value : q.headers_) {
sb << key_value.first << "=" << key_value.second << "\n";
}
sb << "BEGIN CONTENT\n";
sb << q.content_;
sb << "END CONTENT\n";
return sb;
}
} // namespace td