rcpp_framework/libs/brynet/net/wrapper/HttpConnectionBuilder.hpp

#pragma once

#include <brynet/net/http/HttpService.hpp>
#include <brynet/net/wrapper/ConnectionBuilder.hpp>
#include <utility>

class HttpConnectionBuilder {
public:
	HttpConnectionBuilder &WithService(TcpService::Ptr service) {
		mBuilder.WithService(std::move(service));
		return *this;
	}

	HttpConnectionBuilder &WithConnector(AsyncConnector::Ptr connector) {
		mBuilder.WithConnector(std::move(connector));
		return *this;
	}

	HttpConnectionBuilder &WithAddr(std::string ip, size_t port) {
		mBuilder.WithAddr(std::move(ip), port);
		return *this;
	}

	HttpConnectionBuilder &WithTimeout(std::chrono::nanoseconds timeout) {
		mBuilder.WithTimeout(timeout);
		return *this;
	}

	HttpConnectionBuilder &AddSocketProcessCallback(const ProcessTcpSocketCallback &callback) {
		mBuilder.AddSocketProcessCallback(callback);
		return *this;
	}

	HttpConnectionBuilder &WithEnterCallback(HttpSession::EnterCallback &&callback) {
		mHttpEnterCallback = std::move(callback);
		return *this;
	}

	HttpConnectionBuilder &WithFailedCallback(FailedCallback callback) {
		mBuilder.WithFailedCallback(std::move(callback));
		return *this;
	}

	HttpConnectionBuilder &WithMaxRecvBufferSize(size_t size) {
		mBuilder.WithMaxRecvBufferSize(size);
		return *this;
	}

#ifdef BRYNET_USE_OPENSSL
	HttpConnectionBuilder &WithSSL() {
		mBuilder.WithSSL();
		return *this;
	}
#endif
	HttpConnectionBuilder &WithForceSameThreadLoop() {
		mBuilder.WithForceSameThreadLoop();
		return *this;
	}

	void asyncConnect() {
		if (mHttpEnterCallback == nullptr) {
			throw BrynetCommonException("not setting http enter callback");
		}

		auto callback = mHttpEnterCallback;
		auto builder = mBuilder;
		builder.AddEnterCallback([callback](const TcpConnection::Ptr &session) {
			HttpService::setup(session, callback);
		});
		builder.asyncConnect();
	}

private:
	HttpSession::EnterCallback mHttpEnterCallback;
	ConnectionBuilder mBuilder;
};