From 51569ce0a572daa464ae0e32bb896b653050786b Mon Sep 17 00:00:00 2001
From: Jonathan Klabunde Tomer <125505367+jkt-signal@users.noreply.github.com>
Date: Thu, 24 Apr 2025 08:29:58 -0700
Subject: [PATCH] Use cached partition topology for metrics/logs

---
 .../whispersystems/textsecuregcm/storage/MessagesCache.java    | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/service/src/main/java/org/whispersystems/textsecuregcm/storage/MessagesCache.java b/service/src/main/java/org/whispersystems/textsecuregcm/storage/MessagesCache.java
index c6c82d96d..f64a19db1 100644
--- a/service/src/main/java/org/whispersystems/textsecuregcm/storage/MessagesCache.java
+++ b/service/src/main/java/org/whispersystems/textsecuregcm/storage/MessagesCache.java
@@ -670,11 +670,10 @@ public class MessagesCache {
         .thenRun(() -> sample.stop(clearQueueTimer));
   }
 
-  // expensive—use for rare error logging only
   public String shardForSlot(int slot) {
     try {
       return redisCluster.withBinaryCluster(
-          connection -> ClusterPartitionParser.parse(connection.sync().clusterNodes()).getPartitionBySlot(slot).getUri().getHost());
+          connection -> connection.getPartitions().getPartitionBySlot(slot).getUri().getHost());
     } catch (Throwable ignored) {
       return "unknown";
     }