NetworkManager.h

Go to the documentation of this file.

#pragma once
 
#include <algorithm>
#include <condition_variable>
#include <functional>
#include <future>
#include <map>
#include <memory>
#include <mutex>
#include <queue>
#include <string>
#include <thread>
#include <vector>
 
#include "CncTypes.h"
#include "Inventory.h"
#include "NetconfSession.h"
#include "Topology.h"
 
namespace cnc {
 
struct NodeWorker {
    std::shared_ptr<common::NetconfSession> opsSession;
    std::shared_ptr<common::NetconfSession> notifSession;
 
    std::thread workerThread;
    std::queue<std::function<void()>> jobQueue;
    std::mutex queueMutex;
    std::condition_variable cv;
    bool stop = false;
 
    const size_t MAX_QUEUE_SIZE = 10;  // Load shredder: Max 10 pending jobs per node
};
 
class NetworkManager {
   public:
    NetworkManager(Topology& topology);
 
    ~NetworkManager();
 
    bool connectAllNodes(const InventoryMap& inventory);
 
    void fetchLldpData();
 
    void fetchOperationGcl();
 
    void fetchInterfaces();
 
    void fetchPtpData();
 
    void discoverNetwork();
 
    bool fetchLldpDataForNode(const std::string& nodeName);
 
    bool fetchOperationGclForNode(const std::string& nodeName);
 
    bool fetchPtpDataForNode(const std::string& nodeName);
 
    bool deployConfigToNode(const std::string& nodeName);
 
    bool deployInterfaceConfig(const std::string& nodeName, const std::string& ifaceName);
 
    void deployConfigToAll();
 
    std::shared_ptr<common::NetconfSession> getSession(const std::string& nodeName);
 
   private:
    Topology& topology_;  // Reference to the topology managing network nodes.
 
    // Active NodeWorkers keyed by node hostname.
    // Use shared_ptr to manage session lifetimes automatically. (no dangling pointers)
    // std::map<std::string, std::shared_ptr<common::NetconfSession>> sessions_;
    std::map<std::string, std::shared_ptr<NodeWorker>> nodeWorkers_;
 
    bool executeOnNodeWorker(const std::string& nodeName,
                             std::function<void(std::shared_ptr<common::NetconfSession>)> task);
 
    static std::string buildGclXml(const CncNode_t& node);
 
    std::string buildInterfaceGclXml(const ietfInterface_t& iface);
};
}  // namespace cnc