load_balancer.h

Go to the documentation of this file.

#include <iostream>
#include <queue>
#include <string>
#include <vector>
 
#include "request.h"
#include "web_server.h"

 
#ifndef LOAD_BALANCER_H
#define LOAD_BALANCER_H

class LoadBalancer {
 public:
  int num_servers;

  int time_to_run;

  int time_elapsed = 0;

  int min_request_time = 5;

  int max_request_time = 45;

  std::ostream& output_stream;

  std::queue<Request> request_queue;

  std::vector<WebServer*> servers;

  LoadBalancer(int num_servers, int time_to_run, std::ostream& output_stream);

  ~LoadBalancer();

  bool is_finished() { return time_elapsed >= time_to_run; }

  Request generate_random_request();

  void add_request(const Request& req) { this->request_queue.push(req); }

  void handle_requests();
 
 private:
  std::string generate_random_ip();
};
 
#endif  // LOAD_BALANCER_H