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.
88 lines
2.2 KiB
C++
88 lines
2.2 KiB
C++
#ifndef WEBCC_HTTP_CLIENT_H_
|
|
#define WEBCC_HTTP_CLIENT_H_
|
|
|
|
#include <cassert>
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
#include "boost/asio/deadline_timer.hpp"
|
|
#include "boost/asio/io_context.hpp"
|
|
#include "boost/asio/ip/tcp.hpp"
|
|
|
|
#include "webcc/globals.h"
|
|
#include "webcc/http_request.h"
|
|
#include "webcc/http_response.h"
|
|
#include "webcc/http_response_parser.h"
|
|
|
|
namespace webcc {
|
|
|
|
// HTTP client session in synchronous mode.
|
|
// A request will not return until the response is received or timeout occurs.
|
|
// Don't use the same HttpClient object in multiple threads.
|
|
class HttpClient {
|
|
public:
|
|
// The |buffer_size| is the bytes of the buffer for reading response.
|
|
// 0 means default value (e.g., 1024) will be used.
|
|
explicit HttpClient(std::size_t buffer_size = 0);
|
|
|
|
~HttpClient() = default;
|
|
|
|
WEBCC_DELETE_COPY_ASSIGN(HttpClient);
|
|
|
|
// Set the timeout seconds for reading response.
|
|
// The |seconds| is only effective when greater than 0.
|
|
void SetTimeout(int seconds);
|
|
|
|
// Connect to server, send request, wait until response is received.
|
|
// Set |buffer_size| to non-zero to use a different buffer size for this
|
|
// specific request.
|
|
bool Request(const HttpRequest& request, std::size_t buffer_size = 0);
|
|
|
|
HttpResponsePtr response() const { return response_; }
|
|
|
|
bool timed_out() const { return timed_out_; }
|
|
|
|
Error error() const { return error_; }
|
|
|
|
private:
|
|
Error Connect(const HttpRequest& request);
|
|
|
|
Error SendReqeust(const HttpRequest& request);
|
|
|
|
Error ReadResponse();
|
|
|
|
void DoReadResponse(Error* error);
|
|
|
|
void DoWaitDeadline();
|
|
void OnDeadline(boost::system::error_code ec);
|
|
|
|
void Stop();
|
|
|
|
boost::asio::io_context io_context_;
|
|
boost::asio::ip::tcp::socket socket_;
|
|
|
|
std::vector<char> buffer_;
|
|
|
|
HttpResponsePtr response_;
|
|
std::unique_ptr<HttpResponseParser> response_parser_;
|
|
|
|
// Timer for the timeout control.
|
|
boost::asio::deadline_timer deadline_;
|
|
|
|
// Maximum seconds to wait before the client cancels the operation.
|
|
// Only for reading response from server.
|
|
int timeout_seconds_;
|
|
|
|
// Request stopped due to timeout or socket error.
|
|
bool stopped_;
|
|
|
|
// If the error was caused by timeout or not.
|
|
bool timed_out_;
|
|
|
|
Error error_;
|
|
};
|
|
|
|
} // namespace webcc
|
|
|
|
#endif // WEBCC_HTTP_CLIENT_H_
|