From b6eec490f497a980885b3860d588d23973ee4e46 Mon Sep 17 00:00:00 2001
From: gvolfing <gabor.volfinger@memgraph.io>
Date: Fri, 5 Aug 2022 16:45:44 +0200
Subject: [PATCH] Update Coordinator

---
 src/coordinator/coordinator.hpp | 19 +++++++++++++++----
 1 file changed, 15 insertions(+), 4 deletions(-)

diff --git a/src/coordinator/coordinator.hpp b/src/coordinator/coordinator.hpp
index 1fcb23fa8..0e3feef1e 100644
--- a/src/coordinator/coordinator.hpp
+++ b/src/coordinator/coordinator.hpp
@@ -11,6 +11,8 @@
 
 #pragma once
 
+#include <unordered_set>
+
 #include "coordinator/hybrid_logical_clock.hpp"
 #include "coordinator/shard_map.hpp"
 #include "io/simulator/simulator.hpp"
@@ -85,7 +87,11 @@ using ReadRequests = std::variant<HlcRequest, GetShardMapRequest>;
 using ReadResponses = std::variant<HlcResponse, GetShardMapResponse>;
 
 class Coordinator {
+  using StandbySotrageEnginePool = std::unordered_set<Address>;
+
   ShardMap shard_map_;
+  StandbySotrageEnginePool storage_engine_pool_;
+
   /// The highest reserved timestamp / highest allocated timestamp
   /// is a way for minimizing communication involved in query engines
   /// reserving Hlc's for their transaction processing.
@@ -129,10 +135,11 @@ class Coordinator {
 
     if (split_shard_request.previous_shard_map_version != shard_map_.shard_map_version) {
       // TODO reply with failure
+      res.success = false;
+    } else {
+      // TODO apply split
     }
 
-    // TODO apply split
-
     return res;
   }
 
@@ -140,8 +147,10 @@ class Coordinator {
   /// which can be used to rebalance storage over time.
   WriteResponses Apply(RegisterStorageEngineRequest &&register_storage_engine_request) {
     RegisterStorageEngineResponse res{};
-
     // TODO
+    const Address &address = register_storage_engine_request.address;
+    storage_engine_pool_.insert(address);
+    res.success = true;
 
     return res;
   }
@@ -150,8 +159,10 @@ class Coordinator {
   /// clusters that it might be participating in.
   WriteResponses Apply(DeregisterStorageEngineRequest &&register_storage_engine_request) {
     DeregisterStorageEngineResponse res{};
-
     // TODO
+    const Address &address = register_storage_engine_request.address;
+    storage_engine_pool_.erase(address);
+    res.success = true;
 
     return res;
   }