GlobalControlManager.cpp

//
// Created by greg on 9/12/19.
//

#include "GlobalControlManager.h"

ec::GlobalControlManager::GlobalControlManager(std::string ip_addr, uint16_t port, agents_ip_list &_agent_ips, std::vector<uint16_t> &_server_ports)
    : gcm_ip(ec::ip4_addr::from_string(std::move(ip_addr))), mngr(nullptr), gcm_port(port), server_ports(_server_ports), manager_counts(1) {

    for(const auto &i : _agent_ips) {
        agents.emplace_back(new Agent(i));
    }

    if(agents.size() != _agent_ips.size()) {
        SPDLOG_ERROR("ERROR: alloc agents failed!");
        exit(EXIT_FAILURE);
    }
}

ec::GlobalControlManager::GlobalControlManager(std::string ip_addr, uint16_t port, agents_ip_list &_agent_ips, std::vector<ports_t> &_server_ports)
    : gcm_ip(ec::ip4_addr::from_string(std::move(ip_addr))), mngr(nullptr), gcm_port(port), controller_ports(_server_ports), manager_counts(1) {

    for(const auto &i : _agent_ips) {
        agents.emplace_back(new Agent(i));
    }

    if(agents.size() != _agent_ips.size()) {
        SPDLOG_ERROR("ERROR: alloc agents failed!");
        exit(EXIT_FAILURE);
    }
}


uint32_t ec::GlobalControlManager::create_manager() {
    if(managers.find(manager_counts) != managers.end()) {
        SPDLOG_ERROR("Error allocating new Server. Server IDs not correct");
        return 0;
    }

    auto ports = controller_ports[manager_counts - 1];
    mngr = new Manager(manager_counts, gcm_ip, ports, agents);
    managers.insert({manager_counts, mngr});

    manager_counts++;
    return mngr->get_server_id();
}

const ec::Manager &ec::GlobalControlManager::get_manager(const int manager_id) const {
    auto itr = managers.find(manager_id);
    if(itr == managers.end()) {
        SPDLOG_CRITICAL(" No Server with manager_id: {}. Exiting....", manager_id);
        std::exit(EXIT_FAILURE);
    }
    return *itr->second;
}


ec::GlobalControlManager::~GlobalControlManager() {
    for(auto a : agents) {
        delete a;
    }
    agents.clear();
    for(auto &s : managers) {
        delete s.second;
    }
    managers.clear();
}

void ec::GlobalControlManager::run(const std::string &app_name, const std::string &_gcm_ip, const int num_containers) {
    auto m = managers.find(1)->second;
    m->start(app_name, _gcm_ip, num_containers);
}

bool ec::GlobalControlManager::init_agent_connections() {
    int sockfd, i;
    struct sockaddr_in servaddr;
    uint32_t num_connections = 0;

    AgentClientDB* agent_clients_db = AgentClientDB::get_agent_client_db_instance();
    for(const auto &ag : agents) {
        auto grpc_addr = ag->get_ip().to_string() + ":" + std::to_string(ag->get_port());
        auto* ac = new rpc::AgentClient(ag, grpc::CreateChannel(
                grpc_addr, grpc::InsecureChannelCredentials()));

        num_connections++;
        agent_clients_db->add_agent_client(ac);
    }
    return num_connections == agent_clients_db->get_agent_clients_db_size();

}

void ec::GlobalControlManager::join_grpc_threads() {
    AgentClientDB* agent_clients_db = AgentClientDB::get_agent_client_db_instance();

    auto db_map = agent_clients_db->get_db_map();

    for(auto const& [ip, client] : *db_map) {
        client->get_thread()->join();
    }


}