urbackup_backend/ServiceWorker.h

65 lines
1.2 KiB
C++

#pragma once
#include <vector>
#include <utility>
#include <stack>
#include "Interface/Thread.h"
#include "Interface/Mutex.h"
#include "Interface/Condition.h"
#include "Interface/Pipe.h"
#include "socket_header.h"
#include "Interface/CustomClient.h"
const int MAX_CLIENTS=20;
class IService;
class CStreamPipe;
class CServiceWorker : public IThread, IRunOtherCallback
{
public:
CServiceWorker(IService *pService, std::string pName, IPipe * pExit, int pMaxClientsPerThread);
~CServiceWorker();
void operator ()(void);
int getAvailableSlots(void);
void AddClient(SOCKET pSocket, const std::string& endpoint);
void stop(void);
virtual void runOther();
struct SCurrWork
{
ICustomClient* client;
bool did_other_work;
};
private:
void work(ICustomClient* skip_client);
void addNewClients(void);
std::vector<std::pair<ICustomClient*, CStreamPipe*> > clients;
std::vector<std::pair<SOCKET, std::string> > new_clients;
IMutex* mutex;
IMutex* nc_mutex;
ICondition* cond;
IPipe *exit;
int nClients;
int max_clients;
THREAD_ID tid;
std::string name;
IService *service;
volatile bool do_stop;
std::stack<SCurrWork> curr_work;
};