minor cleanup and refactor

flipkart-incubator · Sep 16, 2024 · 1200509 · 1200509
1 parent 58e9c4d
commit 1200509
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 5 deletions.
diff --git a/server/src/main/java/com/flipkart/varadhi/cluster/MessageExchange.java b/server/src/main/java/com/flipkart/varadhi/cluster/MessageExchange.java
@@ -2,7 +2,6 @@
 
 import com.flipkart.varadhi.cluster.messages.ClusterMessage;
 import com.flipkart.varadhi.cluster.messages.ResponseMessage;
-import com.flipkart.varadhi.exceptions.NotImplementedException;
 import com.flipkart.varadhi.utils.JsonMapper;
 import io.vertx.core.eventbus.DeliveryOptions;
 import io.vertx.core.eventbus.EventBus;

diff --git a/server/src/main/java/com/flipkart/varadhi/cluster/MessageRouter.java b/server/src/main/java/com/flipkart/varadhi/cluster/MessageRouter.java
@@ -1,8 +1,10 @@
 package com.flipkart.varadhi.cluster;
 
 
-import com.flipkart.varadhi.cluster.messages.*;
-import com.flipkart.varadhi.exceptions.NotImplementedException;
+import com.flipkart.varadhi.cluster.messages.ClusterMessage;
+import com.flipkart.varadhi.cluster.messages.MsgHandler;
+import com.flipkart.varadhi.cluster.messages.RequestHandler;
+import com.flipkart.varadhi.cluster.messages.ResponseMessage;
 import com.flipkart.varadhi.exceptions.VaradhiException;
 import com.flipkart.varadhi.utils.JsonMapper;
 import io.vertx.core.eventbus.DeliveryOptions;

diff --git a/server/src/main/java/com/flipkart/varadhi/qos/RateLimiter.java b/server/src/main/java/com/flipkart/varadhi/qos/RateLimiter.java
@@ -53,7 +53,7 @@ public RateLimiter(VaradhiTopicService varadhiTopicService, int windowSize, Weig
         this.windowSize = windowSize;
         this.weightFunction = weightFunction;
         this.loadHistory = TreeMultiset.create();
-        this.suppressionData = new SuppressionData();
+        this.suppressionData = new SuppressionData<>();
         sendSuppressionFactorToWeb();
     }
 

diff --git a/server/src/test/java/com/flipkart/varadhi/web/produce/ProduceTestBase.java b/server/src/test/java/com/flipkart/varadhi/web/produce/ProduceTestBase.java
@@ -33,7 +33,7 @@ public class ProduceTestBase extends WebTestBase {
     String deployedRegion = "region1";
     String serviceHost = "localhost";
     TrafficAggregator trafficAggregator;
-    SuppressorHandler suppressorHandler;
+    SuppressorHandler<Float> suppressorHandler;
     ArgumentCaptor<Message> msgCapture;
     String topicPath = "/projects/project1/topics/topic1/produce";
     String topicFullName = "project1.topic1";