fixed cr comments

Author: RoeyoOgen

gateway-ha/src/main/java/io/trino/gateway/ha/resource/GatewayWebAppResource.java

@@ -16,10 +16,7 @@
 import com.google.common.base.Strings;
 import com.google.inject.Inject;
 import io.trino.gateway.ha.clustermonitor.ClusterStats;
-import io.trino.gateway.ha.config.HaGatewayConfiguration;
-import io.trino.gateway.ha.config.ProxyBackendConfiguration;
-import io.trino.gateway.ha.config.RulesType;
-import io.trino.gateway.ha.config.UIConfiguration;
+import io.trino.gateway.ha.config.*;
 import io.trino.gateway.ha.domain.Result;
 import io.trino.gateway.ha.domain.RoutingRule;
 import io.trino.gateway.ha.domain.TableData;
@@ -74,10 +71,10 @@ public class GatewayWebAppResource
     private final QueryHistoryManager queryHistoryManager;
     private final BackendStateManager backendStateManager;
     private final ResourceGroupsManager resourceGroupsManager;
+    private final RoutingRulesConfiguration routingRules;
     // TODO Avoid putting mutable objects in fields
     private final UIConfiguration uiConfiguration;
     private final RoutingRulesManager routingRulesManager;
-    private final HaGatewayConfiguration haGatewayConfiguration;
 
     @Inject
     public GatewayWebAppResource(
@@ -94,7 +91,7 @@ public GatewayWebAppResource(
         this.resourceGroupsManager = requireNonNull(resourceGroupsManager, "resourceGroupsManager is null");
         this.uiConfiguration = configuration.getUiConfiguration();
         this.routingRulesManager = requireNonNull(routingRulesManager, "routingRulesManager is null");
-        this.haGatewayConfiguration = configuration;
+        this.routingRules = configuration.getRoutingRules();
     }
 
     @POST
@@ -163,7 +160,7 @@ public Response getDistribution(QueryDistributionRequest query)
         Map<String, List<DistributionResponse.LineChart>> lineChartMap = lineChart.stream().collect(Collectors.groupingBy(DistributionResponse.LineChart::getName));
         List<DistributionResponse.DistributionChart> distributionChart = lineChartMap.values().stream().map(d -> {
             DistributionResponse.DistributionChart dc = new DistributionResponse.DistributionChart();
-            DistributionResponse.LineChart lc = d.getFirst();
+            DistributionResponse.LineChart lc = d.get(0);
             long sum = d.stream().collect(Collectors.summarizingLong(DistributionResponse.LineChart::getQueryCount)).getSum();
             dc.setQueryCount(sum);
             dc.setBackendUrl(lc.getBackendUrl());
@@ -449,7 +446,7 @@ public Response readExactMatchSourceSelector()
     public Response getRoutingRules()
             throws IOException
     {
-        if (haGatewayConfiguration.getRoutingRules().isRulesEngineEnabled() && haGatewayConfiguration.getRoutingRules().getRulesType() == RulesType.EXTERNAL) {
+        if (routingRules.isRulesEngineEnabled() && routingRules.getRulesType() == RulesType.EXTERNAL) {
             return Response.status(Response.Status.NO_CONTENT)
                     .entity(Result.fail("Routing rules are managed by an external service")).build();
         }