socketengine.h

WOW 服务模拟端 支持2.4.3版本 来自开源的ASCENT 自己REPACK

/****************************************************************************
 *
 * Multiplatform High-Performance Async Network Library
 * Base Socket Engine Interface Class
 * Copyright (c) 2007 Burlex
 *
 * This file may be distributed under the terms of the Q Public License
 * as defined by Trolltech ASA of Norway and appearing in the file
 * COPYING included in the packaging of this file.
 *
 * This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
 * WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
 *
 */

#ifndef _NETLIB_SOCKETENGINE_H
#define _NETLIB_SOCKETENGINE_H

class SocketEngine : public Singleton<SocketEngine>
{
public:
	virtual ~SocketEngine() {}

	/** Adds a socket to the engine.
	 */
	virtual void AddSocket(BaseSocket * s) = 0;

	/** Removes a socket from the engine. It should not receive any more events.
	 */
	virtual void RemoveSocket(BaseSocket * s) = 0;

	/** This is called when a socket has data to write for the first time.
	 */
	virtual void WantWrite(BaseSocket * s) = 0;

	/** Spawn however many worker threads this engine requires
	 */
	virtual void SpawnThreads() = 0;

	/** Shutdown the socket engine, disconnect any associated sockets and 
	 * deletes itself and the socket deleter.
	 */
	virtual void Shutdown() = 0;

	/** Called by SocketWorkerThread, this is the network loop.
	 */
	virtual void MessageLoop() = 0;
};

class SocketEngineThread : public ThreadBase
{
	SocketEngine * se;
public:
	SocketEngineThread(SocketEngine * s) : se(s) {}
	void run()
	{
		se->MessageLoop();
	}
};

class SocketDeleter : public Singleton<SocketDeleter>
{
	typedef map<BaseSocket*, time_t> SocketDeleteMap;
	SocketDeleteMap _map;
	Mutex _lock;
public:
	/** Call this every loop of your program to delete old sockets
	 */
	void Update()
	{
		time_t ct = time(NULL);
        _lock.Acquire();
		SocketDeleteMap::iterator it, it2;
		for(it = _map.begin(); it != _map.end();)
		{
			it2 = it++;
            if(it2->second <= ct)
			{
				delete it2->first;
				_map.erase(it2);
			}
		}

		_lock.Release();
	}

	/** Add a socket for deletion in time.
	 */
	void Add(BaseSocket * s)
	{
		_lock.Acquire();
		_map[s] = time(NULL) + 15;
		_lock.Release();
	}

	/** Delete all sockets in this socket deleter, regardless of time
	 */
	void Kill()
	{
		SocketDeleteMap::iterator itr = _map.begin();
		for(; itr != _map.end(); ++itr)
			delete itr->first;

		_map.clear();
	}
};

#define sSocketEngine SocketEngine::getSingleton()
#define sSocketDeleter SocketDeleter::getSingleton()

#endif		// _NETLIB_SOCKETENGINE_H