mirror of
https://github.com/VCMP-SqMod/SqMod.git
synced 2025-02-22 20:57:12 +01:00
156 lines
3.7 KiB
C++
156 lines
3.7 KiB
C++
//
|
|
// TCPServerDispatcher.h
|
|
//
|
|
// Library: Net
|
|
// Package: TCPServer
|
|
// Module: TCPServerDispatcher
|
|
//
|
|
// Definition of the TCPServerDispatcher class.
|
|
//
|
|
// Copyright (c) 2005-2006, Applied Informatics Software Engineering GmbH.
|
|
// and Contributors.
|
|
//
|
|
// SPDX-License-Identifier: BSL-1.0
|
|
//
|
|
|
|
|
|
#ifndef Net_TCPServerDispatcher_INCLUDED
|
|
#define Net_TCPServerDispatcher_INCLUDED
|
|
|
|
|
|
#include "Poco/Net/Net.h"
|
|
#include "Poco/Net/StreamSocket.h"
|
|
#include "Poco/Net/TCPServerConnectionFactory.h"
|
|
#include "Poco/Net/TCPServerParams.h"
|
|
#include "Poco/Runnable.h"
|
|
#include "Poco/NotificationQueue.h"
|
|
#include "Poco/ThreadPool.h"
|
|
#include "Poco/Mutex.h"
|
|
#include <atomic>
|
|
|
|
|
|
namespace Poco {
|
|
namespace Net {
|
|
|
|
|
|
class Net_API TCPServerDispatcher: public Poco::Runnable
|
|
/// A helper class for TCPServer that dispatches
|
|
/// connections to server connection threads.
|
|
{
|
|
public:
|
|
TCPServerDispatcher(TCPServerConnectionFactory::Ptr pFactory, Poco::ThreadPool& threadPool, TCPServerParams::Ptr pParams);
|
|
/// Creates the TCPServerDispatcher.
|
|
///
|
|
/// The dispatcher takes ownership of the TCPServerParams object.
|
|
/// If no TCPServerParams object is supplied, the TCPServerDispatcher
|
|
/// creates one.
|
|
|
|
void duplicate();
|
|
/// Increments the object's reference count.
|
|
|
|
void release();
|
|
/// Decrements the object's reference count
|
|
/// and deletes the object if the count
|
|
/// reaches zero.
|
|
|
|
void run();
|
|
/// Runs the dispatcher.
|
|
|
|
void enqueue(const StreamSocket& socket);
|
|
/// Queues the given socket connection.
|
|
|
|
void stop();
|
|
/// Stops the dispatcher.
|
|
|
|
int currentThreads() const;
|
|
/// Returns the number of currently used threads.
|
|
|
|
int maxThreads() const;
|
|
/// Returns the maximum number of threads available.
|
|
|
|
int totalConnections() const;
|
|
/// Returns the total number of handled connections.
|
|
|
|
int currentConnections() const;
|
|
/// Returns the number of currently handled connections.
|
|
|
|
int maxConcurrentConnections() const;
|
|
/// Returns the maximum number of concurrently handled connections.
|
|
|
|
int queuedConnections() const;
|
|
/// Returns the number of queued connections.
|
|
|
|
int refusedConnections() const;
|
|
/// Returns the number of refused connections.
|
|
|
|
const TCPServerParams& params() const;
|
|
/// Returns a const reference to the TCPServerParam object.
|
|
|
|
protected:
|
|
~TCPServerDispatcher();
|
|
/// Destroys the TCPServerDispatcher.
|
|
|
|
void beginConnection();
|
|
/// Updates the performance counters.
|
|
|
|
void endConnection();
|
|
/// Updates the performance counters.
|
|
|
|
private:
|
|
TCPServerDispatcher();
|
|
TCPServerDispatcher(const TCPServerDispatcher&);
|
|
TCPServerDispatcher& operator = (const TCPServerDispatcher&);
|
|
|
|
class ThreadCountWatcher
|
|
{
|
|
public:
|
|
ThreadCountWatcher(TCPServerDispatcher* pDisp) : _pDisp(pDisp)
|
|
{
|
|
}
|
|
|
|
~ThreadCountWatcher()
|
|
{
|
|
FastMutex::ScopedLock lock(_pDisp->_mutex);
|
|
if (_pDisp->_currentThreads > 1 && _pDisp->_queue.empty())
|
|
{
|
|
--_pDisp->_currentThreads;
|
|
}
|
|
}
|
|
|
|
private:
|
|
ThreadCountWatcher();
|
|
ThreadCountWatcher(const ThreadCountWatcher&);
|
|
ThreadCountWatcher& operator=(const ThreadCountWatcher&);
|
|
|
|
TCPServerDispatcher* _pDisp;
|
|
};
|
|
|
|
std::atomic<int> _rc;
|
|
TCPServerParams::Ptr _pParams;
|
|
std::atomic<int> _currentThreads;
|
|
std::atomic<int> _totalConnections;
|
|
std::atomic<int> _currentConnections;
|
|
std::atomic<int> _maxConcurrentConnections;
|
|
std::atomic<int> _refusedConnections;
|
|
std::atomic<bool> _stopped;
|
|
Poco::NotificationQueue _queue;
|
|
TCPServerConnectionFactory::Ptr _pConnectionFactory;
|
|
Poco::ThreadPool& _threadPool;
|
|
mutable Poco::FastMutex _mutex;
|
|
};
|
|
|
|
|
|
//
|
|
// inlines
|
|
//
|
|
inline const TCPServerParams& TCPServerDispatcher::params() const
|
|
{
|
|
return *_pParams;
|
|
}
|
|
|
|
|
|
} } // namespace Poco::Net
|
|
|
|
|
|
#endif // Net_TCPServerDispatcher_INCLUDED
|