rework client classes
Chunting Gu
4 years ago
parent
d294cda3c1
commit
28ae628d56
@ -0,0 +1,164 @@
|
||||
#ifndef WEBCC_CLIENT_BASE_H_
|
||||
#define WEBCC_CLIENT_BASE_H_
|
||||
|
||||
#include <condition_variable>
|
||||
#include <memory>
|
||||
#include <mutex>
|
||||
#include <string>
|
||||
#include <vector>
|
||||
|
||||
#include "boost/asio/io_context.hpp"
|
||||
#include "boost/asio/ip/tcp.hpp"
|
||||
#include "boost/asio/steady_timer.hpp"
|
||||
|
||||
#include "webcc/globals.h"
|
||||
#include "webcc/request.h"
|
||||
#include "webcc/response.h"
|
||||
#include "webcc/response_parser.h"
|
||||
#include "webcc/socket_base.h"
|
||||
|
||||
namespace webcc {
|
||||
|
||||
class ClientBase {
|
||||
public:
|
||||
explicit ClientBase(boost::asio::io_context& io_context);
|
||||
|
||||
ClientBase(const ClientBase&) = delete;
|
||||
ClientBase& operator=(const ClientBase&) = delete;
|
||||
|
||||
~ClientBase() = default;
|
||||
|
||||
void set_buffer_size(std::size_t buffer_size) {
|
||||
if (buffer_size > 0) {
|
||||
buffer_size_ = buffer_size;
|
||||
}
|
||||
}
|
||||
|
||||
void set_connect_timeout(int timeout) {
|
||||
if (timeout > 0) {
|
||||
connect_timeout_ = timeout;
|
||||
}
|
||||
}
|
||||
|
||||
void set_read_timeout(int timeout) {
|
||||
if (timeout > 0) {
|
||||
read_timeout_ = timeout;
|
||||
}
|
||||
}
|
||||
|
||||
// Set progress callback to be informed about the read progress.
|
||||
// NOTE: Don't use move semantics because in practice, there is no difference
|
||||
// between copying and moving an object of a closure type.
|
||||
// TODO: Support write progress
|
||||
void set_progress_callback(ProgressCallback callback) {
|
||||
progress_callback_ = callback;
|
||||
}
|
||||
|
||||
// Connect, send request, wait until response is received.
|
||||
Error Request(RequestPtr request, bool stream = false);
|
||||
|
||||
// Close the connection.
|
||||
// The async operation on the socket will be canceled.
|
||||
void Close();
|
||||
|
||||
bool connected() const {
|
||||
return connected_;
|
||||
}
|
||||
|
||||
ResponsePtr response() const {
|
||||
return response_;
|
||||
}
|
||||
|
||||
// Reset response object.
|
||||
// Used to make sure the response object will released even the client object
|
||||
// itself will be cached for keep-alive purpose.
|
||||
void Reset() {
|
||||
response_.reset();
|
||||
response_parser_.Init(nullptr, false);
|
||||
}
|
||||
|
||||
protected:
|
||||
// Create Socket or SslSocket.
|
||||
virtual void CreateSocket() = 0;
|
||||
|
||||
// Resolve host.
|
||||
virtual void Resolve() = 0;
|
||||
|
||||
void CloseSocket();
|
||||
|
||||
void AsyncResolve(string_view default_port);
|
||||
|
||||
void OnResolve(boost::system::error_code ec,
|
||||
boost::asio::ip::tcp::resolver::results_type endpoints);
|
||||
|
||||
void OnConnect(boost::system::error_code ec, boost::asio::ip::tcp::endpoint);
|
||||
|
||||
void AsyncWrite();
|
||||
void OnWrite(boost::system::error_code ec, std::size_t length);
|
||||
|
||||
void AsyncWriteBody();
|
||||
void OnWriteBody(boost::system::error_code ec, std::size_t length);
|
||||
|
||||
void HandleWriteError(boost::system::error_code ec);
|
||||
|
||||
void AsyncRead();
|
||||
void OnRead(boost::system::error_code ec, std::size_t length);
|
||||
|
||||
void AsyncWaitDeadlineTimer(int seconds);
|
||||
void OnDeadlineTimer(boost::system::error_code ec);
|
||||
void StopDeadlineTimer();
|
||||
|
||||
void FinishRequest();
|
||||
|
||||
protected:
|
||||
boost::asio::io_context& io_context_;
|
||||
|
||||
std::unique_ptr<SocketBase> socket_;
|
||||
|
||||
boost::asio::ip::tcp::resolver resolver_;
|
||||
|
||||
bool request_finished_ = true;
|
||||
std::condition_variable request_cv_;
|
||||
std::mutex request_mutex_;
|
||||
|
||||
RequestPtr request_;
|
||||
|
||||
ResponsePtr response_;
|
||||
ResponseParser response_parser_;
|
||||
|
||||
// The length already read.
|
||||
std::size_t length_read_ = 0;
|
||||
|
||||
// The buffer for reading response.
|
||||
std::vector<char> buffer_;
|
||||
|
||||
// The size of the buffer for reading response.
|
||||
// 0 means default value will be used.
|
||||
std::size_t buffer_size_ = kBufferSize;
|
||||
|
||||
// Timeout (seconds) for connecting to server.
|
||||
// Default as 0 to disable our own control (i.e., deadline_timer_).
|
||||
int connect_timeout_ = 0;
|
||||
|
||||
// Timeout (seconds) for reading response.
|
||||
int read_timeout_ = kMaxReadSeconds;
|
||||
|
||||
// Deadline timer for connecting to server.
|
||||
boost::asio::steady_timer deadline_timer_;
|
||||
bool deadline_timer_stopped_ = true;
|
||||
|
||||
// Socket connected or not.
|
||||
bool connected_ = false;
|
||||
|
||||
// Progress callback (optional).
|
||||
ProgressCallback progress_callback_;
|
||||
|
||||
// Current error.
|
||||
Error error_;
|
||||
};
|
||||
|
||||
using ClientPtr = std::shared_ptr<ClientBase>;
|
||||
|
||||
} // namespace webcc
|
||||
|
||||
#endif // WEBCC_CLIENT_BASE_H_
|
||||
@ -1,23 +0,0 @@
|
||||
#include "webcc/ssl_client.h"
|
||||
|
||||
#include "boost/algorithm/string.hpp"
|
||||
|
||||
#include "webcc/ssl_socket.h"
|
||||
|
||||
namespace webcc {
|
||||
|
||||
SslClient::SslClient(boost::asio::io_context& io_context,
|
||||
boost::asio::ssl::context& ssl_context)
|
||||
: Client(io_context), ssl_context_(ssl_context) {
|
||||
}
|
||||
|
||||
void SslClient::AsyncConnect() {
|
||||
if (boost::iequals(request_->url().scheme(), "https")) {
|
||||
socket_.reset(new SslSocket{ io_context_, ssl_context_ });
|
||||
AsyncResolve("443");
|
||||
} else {
|
||||
Client::AsyncConnect();
|
||||
}
|
||||
}
|
||||
|
||||
} // namespace webcc
|
||||
Loading…
Reference in New Issue