ZOOKEEPER-5010: purge orphaned ephemerals after DIFF sync

Author: seekskyworld

zookeeper-server/src/main/java/org/apache/zookeeper/server/quorum/Learner.java

@@ -31,6 +31,7 @@
 import java.nio.ByteBuffer;
 import java.util.ArrayDeque;
 import java.util.Deque;
+import java.util.HashSet;
 import java.util.Map;
 import java.util.Map.Entry;
 import java.util.Set;
@@ -54,7 +55,9 @@
 import org.apache.zookeeper.server.Request;
 import org.apache.zookeeper.server.ServerCnxn;
 import org.apache.zookeeper.server.ServerMetrics;
+import org.apache.zookeeper.server.SessionTracker;
 import org.apache.zookeeper.server.TxnLogEntry;
+import org.apache.zookeeper.server.ZKDatabase;
 import org.apache.zookeeper.server.ZooTrace;
 import org.apache.zookeeper.server.quorum.QuorumPeer.QuorumServer;
 import org.apache.zookeeper.server.quorum.flexible.QuorumVerifier;
@@ -558,6 +561,7 @@ protected void syncWithLeader(long newLeaderZxid) throws Exception {
         boolean snapshotNeeded = true;
         boolean syncSnapshot = false;
         readPacket(qp);
+        boolean diffSync = qp.getType() == Leader.DIFF;
         Deque<Long> packetsCommitted = new ArrayDeque<>();
         Deque<PacketInFlight> packetsNotCommitted = new ArrayDeque<>();
         Deque<Request> requestsToAck = new ArrayDeque<>();
@@ -613,6 +617,10 @@ protected void syncWithLeader(long newLeaderZxid) throws Exception {
             }
             zk.getZKDatabase().initConfigInZKDatabase(self.getQuorumVerifier());
             zk.createSessionTracker();
+            // DIFF keeps the local tree; clear ephemerals without sessions before applying new transactions.
+            if (diffSync) {
+                purgeOrphanedEphemerals();
+            }
 
             long lastQueued = 0;
 
@@ -845,6 +853,43 @@ protected void syncWithLeader(long newLeaderZxid) throws Exception {
             // New server type need to handle in-flight packets
             throw new UnsupportedOperationException("Unknown server type");
         }
+
+    }
+
+    void purgeOrphanedEphemerals() {
+        if (zk == null) {
+            return;
+        }
+        SessionTracker sessionTracker = zk.getSessionTracker();
+        if (sessionTracker == null) {
+            return;
+        }
+        ZKDatabase zkDatabase = zk.getZKDatabase();
+        if (zkDatabase == null) {
+            return;
+        }
+
+        Set<Long> globalSessions = sessionTracker.globalSessions();
+        Set<Long> localSessions = sessionTracker.localSessions();
+        Set<Long> sessionsWithEphemerals = new HashSet<>(zkDatabase.getSessions());
+        if (sessionsWithEphemerals.isEmpty()) {
+            return;
+        }
+
+        long zxid = zkDatabase.getDataTreeLastProcessedZxid();
+        for (Long sessionId : sessionsWithEphemerals) {
+            if (globalSessions.contains(sessionId)
+                || localSessions.contains(sessionId)
+                || (sessionTracker instanceof UpgradeableSessionTracker
+                    && ((UpgradeableSessionTracker) sessionTracker).isUpgradingSession(sessionId))) {
+                continue;
+            }
+            LOG.warn(
+                "Removing ephemeral nodes for unknown session 0x{} after DIFF sync",
+                Long.toHexString(sessionId));
+            zkDatabase.killSession(sessionId, zxid);
+            sessionTracker.removeSession(sessionId);
+        }
     }
 
     protected void revalidate(QuorumPacket qp) throws IOException {

zookeeper-server/src/test/java/org/apache/zookeeper/server/quorum/LearnerTest.java

@@ -24,6 +24,8 @@
 import static org.hamcrest.CoreMatchers.is;
 import static org.hamcrest.MatcherAssert.assertThat;
 import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static org.junit.jupiter.api.Assertions.assertNull;
 import static org.junit.jupiter.api.Assertions.assertThrows;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 import static org.junit.jupiter.api.Assertions.fail;
@@ -317,6 +319,39 @@ public void syncTest() throws Exception {
         }
     }
 
+    @Test
+    public void testPurgeOrphanedEphemerals() throws Exception {
+        File tmpFile = File.createTempFile("test", ".dir", testData);
+        tmpFile.delete();
+        SimpleLearner sl = null;
+        try {
+            FileTxnSnapLog ftsl = new FileTxnSnapLog(tmpFile, tmpFile);
+            sl = new SimpleLearner(ftsl);
+
+            long sessionId = 0x1234L;
+            TxnHeader hdr = new TxnHeader(sessionId, 1, 1L, 1L, ZooDefs.OpCode.create);
+            CreateTxn txn = new CreateTxn(
+                "/eph",
+                new byte[0],
+                new ArrayList<ACL>(),
+                true,
+                sl.zk.getZKDatabase().getNode("/").stat.getCversion());
+            sl.zk.getZKDatabase().processTxn(hdr, txn, null);
+
+            assertNotNull(sl.zk.getZKDatabase().getNode("/eph"), "Ephemeral node should exist before cleanup");
+
+            sl.zk.startupWithoutServing();
+            sl.purgeOrphanedEphemerals();
+
+            assertNull(sl.zk.getZKDatabase().getNode("/eph"), "Ephemeral node should be removed for unknown session");
+        } finally {
+            if (sl != null) {
+                sl.zk.shutdown();
+            }
+            TestUtils.deleteFileRecursively(tmpFile);
+        }
+    }
+
     @Test
     public void truncFailTest() throws Exception {
         final boolean[] exitProcCalled = {false};