LoadBalancer.h

Go to the documentation of this file.

 
#ifndef LOAD_BALANCER_H
#define LOAD_BALANCER_H
 
#include "Request.h"
#include "WebServer.h"
#include "Config.h"
#include <queue>
#include <vector>
#include <memory>
#include <random>

class LoadBalancer {
public:
    LoadBalancer(int initialServers, const LoadBalancerConfig& config);

    void run(int totalClockCycles);

    void prefillQueue(int count);

    void generateRequest();

    void scaleServers();

    void assignRequests();

    void printSummary(int initialServerCount) const;
 
private:
    Request createRandomRequest();

    std::string generateRandomIP();

    void log(const char* colorCode, const char* msg) const;
 
    LoadBalancerConfig config_;                    
    std::queue<Request> requestQueue_;             
    std::vector<std::unique_ptr<WebServer>> servers_;  
    int currentTime_;                              
    int minQueueThreshold_;                        
    int maxQueueThreshold_;                        
    int lastScaleTime_;                            
    int nextServerId_;                             
 
    std::mt19937 rng_;                             
    std::bernoulli_distribution requestDist_;      
    std::uniform_int_distribution<int> timeDist_;  
    std::uniform_int_distribution<int> octetDist_; 
    std::uniform_int_distribution<int> jobTypeDist_;  
 
    int totalProcessed_;       
    int totalBlocked_;         
    int requestsAddedToQueue_; 
    int serversAdded_;         
    int serversRemoved_;       
};
 
#endif