31 lines
869 B
C++
31 lines
869 B
C++
#pragma once
|
|
#include "Globals.h"
|
|
#include "NodeContext.h"
|
|
#include "NodeInfo.h"
|
|
#include "TaskManager.h"
|
|
#include <Arduino.h>
|
|
#include <ESP8266WiFi.h>
|
|
#include <ArduinoJson.h>
|
|
#include <ESP8266HTTPClient.h>
|
|
#include <map>
|
|
|
|
class ClusterManager {
|
|
public:
|
|
ClusterManager(NodeContext& ctx, TaskManager& taskMgr);
|
|
void registerTasks();
|
|
void sendDiscovery();
|
|
void listenForDiscovery();
|
|
void addOrUpdateNode(const String& nodeHost, IPAddress nodeIP);
|
|
void updateAllNodeStatuses();
|
|
void removeDeadNodes();
|
|
void printMemberList();
|
|
const std::map<String, NodeInfo>& getMemberList() const { return *ctx.memberList; }
|
|
void fetchNodeInfo(const IPAddress& ip);
|
|
void updateLocalNodeResources();
|
|
void heartbeatTaskCallback();
|
|
void updateAllMembersInfoTaskCallback();
|
|
private:
|
|
NodeContext& ctx;
|
|
TaskManager& taskManager;
|
|
};
|