1 files changed, 17 insertions, 1 deletions
diff --git a/container-search/src/main/java/com/yahoo/search/dispatch/rpc/RpcConnectionPool.java b/container-search/src/main/java/com/yahoo/search/dispatch/rpc/RpcConnectionPool.java
index fd8e0e4f81a..a93ddb0b360 100644
--- a/container-search/src/main/java/com/yahoo/search/dispatch/rpc/RpcConnectionPool.java
+++ b/container-search/src/main/java/com/yahoo/search/dispatch/rpc/RpcConnectionPool.java
@@ -1,11 +1,27 @@
 // Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
 package com.yahoo.search.dispatch.rpc;
 
+import com.yahoo.vespa.config.search.DispatchNodesConfig;
+
+import java.util.Collection;
+import java.util.List;
+
 /**
  * Interface for getting a connection given a node id.
  *
  * @author balderersheim
  */
-public interface RpcConnectionPool {
+public interface RpcConnectionPool extends AutoCloseable {
+
+    /** Returns a connection to the given node id. */
     Client.NodeConnection getConnection(int nodeId);
+
+
+    /** Will return a list of items that need a delayed close when updating node set. */
+    default Collection<? extends AutoCloseable> updateNodes(DispatchNodesConfig nodesConfig) { return List.of(); }
+
+    /** Shuts down all connections in the pool, and the underlying RPC client. */
+    @Override
+    void close();
+
 }