loadbalancer.h

Go to the documentation of this file.

#ifndef LOADBALANCER_H
#define LOADBALANCER_H
 
#include "request.h"
#include "requestqueue.h"
#include "webserver.h"
#include "logmanager.h"
#include <vector>
#include <string>
 
 //all doxygen comments are generated with AI assistance
class LoadBalancer {
public:
    LoadBalancer(LogManager& logger);
 
    int getTime() const;
 
    void incTime();
 
 
    void addRequest(const Request& r);
 
    Request getRequest();
 
    bool isRequestQueueEmpty() const;
 
    size_t getRequestQueueSize() const;
 
    int getProcessedRequests() const;
 
 
    void incrementProcessedRequests();
 
 
    WebServer& getNextServer();
 
    void allocateServer();
 
    void deallocateServer();
 
    // IP blocking
 
    bool isIpBlocked(const std::string& ip) const;   //implemented with the assistance of AI
 
 
    int getRejectedRequests() const;
 
    int getActiveServers() const;
 
    int getInactiveServers() const;
 
    void setTotalServers(int total);
 
private:
    LogManager& logger;  
    int currentTime = 0; 
    RequestQueue requestQueue; 
    int currentServerIndex = 0; 
    int processedRequests = 0; 
    int rejectedRequests = 0; 
    int activeServers = 0; 
    int totalServers = 0; 
    const int maxServers = 50; 
    const int minServers = 1; 
    std::vector<std::string> blockedIpRanges; 
    std::vector<WebServer> servers; 
    void initializeBlockedIpRanges();        //implemented with the assistance of AI
 
    void logRejectedRequest(const Request& r);         //implemented with the assistance of AI
};
 
#endif // LOADBALANCER_H