/*
Time  : 2016/06/13
Author: Kay Yang (1025115216@qq.com)
*/

#include "PredictServerImpl.h"
#include "PredictSession.h"
#include "PredictPrint.h"
#include "FormatString.h"
struct event ev;
struct timeval tv; 
bool g_break = false;
evutil_socket_t listener;
struct event_base *base;
struct event *listener_event;

// holy shit. in multiple threads, should add a timer to break event_base in the same thread.
// otherwise in different threads, there is no effect when calling event_base_loopbreak
void time_cb(int fd, short _event, void *argc)
{
	if (g_break && base)
	{
		event_del(listener_event);
		evutil_closesocket(listener);
		event_base_loopbreak(base);
	}
	else
	{
		// it will do work that check the channel is active by write one byte to client
		// if something is wrong, maybe the connection is disconnected
		DataStructures::ThreadSafeList<PredictSession*>* sessions = PredictSessionManager::Instance()->GetSessions();
		DataStructures::List<PredictSession*>* list = sessions->LockList();
		int size = list->Size();
		for (int i = 0; i < size; ++i)
		{
			PredictSession* session = (*list)[i];
			if (!session->IsNeedRemove())
			{
				bool res = session->IsActive();
				if (!res)
				{
					ClientPrint::PrintMessage("", "write wrong 1");
				}
			}
		}
		sessions->ReleaseList();
		event_add(&ev, &tv);
	}
}

PredictServerImpl::ListenThread::ListenThread(PredictServerImpl* owner) : DataStructures::HexCommonThread(), mOwner(owner)
{

}

PredictServerImpl::ListenThread::~ListenThread()
{
	Stop();
	mOwner = 0;
}

// may call Kill() directly to Stop
void PredictServerImpl::ListenThread::Stop()
{
	if (base)
	{
		g_break = true;
		// current thread sleep 2 ms
		// the timer interval time is 1 ms
		DataStructures::HexCommonThread::ThreadSleep(10);
		if (base)
		{
			DataStructures::HexCommonThread::Kill();
			event_base_free(base);
			base = 0;
		}
	}
}

void PredictServerImpl::ListenThread::Run()
{
#ifdef WIN32
	WSADATA wsa_data;
	WSAStartup(0x0201, &wsa_data);
#endif
	struct sockaddr_in sin;
	base = event_base_new();
	if (!base)
		return;
	
	sin.sin_family = AF_INET;
	sin.sin_addr.s_addr = inet_addr(mOwner->mConfigData.mIP);
	sin.sin_port = htons(mOwner->mConfigData.mPort);
	listener = socket(AF_INET, SOCK_STREAM, 0);
	evutil_make_socket_nonblocking(listener);
	int one = 1;
	int neg = 0;
	setsockopt(listener, SOL_SOCKET, SO_REUSEADDR, (char*)&neg, sizeof(neg));
	setsockopt(listener, SOL_SOCKET, SO_KEEPALIVE, (char*)&one, sizeof(one));
	setsockopt(listener, IPPROTO_TCP, TCP_NODELAY, (char*)&one, sizeof(one));
	if (bind(listener, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ClientPrint::PrintMessage("", FormatString("端口 %d 绑定错误，检查是否已经被绑定", mOwner->mConfigData.mPort));
		return;
	}
	if (listen(listener, 16) < 0)
	{
		ClientPrint::PrintMessage("", "listen error");
		return;
	}
	listener_event = event_new(base, listener, EV_READ | EV_PERSIST | EV_WRITE, Connecting, (void*)base);
	event_add(listener_event, NULL);
	tv.tv_sec = 1;
	// micro second
	tv.tv_usec = 0;
	evtimer_assign(&ev, base, time_cb, 0);
	event_add(&ev, &tv);
	event_base_dispatch(base);
	if (base)
	{
		event_del(&ev);
		event_base_free(base);
		base = 0;
	}
#ifdef WIN32
	WSACleanup();
#endif
}

PredictServerImpl::PredictServerImpl() : mListenThread(0)
{
}

PredictServerImpl::~PredictServerImpl()
{
	Stop();
}

void PredictServerImpl::Start()
{
	// stop first
	Stop();
	if (!mListenThread)
	{
		mListenThread = new ListenThread(this);
	}
	mListenThread->Start();
}

void PredictServerImpl::Stop()
{
	if (mListenThread)
	{
		delete mListenThread;
	}
	mListenThread = 0;
}

void PredictServerImpl::SetConfigData(const PredictServerConfigData& data)
{
	mConfigData = data;
}