Close connections when shutting down the the cluster (which will trig…

…ger sending a close announcement) Signed-off-by: Florian Fendt <Florian.Fendt@bosch.io>
eclipse-ditto · May 7, 2021 · d80c088 · d80c088
1 parent f3cbc12
commit d80c088
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/...rvice/src/main/java/org/eclipse/ditto/connectivity/service/messaging/BaseClientActor.java b/...rvice/src/main/java/org/eclipse/ditto/connectivity/service/messaging/BaseClientActor.java
@@ -131,6 +131,7 @@
 import akka.actor.ActorRef;
 import akka.actor.ActorSelection;
 import akka.actor.ActorSystem;
+import akka.actor.CoordinatedShutdown;
 import akka.actor.FSM;
 import akka.actor.OneForOneStrategy;
 import akka.actor.Props;
@@ -318,6 +319,17 @@ public void preStart() throws Exception {
                     clientActorRefsNotificationDelay);
         }
         clientActorRefs.add(getSelf());
+
+        closeConnectionBeforeTerminatingCluster();
+    }
+
+    private void closeConnectionBeforeTerminatingCluster() {
+        CoordinatedShutdown.get(getContext().getSystem())
+                .addActorTerminationTask(
+                        CoordinatedShutdown.PhaseBeforeServiceUnbind(),
+                        "closeConnection",
+                        getSelf(),
+                        Optional.of(CloseConnection.of(connectionId(), DittoHeaders.empty())));
     }
 
     @Override