0ct0pu5/kafka-ui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

ISSUE-1128: Add respect to cluster's order in ClustersStorage (#1130)

Browse source 
* add respect to cluster's order in ClustersStorage
* minor cleaning
This commit is contained in:
Ilya Kuramshin 2021-11-26 14:50:32 +03:00 • committed by GitHub
parent 53a641cb58
commit d674808f3c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 24 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
kafka-ui-api/src/main/java/com/provectus/kafka/ui/service/ClustersMetricsScheduler.java
View file

@ -1,6 +1,5 @@
package com.provectus.kafka.ui.service;
import java.util.Map;
import javax.annotation.PostConstruct;
import lombok.RequiredArgsConstructor;
import lombok.extern.log4j.Log4j2;
@ -24,10 +23,9 @@ public class ClustersMetricsScheduler {
initialDelayString = "${kafka.update-metrics-rate-millis:30000}"
)
public void updateMetrics() {
Flux.fromIterable(clustersStorage.getKafkaClustersMap().entrySet())
Flux.fromIterable(clustersStorage.getKafkaClusters())
.parallel()
.runOn(Schedulers.parallel())
.map(Map.Entry::getValue)
.flatMap(cluster -> {
log.debug("Start getting metrics for kafkaCluster: {}", cluster.getName());
return metricsService.updateCache(cluster)

27
kafka-ui-api/src/main/java/com/provectus/kafka/ui/service/ClustersStorage.java
View file

@ -1,33 +1,22 @@
package com.provectus.kafka.ui.service;
import com.google.common.collect.ImmutableMap;
import com.provectus.kafka.ui.config.ClustersProperties;
import com.provectus.kafka.ui.mapper.ClusterMapper;
import com.provectus.kafka.ui.model.KafkaCluster;
import java.util.Collection;
import java.util.Map;
import java.util.Optional;
import java.util.concurrent.ConcurrentHashMap;
import javax.annotation.PostConstruct;
import lombok.RequiredArgsConstructor;
import org.mapstruct.factory.Mappers;
import org.springframework.stereotype.Component;
@Component
@RequiredArgsConstructor
public class ClustersStorage {
private final Map<String, KafkaCluster> kafkaClusters = new ConcurrentHashMap<>();
private final ImmutableMap<String, KafkaCluster> kafkaClusters;
private final ClustersProperties clusterProperties;
private final ClusterMapper clusterMapper = Mappers.getMapper(ClusterMapper.class);
@PostConstruct
public void init() {
for (ClustersProperties.Cluster clusterProperties : clusterProperties.getClusters()) {
KafkaCluster cluster = clusterMapper.toKafkaCluster(clusterProperties);
kafkaClusters.put(clusterProperties.getName(), cluster);
}
public ClustersStorage(ClustersProperties properties, ClusterMapper mapper) {
var builder = ImmutableMap.<String, KafkaCluster>builder();
properties.getClusters().forEach(c -> builder.put(c.getName(), mapper.toKafkaCluster(c)));
this.kafkaClusters = builder.build();
}
public Collection<KafkaCluster> getKafkaClusters() {
@ -37,8 +26,4 @@ public class ClustersStorage {
public Optional<KafkaCluster> getClusterByName(String clusterName) {
return Optional.ofNullable(kafkaClusters.get(clusterName));
}
public Map<String, KafkaCluster> getKafkaClustersMap() {
return kafkaClusters;
}
}