|
5 | 5 | import ai.superstream.model.MetadataMessage; |
6 | 6 | import ai.superstream.util.SuperstreamLogger; |
7 | 7 | import net.bytebuddy.asm.Advice; |
| 8 | +import java.util.AbstractMap; |
8 | 9 |
|
9 | 10 | import java.lang.reflect.Field; |
10 | 11 | import java.lang.reflect.Method; |
@@ -173,7 +174,8 @@ public static void onEnter(@Advice.AllArguments Object[] args) { |
173 | 174 | // Report the error to the client |
174 | 175 | try { |
175 | 176 | // Get metadata message before reporting |
176 | | - MetadataMessage metadataMessage = SuperstreamManager.getInstance().getOrFetchMetadataMessage(bootstrapServers, properties); |
| 177 | + AbstractMap.SimpleEntry<MetadataMessage, String> metadataResult = SuperstreamManager.getInstance().getOrFetchMetadataMessage(bootstrapServers, properties); |
| 178 | + MetadataMessage metadataMessage = metadataResult.getKey(); |
177 | 179 | SuperstreamManager.getInstance().reportClientInformation( |
178 | 180 | bootstrapServers, |
179 | 181 | properties, |
@@ -302,7 +304,8 @@ public static void onExit(@Advice.This Object producer) { |
302 | 304 | if (producerProps != null) { |
303 | 305 | String bootstrapServersProp = producerProps.getProperty("bootstrap.servers"); |
304 | 306 | if (bootstrapServersProp != null) { |
305 | | - metadataMessage = SuperstreamManager.getInstance().getOrFetchMetadataMessage(bootstrapServersProp, producerProps); |
| 307 | + AbstractMap.SimpleEntry<MetadataMessage, String> metadataResult = SuperstreamManager.getInstance().getOrFetchMetadataMessage(bootstrapServersProp, producerProps); |
| 308 | + metadataMessage = metadataResult.getKey(); |
306 | 309 | } |
307 | 310 | String topicsEnv = System.getenv("SUPERSTREAM_TOPICS_LIST"); |
308 | 311 | if (topicsEnv != null && !topicsEnv.trim().isEmpty()) { |
|
0 commit comments