#ifndef SOCKET_H

#define SOCKET_H

#include <thread></thread>

#include <atomic></atomic>

#include "protocol.h"

enum : ErrorCode  {

	ERR_TCP_COMMON = ERR_TCP,

	ERR_TCP_WORKER_FAILED,

};

class TcpProtocol;

class TcpSocket final: public Socket {

public:

	const int sockId;

protected:

	friend class TcpProtocol;

	TcpSocket(

		Protocol &protocol,

		const Connection::Handle &connection,

		const Server::Handle &server,

		const Address &address,

		int sockId );

	TcpProtocol& getProtocol()

		{ return *(TcpProtocol*)&protocol; }

};

class TcpProtocol: public Protocol {

private:

	friend class TcpSocket;

	std::mutex mutexIdle;

	std::mutex mutexRun;

	std::atomic<bool> wantRun;</bool>

	std::thread *thread;

	int epollId;

	int eventsId;

	void controlCommand(unsigned long long cmd);

	void threadConnectionWrite(TcpSocket *socket, const Connection::Handle &connection);

	void threadConnectionRead(TcpSocket *socket, const Connection::Handle &connection);

	void threadProcessEvents(void *eventsPtr, int count);

	void threadRun();

public:

	TcpProtocol();

	~TcpProtocol();

	ErrorCode resolve(Address &address) override;

protected:

	ErrorCode onStart() override;

	void onStop(ErrorCode) override;

	void onSocketChanged(Socket *socket) override;

	ErrorCode onConnect(const Connection::Handle &connection, const Address &address) override;

	ErrorCode onListen(const Server::Handle &server, const Address &address) override;

};

#endif