rebase

wepay · Dec 8, 2021 · 09a7a01 · 09a7a01
1 parent a8ac9f3
commit 09a7a01
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/waltz-server/src/main/java/com/wepay/waltz/server/internal/Partition.java b/waltz-server/src/main/java/com/wepay/waltz/server/internal/Partition.java
@@ -279,13 +279,16 @@ public void setPartitionClient(PartitionClient client) {
     /**
      * Removes the client from the list of clients that are part of this partition.
      * @param client The client that has to be removed from this partition.
+     * @return True if the partition is removed, otherwise returns False.
      */
-    public void removePartitionClient(PartitionClient client) {
+    public boolean removePartitionClient(PartitionClient client) {
         synchronized (partitionClientSeqNums) {
             Long currentSeqNum = partitionClientSeqNums.get(client.clientId());
             if (currentSeqNum != null && currentSeqNum.equals(client.seqNum())) {
                 partitionClientSeqNums.remove(client.clientId());
+                return true;
             }
+            return false;
         }
     }
 

diff --git a/waltz-server/src/main/java/com/wepay/waltz/server/internal/WaltzServerHandler.java b/waltz-server/src/main/java/com/wepay/waltz/server/internal/WaltzServerHandler.java
@@ -235,8 +235,15 @@ public void onChannelActive() {
         @Override
         public void onChannelInactive() {
             synchronized (partitions) {
+                List<Integer> partitionClientsNotRemoved = new ArrayList<>();
                 for (Partition partition : partitions.values()) {
-                    partition.removePartitionClient(handler);
+                    if (!partition.removePartitionClient(handler)) {
+                        partitionClientsNotRemoved.add(partition.partitionId);
+                    }
+                }
+                if (!partitionClientsNotRemoved.isEmpty()) {
+                    LOGGER.info(String.format("WaltzServerHandler ClientId: %d, SeqNum: %d, PartitionsNotRemoved: %s",
+                        handler.clientId, handler.seqNum, partitionClientsNotRemoved));
                 }
             }
         }