From 453bd6064b6ae04d59c96792273cbbb52728ac28 Mon Sep 17 00:00:00 2001
From: Viktor Lofgren <vlofgren@marginalia.nu>
Date: Tue, 20 Feb 2024 15:45:16 +0100
Subject: [PATCH] (minor) Add warm-up to GrpcMultiNodeChannelPool to speed up
 the initial messages

Without doing this, connections would be created lazily, which is probably never desirable.
---
 .../marginalia/service/client/GrpcMultiNodeChannelPool.java  | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/code/common/service-discovery/src/main/java/nu/marginalia/service/client/GrpcMultiNodeChannelPool.java b/code/common/service-discovery/src/main/java/nu/marginalia/service/client/GrpcMultiNodeChannelPool.java
index e816bd1a..2552dcf0 100644
--- a/code/common/service-discovery/src/main/java/nu/marginalia/service/client/GrpcMultiNodeChannelPool.java
+++ b/code/common/service-discovery/src/main/java/nu/marginalia/service/client/GrpcMultiNodeChannelPool.java
@@ -40,6 +40,11 @@ public class GrpcMultiNodeChannelPool<STUB> {
         this.channelConstructor = channelConstructor;
         this.stubConstructor = stubConstructor;
         this.nodeConfigurationWatcher = nodeConfigurationWatcher;
+
+        // Warm up the pool to reduce latency for the initial request
+        for (var node : nodeConfigurationWatcher.getQueryNodes()) {
+            apiForNode(node);
+        }
     }