Added rate limit warning

Author: pokki-deploy

config-layers/test/io/vertx/ext/web/handler/sockjs/processor/RateLimiterProcessor.properties

@@ -0,0 +1,3 @@
+#/io/vertx/ext/web/handler/sockjs/processor/RateLimiterProcessor
+
+warningOnly=false

src/main/java/io/vertx/reactivex/ext/web/handler/sockjs/processor/RateLimiterProcessor.java

@@ -32,6 +32,7 @@ public class RateLimiterProcessor implements BridgeEventProcessor {
   private int mMaximumSize = 50000;
   private int mExpireAfterAccess = 3600;
   private boolean mEnable = true;
+  private boolean mWarningOnly = true;
   private Logger mLog;
 
   @StartService
@@ -61,14 +62,16 @@ public Single<BridgeEventContext> process(Single<BridgeEventContext> pContext) {
 
               Bucket bucket = getCache().get(eventType).get(ctx.getBridgeEvent().socket().writeHandlerID(), () -> {
                 Refill refill = Refill.smooth(getRateOfTokenFill().getOrDefault(eventType, getDefaultRateOfTokenFill()),
-                         Duration.ofSeconds(getRateOfTokenFillDuration().getOrDefault(eventType, getDefaultRateOfTokenFillDuration())));
+                        Duration.ofSeconds(getRateOfTokenFillDuration().getOrDefault(eventType, getDefaultRateOfTokenFillDuration())));
                 Bandwidth limit = Bandwidth.classic(getTokenBucketSize().getOrDefault(eventType, getDefaultTokenBucketSize()), refill);
                 return Bucket4j.builder().addLimit(limit).build();
               });
 
               if (!bucket.tryConsume(1)) {
-                getLog().debug(() -> String.format("[%s] Rate limit crossed for connection:[%s],event:[%s]", ctx.getId(), ctx.getBridgeEvent().socket().writeHandlerID(), eventType.toString()));
-                ctx.getBridgeEvent().fail("Rate Limit Crossed");
+                if (!isWarningOnly()) {
+                  ctx.getBridgeEvent().fail("Rate Limit Crossed");
+                }
+                getLog().warn(() -> String.format("[%s] Rate limit crossed for connection:[%s],event:[%s]", ctx.getId(), ctx.getBridgeEvent().socket().writeHandlerID(), eventType.toString()));
               }
 
               return ctx;
@@ -181,4 +184,12 @@ public void setDefaultRateOfTokenFillDuration(int pDefaultRateOfTokenFillDuratio
     this.mDefaultRateOfTokenFillDuration = pDefaultRateOfTokenFillDuration;
   }
 
+  public boolean isWarningOnly() {
+    return mWarningOnly;
+  }
+
+  public void setWarningOnly(boolean pWarningOnly) {
+    this.mWarningOnly = pWarningOnly;
+  }
+
 }