From 27576981b855417ca3bf1ab6d99675d3a49e2996 Mon Sep 17 00:00:00 2001
From: Andrew Prudhomme <asp@yelp.com>
Date: Mon, 21 Oct 2024 10:47:46 -0700
Subject: [PATCH] Remove replicas on NOT_FOUND status exception

---
 .../com/yelp/nrtsearch/server/nrt/NRTPrimaryNode.java  | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/src/main/java/com/yelp/nrtsearch/server/nrt/NRTPrimaryNode.java b/src/main/java/com/yelp/nrtsearch/server/nrt/NRTPrimaryNode.java
index 02126728c..2b80121f1 100644
--- a/src/main/java/com/yelp/nrtsearch/server/nrt/NRTPrimaryNode.java
+++ b/src/main/java/com/yelp/nrtsearch/server/nrt/NRTPrimaryNode.java
@@ -216,18 +216,20 @@ void sendNewNRTPointToReplicas() {
         Status status = e.getStatus();
         if (status.getCode().equals(Status.UNAVAILABLE.getCode())) {
           logger.warn(
-              "NRTPRimaryNode: sendNRTPoint, lost connection to replicaId: {} host: {} port: {}",
+              "NRTPrimaryNode: sendNRTPoint, lost connection to replicaId: {} host: {} port: {}",
               replicaDetails.replicaId,
               replicaDetails.replicationServerClient.getHost(),
               replicaDetails.replicationServerClient.getPort());
           currentReplicaServerClient.close();
           it.remove();
-        } else if (status.getCode().equals(Status.FAILED_PRECONDITION.getCode())) {
+        } else if (status.getCode().equals(Status.FAILED_PRECONDITION.getCode())
+            || status.getCode().equals(Status.NOT_FOUND.getCode())) {
           logger.warn(
-              "NRTPRimaryNode: sendNRTPoint, replicaId: {} host: {} port: {} cannot process nrt point, closing connection",
+              "NRTPrimaryNode: sendNRTPoint, replicaId: {} host: {} port: {} cannot process nrt point, closing connection: {}",
               replicaDetails.replicaId,
               replicaDetails.replicationServerClient.getHost(),
-              replicaDetails.replicationServerClient.getPort());
+              replicaDetails.replicationServerClient.getPort(),
+              status);
           currentReplicaServerClient.close();
           it.remove();
         }