pokerth/src/net/senderthread.h

/***************************************************************************
 *   Copyright (C) 2007 by Lothar May                                      *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/
/* Network sender thread. */

#ifndef _SENDERTHREAD_H_
#define _SENDERTHREAD_H_

#include <core/thread.h>
#include <net/socket_helper.h>
#include <net/netpacket.h>
#include <net/sendercallback.h>

#include <deque>
#include <boost/shared_ptr.hpp>

#define SENDER_THREAD_TERMINATE_TIMEOUT		200
#define SEND_TIMEOUT_MSEC					10
#define SEND_QUEUE_SIZE						1000
#define SEND_LOW_PRIO_QUEUE_SIZE			50000

class SenderThread : public Thread
{
public:
	SenderThread(SenderCallback &cb);
	virtual ~SenderThread();

	void Send(SOCKET sock, boost::shared_ptr<NetPacket> packet);
	void Send(SOCKET sock, const NetPacketList &packetList);

	void SendLowPrio(SOCKET sock, boost::shared_ptr<NetPacket> packet);
	void SendLowPrio(SOCKET sock, const NetPacketList &packetList);

protected:
	typedef std::pair<boost::shared_ptr<NetPacket>, SOCKET> SendData;
	typedef std::deque<SendData> SendDataDeque;

	// Main function of the thread.
	virtual void Main();

	void InternalStore(SendDataDeque &sendQueue, unsigned maxQueueSize, SOCKET sock, boost::shared_ptr<NetPacket> packet);
	void InternalStore(SendDataDeque &sendQueue, unsigned maxQueueSize, SOCKET sock, const NetPacketList &packetList);

private:

	SOCKET m_curSocket;

	std::deque<SendData> m_outBuf;
	mutable boost::mutex m_outBufMutex;

	std::deque<SendData> m_lowPrioOutBuf;
	mutable boost::mutex m_lowPrioOutBufMutex;

	char m_tmpOutBuf[MAX_PACKET_SIZE];
	unsigned m_tmpOutBufSize;

	SenderCallback &m_callback;
};

#endif