78 lines
2.9 KiB
C++
78 lines
2.9 KiB
C++
/***************************************************************************
|
|
* 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/sessiondata.h>
|
|
#include <net/netpacket.h>
|
|
#include <net/sendercallback.h>
|
|
|
|
#include <list>
|
|
#include <boost/shared_ptr.hpp>
|
|
|
|
#define SENDER_THREAD_TERMINATE_TIMEOUT THREAD_WAIT_INFINITE
|
|
|
|
class SenderThread : public Thread
|
|
{
|
|
public:
|
|
SenderThread(SenderCallback &cb);
|
|
virtual ~SenderThread();
|
|
|
|
void Send(boost::shared_ptr<SessionData> session, boost::shared_ptr<NetPacket> packet);
|
|
void Send(boost::shared_ptr<SessionData> session, const NetPacketList &packetList);
|
|
|
|
unsigned GetNumPacketsInQueue() const;
|
|
bool operator<(const SenderThread &other) const;
|
|
|
|
protected:
|
|
struct SendData
|
|
{
|
|
SendData()
|
|
: bytesSent(0) {}
|
|
SendData(boost::shared_ptr<NetPacket> p, boost::shared_ptr<SessionData> s)
|
|
: packet(p), session(s), bytesSent(0) {}
|
|
boost::shared_ptr<NetPacket> packet;
|
|
boost::shared_ptr<SessionData> session;
|
|
unsigned bytesSent;
|
|
};
|
|
typedef std::list<SendData> SendDataList;
|
|
|
|
// Main function of the thread.
|
|
virtual void Main();
|
|
|
|
void InternalStore(SendDataList &sendQueue, unsigned maxQueueSize, boost::shared_ptr<SessionData> session, boost::shared_ptr<NetPacket> packet);
|
|
void InternalStore(SendDataList &sendQueue, unsigned maxQueueSize, boost::shared_ptr<SessionData> session, const NetPacketList &packetList);
|
|
|
|
private:
|
|
|
|
SendDataList m_sendQueue;
|
|
mutable boost::mutex m_sendQueueMutex;
|
|
|
|
SendDataList m_stalledQueue;
|
|
mutable boost::mutex m_stalledQueueMutex;
|
|
|
|
SenderCallback &m_callback;
|
|
};
|
|
|
|
#endif
|
|
|