mirror of
https://github.com/nzbget/nzbget.git
synced 2026-04-18 11:56:53 -04:00
87 lines
2.6 KiB
C++
87 lines
2.6 KiB
C++
/*
|
|
* This file is part of nzbget. See <http://nzbget.net>.
|
|
*
|
|
* Copyright (C) 2004 Sven Henkel <sidddy@users.sourceforge.net>
|
|
* Copyright (C) 2007-2016 Andrey Prygunkov <hugbug@users.sourceforge.net>
|
|
*
|
|
* 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, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
|
|
#ifndef SERVERPOOL_H
|
|
#define SERVERPOOL_H
|
|
|
|
#include "Log.h"
|
|
#include "Container.h"
|
|
#include "Thread.h"
|
|
#include "NewsServer.h"
|
|
#include "NntpConnection.h"
|
|
|
|
class ServerPool : public Debuggable
|
|
{
|
|
public:
|
|
typedef std::vector<NewsServer*> RawServerList;
|
|
|
|
void SetTimeout(int timeout) { m_timeout = timeout; }
|
|
void SetRetryInterval(int retryInterval) { m_retryInterval = retryInterval; }
|
|
void AddServer(std::unique_ptr<NewsServer> newsServer);
|
|
void InitConnections();
|
|
int GetMaxNormLevel() { return m_maxNormLevel; }
|
|
Servers* GetServers() { return &m_servers; } // Only for read access (no lockings)
|
|
NntpConnection* GetConnection(int level, NewsServer* wantServer, RawServerList* ignoreServers);
|
|
void FreeConnection(NntpConnection* connection, bool used);
|
|
void CloseUnusedConnections();
|
|
void Changed();
|
|
int GetGeneration() { return m_generation; }
|
|
void BlockServer(NewsServer* newsServer);
|
|
bool IsServerBlocked(NewsServer* newsServer);
|
|
|
|
protected:
|
|
virtual void LogDebugInfo();
|
|
|
|
private:
|
|
class PooledConnection : public NntpConnection
|
|
{
|
|
public:
|
|
using NntpConnection::NntpConnection;
|
|
bool GetInUse() { return m_inUse; }
|
|
void SetInUse(bool inUse) { m_inUse = inUse; }
|
|
time_t GetFreeTime() { return m_freeTime; }
|
|
void SetFreeTimeNow();
|
|
private:
|
|
bool m_inUse = false;
|
|
time_t m_freeTime = 0;
|
|
};
|
|
|
|
typedef std::vector<int> Levels;
|
|
typedef std::vector<std::unique_ptr<PooledConnection>> Connections;
|
|
|
|
Servers m_servers;
|
|
RawServerList m_sortedServers;
|
|
Connections m_connections;
|
|
Levels m_levels;
|
|
int m_maxNormLevel = 0;
|
|
Mutex m_connectionsMutex;
|
|
int m_timeout = 60;
|
|
int m_retryInterval = 0;
|
|
int m_generation = 0;
|
|
|
|
void NormalizeLevels();
|
|
NntpConnection* LockedGetConnection(int level, NewsServer* wantServer, RawServerList* ignoreServers);
|
|
};
|
|
|
|
extern ServerPool* g_ServerPool;
|
|
|
|
#endif
|