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

#pragma once

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

namespace brynet {
namespace net {
namespace wrapper {

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

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

	HttpListenerBuilder &AddSocketProcess(const ListenThread::TcpSocketProcessCallback &callback) {
		mBuilder.AddSocketProcess(callback);
		return *this;
	}

	HttpListenerBuilder &WithMaxRecvBufferSize(size_t size) {
		mBuilder.WithMaxRecvBufferSize(size);
		return *this;
	}
#ifdef BRYNET_USE_OPENSSL
	HttpListenerBuilder &WithSSL(SSLHelper::Ptr sslHelper) {
		mBuilder.WithSSL(std::move(sslHelper));
		return *this;
	}
#endif
	HttpListenerBuilder &WithForceSameThreadLoop() {
		mBuilder.WithForceSameThreadLoop();
		return *this;
	}

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

	HttpListenerBuilder &WithReusePort() {
		mBuilder.WithReusePort();
		return *this;
	}

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

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

private:
	http::HttpSession::EnterCallback mHttpEnterCallback;
	ListenerBuilder mBuilder;
};

} // namespace wrapper
} // namespace net
} // namespace brynet