Commit dd741a58 authored by Tatsuhiro Tsujikawa's avatar Tatsuhiro Tsujikawa

Use header_map instead of wrapping it

parent 26304546
......@@ -36,8 +36,8 @@ using boost::asio::ip::tcp;
using namespace nghttp2::asio_http2;
using namespace nghttp2::asio_http2::client;
void print_header(const http_header &h) {
for (auto &kv : h.items()) {
void print_header(const header_map &h) {
for (auto &kv : h) {
std::cerr << kv.first << ": " << kv.second.value << "\n";
}
std::cerr << std::endl;
......
......@@ -54,7 +54,7 @@ const std::string &request::authority() const { return impl_->authority(); }
const std::string &request::host() const { return impl_->host(); }
const http_header &request::header() const { return impl_->header(); }
const header_map &request::header() const { return impl_->header(); }
request_impl &request::impl() { return *impl_; }
......
......@@ -68,11 +68,11 @@ read_cb::result_type request_impl::call_on_read(uint8_t *buf, std::size_t len) {
return read_cb::result_type{};
}
void request_impl::header(http_header h) { header_ = std::move(h); }
void request_impl::header(header_map h) { header_ = std::move(h); }
http_header &request_impl::header() { return header_; }
header_map &request_impl::header() { return header_; }
const http_header &request_impl::header() const { return header_; }
const header_map &request_impl::header() const { return header_; }
void request_impl::stream(class stream *strm) { strm_ = strm; }
......
......@@ -57,9 +57,9 @@ public:
void on_read(read_cb cb);
read_cb::result_type call_on_read(uint8_t *buf, std::size_t len);
void header(http_header h);
http_header &header();
const http_header &header() const;
void header(header_map h);
header_map &header();
const header_map &header() const;
void stream(class stream *strm);
......@@ -79,7 +79,7 @@ public:
const std::string &host() const;
private:
http_header header_;
header_map header_;
response_cb response_cb_;
request_cb push_request_cb_;
close_cb close_cb_;
......
......@@ -42,7 +42,7 @@ int response::status_code() const { return impl_->status_code(); }
int64_t response::content_length() const { return impl_->content_length(); }
const http_header &response::header() const { return impl_->header(); }
const header_map &response::header() const { return impl_->header(); }
response_impl &response::impl() { return *impl_; }
......
......@@ -48,9 +48,9 @@ void response_impl::content_length(int64_t n) { content_length_ = n; }
int64_t response_impl::content_length() const { return content_length_; }
http_header &response_impl::header() { return header_; }
header_map &response_impl::header() { return header_; }
const http_header &response_impl::header() const { return header_; }
const header_map &response_impl::header() const { return header_; }
} // namespace client
} // namespace asio_http2
......
......@@ -50,13 +50,13 @@ public:
void content_length(int64_t n);
int64_t content_length() const;
http_header &header();
const http_header &header() const;
header_map &header();
const header_map &header() const;
private:
data_cb data_cb_;
http_header header_;
header_map header_;
int64_t content_length_;
int status_code_;
......
......@@ -56,20 +56,20 @@ void session::shutdown() { impl_->shutdown(); }
request *session::submit(boost::system::error_code &ec,
const std::string &method, const std::string &uri,
http_header h) {
header_map h) {
return impl_->submit(ec, method, uri, read_cb(), std::move(h));
}
request *session::submit(boost::system::error_code &ec,
const std::string &method, const std::string &uri,
std::string data, http_header h) {
std::string data, header_map h) {
return impl_->submit(ec, method, uri, string_reader(std::move(data)),
std::move(h));
}
request *session::submit(boost::system::error_code &ec,
const std::string &method, const std::string &uri,
read_cb cb, http_header h) {
read_cb cb, header_map h) {
return impl_->submit(ec, method, uri, std::move(cb), std::move(h));
}
......
......@@ -134,9 +134,9 @@ int on_header_callback(nghttp2_session *session, const nghttp2_frame *frame,
res.content_length(util::parse_uint(value, valuelen));
}
res.header().add(std::string(name, name + namelen),
std::string(value, value + valuelen),
flags & NGHTTP2_NV_FLAG_NO_INDEX);
res.header().emplace(std::string(name, name + namelen),
header_value(std::string(value, value + valuelen),
flags & NGHTTP2_NV_FLAG_NO_INDEX));
}
break;
}
......@@ -167,9 +167,9 @@ int on_header_callback(nghttp2_session *session, const nghttp2_frame *frame,
req.host(std::string(value, value + valuelen));
// fall through
default:
req.header().add(std::string(name, name + namelen),
std::string(value, value + valuelen),
flags & NGHTTP2_NV_FLAG_NO_INDEX);
req.header().emplace(std::string(name, name + namelen),
header_value(std::string(value, value + valuelen),
flags & NGHTTP2_NV_FLAG_NO_INDEX));
}
break;
......@@ -345,7 +345,7 @@ std::unique_ptr<stream> session_impl::create_stream() {
request *session_impl::submit(boost::system::error_code &ec,
const std::string &method, const std::string &uri,
read_cb cb, http_header h) {
read_cb cb, header_map h) {
ec.clear();
auto nva = std::vector<nghttp2_nv>();
......@@ -354,7 +354,7 @@ request *session_impl::submit(boost::system::error_code &ec,
nva.push_back(http2::make_nv_ll(":scheme", "https"));
nva.push_back(http2::make_nv_ll(":path", "/"));
nva.push_back(http2::make_nv_ll(":authority", "localhost:3000"));
for (auto &kv : h.items()) {
for (auto &kv : h) {
nva.push_back(
http2::make_nv(kv.first, kv.second.value, kv.second.sensitive));
}
......
......@@ -65,7 +65,7 @@ public:
stream *find_stream(int32_t stream_id);
request *submit(boost::system::error_code &ec, const std::string &method,
const std::string &uri, read_cb cb, http_header h);
const std::string &uri, read_cb cb, header_map h);
virtual tcp::socket &socket() = 0;
virtual void read_socket(std::function<
......
......@@ -57,46 +57,5 @@ read_cb string_reader(std::string data) {
};
}
http_header::http_header() {}
http_header::http_header(
std::initializer_list<std::pair<std::string, header_value>> ilist) {
for (auto &kv : ilist) {
auto name = kv.first;
util::inp_strlower(name);
items_.emplace(std::move(name), kv.second);
}
}
http_header &http_header::
operator=(std::initializer_list<std::pair<std::string, header_value>> ilist) {
items_.clear();
for (auto &kv : ilist) {
auto name = kv.first;
util::inp_strlower(name);
items_.emplace(std::move(name), kv.second);
}
return *this;
}
const header_map &http_header::items() const { return items_; }
void http_header::add(std::string name, std::string value, bool sensitive) {
util::inp_strlower(name);
items_.emplace(name, header_value(value, sensitive));
}
const header_value *http_header::get(const std::string &name) const {
auto it = items_.find(name);
if (it == std::end(items_)) {
return nullptr;
}
return &(*it).second;
}
std::size_t http_header::size() const { return items_.size(); }
bool http_header::empty() const { return items_.empty(); }
} // namespace asio_http2
} // namespace nghttp2
......@@ -72,31 +72,6 @@ struct header_value {
using header_map = std::multimap<std::string, header_value>;
class http_header {
public:
http_header();
http_header(const http_header &other) = default;
http_header(http_header &&other) = default;
http_header(
std::initializer_list<std::pair<std::string, header_value>> ilist);
http_header &operator=(const http_header &other) = default;
http_header &operator=(http_header &&other) = default;
http_header &
operator=(std::initializer_list<std::pair<std::string, header_value>> ilist);
const header_map &items() const;
void add(std::string name, std::string value, bool sensitive);
const header_value *get(const std::string &name) const;
std::size_t size() const;
bool empty() const;
private:
header_map items_;
};
const boost::system::error_category &nghttp2_category() noexcept;
typedef std::function<void(const uint8_t *, std::size_t)> data_cb;
......@@ -318,7 +293,7 @@ public:
int64_t content_length() const;
const http_header &header() const;
const header_map &header() const;
response_impl &impl();
......@@ -350,7 +325,7 @@ public:
const std::string &authority() const;
const std::string &host() const;
const http_header &header() const;
const header_map &header() const;
request_impl &impl();
......@@ -375,11 +350,11 @@ public:
void shutdown();
request *submit(boost::system::error_code &ec, const std::string &method,
const std::string &uri, http_header h = {});
const std::string &uri, header_map h = {});
request *submit(boost::system::error_code &ec, const std::string &method,
const std::string &uri, std::string data, http_header h = {});
const std::string &uri, std::string data, header_map h = {});
request *submit(boost::system::error_code &ec, const std::string &method,
const std::string &uri, read_cb cb, http_header h = {});
const std::string &uri, read_cb cb, header_map h = {});
private:
std::unique_ptr<session_impl> impl_;
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment