From daedc4d7c77f7967d9da1f77eedd19bf30f4d68b Mon Sep 17 00:00:00 2001
From: iliax <ikuramshin@provectus.com>
Date: Wed, 21 Jun 2023 11:57:18 +0400
Subject: [PATCH] test fix

---
 .../provectus/kafka/ui/service/StatisticsService.java  | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/kafka-ui-api/src/main/java/com/provectus/kafka/ui/service/StatisticsService.java b/kafka-ui-api/src/main/java/com/provectus/kafka/ui/service/StatisticsService.java
index 19d946590c..0076ebbb88 100644
--- a/kafka-ui-api/src/main/java/com/provectus/kafka/ui/service/StatisticsService.java
+++ b/kafka-ui-api/src/main/java/com/provectus/kafka/ui/service/StatisticsService.java
@@ -9,6 +9,7 @@ import com.provectus.kafka.ui.model.Metrics;
 import com.provectus.kafka.ui.model.ServerStatusDTO;
 import com.provectus.kafka.ui.model.Statistics;
 import com.provectus.kafka.ui.service.metrics.MetricsCollector;
+import com.provectus.kafka.ui.service.metrics.v2.scrape.ScrapedClusterState;
 import java.util.List;
 import java.util.Map;
 import java.util.stream.Collectors;
@@ -44,7 +45,9 @@ public class StatisticsService {
                             getLogDirInfo(description, ac),
                             featureService.getAvailableFeatures(ac, cluster, description),
                             loadTopicConfigs(cluster),
-                            describeTopics(cluster)),
+                            describeTopics(cluster),
+                            loadClusterState(ac)
+                        ),
                         results ->
                             Statistics.builder()
                                 .status(ServerStatusDTO.ONLINE)
@@ -55,6 +58,7 @@ public class StatisticsService {
                                 .features((List<ClusterFeature>) results[2])
                                 .topicConfigs((Map<String, List<ConfigEntry>>) results[3])
                                 .topicDescriptions((Map<String, TopicDescription>) results[4])
+                                .clusterState((ScrapedClusterState) results[5])
                                 .build()
                     ))))
         .doOnError(e ->
@@ -76,4 +80,8 @@ public class StatisticsService {
     return adminClientService.get(c).flatMap(ReactiveAdminClient::getTopicsConfig);
   }
 
+  private Mono<ScrapedClusterState> loadClusterState(ReactiveAdminClient ac){
+    return ScrapedClusterState.scrape(ac);
+  }
+
 }