TCPServer.h

#pragma once

#include "mg/aio/IOTask.h"
#include "mg/box/Mutex.h"

namespace mg {
namespace aio {

	class IOCore;

	enum TCPServerState
	{
		TCP_SERVER_STATE_NEW,
		TCP_SERVER_STATE_BOUND,
		TCP_SERVER_STATE_LISTENING,
		TCP_SERVER_STATE_CLOSING,
		TCP_SERVER_STATE_CLOSED,
	};

	class TCPServerSubscription
	{
	protected:
		~TCPServerSubscription() = default;
	public:
		virtual void OnAccept(
			mg::net::Socket aSock,
			const mg::net::Host& aHost) = 0;
		// Close is signaled only if Listen() was successful beforehand.
		virtual void OnClose() = 0;
	};

	class TCPServer
		: private IOSubscription
	{
	public:
		SHARED_PTR_RENEW_API(TCPServer)

		bool Bind(
			const mg::net::Host& aHost,
			mg::box::Error::Ptr& aOutErr);

		uint16_t GetPort() const;

		bool Listen(
			uint32_t aBacklog,
			TCPServerSubscription* aSub,
			mg::box::Error::Ptr& aOutErr);

		void PostClose();
		bool IsClosed() const;
		IOCore& GetCore();

	private:
		TCPServer(IOCore& aCore);
		~TCPServer() override;

		void OnEvent(
			const IOArgs& aArgs) override;

		mutable mg::box::Mutex myMutex;
		TCPServerState myState;
		IOTask myTask;
		IOEvent myAcceptEvent;
		TCPServerSubscription* mySub;
		IOServerSocket* myBoundSocket;
	};

	inline IOCore&
	TCPServer::GetCore()
	{
		return myTask.GetCore();
	}

}
}