[Feature] Add Burst control capability for Applications

components/apimgt/org.wso2.carbon.apimgt.api/src/main/java/org/wso2/carbon/apimgt/api/model/policy/ApplicationPolicy.java

@@ -19,6 +19,8 @@
 package org.wso2.carbon.apimgt.api.model.policy;
 
 public class ApplicationPolicy extends Policy {
+    private int rateLimitCount;
+    private String rateLimitTimeUnit;
     private int applicationId;
 
     private byte[] customAttributes;
@@ -43,10 +45,24 @@
 		this.customAttributes = customAttributes;
 	}
 
+    public String getRateLimitTimeUnit() {
+        return rateLimitTimeUnit;
+    }
+    public void setRateLimitTimeUnit(String rateLimitTimeUnit) {
+        this.rateLimitTimeUnit = rateLimitTimeUnit;
+    }
+    public int getRateLimitCount() {
+        return rateLimitCount;
+    }
+    public void setRateLimitCount(int rateLimitCount) {
+        this.rateLimitCount = rateLimitCount;
+    }
+
     @Override
     public String toString() {
         return "ApplicationPolicy [policyName=" + getPolicyName()
                 + ", applicationId =" + applicationId + ", description=" + getDescription() + ", defaultQuotaPolicy="
-                + getDefaultQuotaPolicy() + "]";
+                + getDefaultQuotaPolicy() + ", rateLimitCount=" + getRateLimitCount() + ", rateLimitTimeUnit="
+                + getRateLimitTimeUnit() + "]";
     }
 }

components/apimgt/org.wso2.carbon.apimgt.api/src/main/java/org/wso2/carbon/apimgt/api/model/subscription/ApplicationPolicy.java

@@ -23,9 +23,28 @@
  */
 public class ApplicationPolicy extends Policy {
 
+    private int rateLimitCount = -1;
+    private String rateLimitTimeUnit = null;
+
     @Override
     public String getCacheKey() {
 
         return POLICY_TYPE.APPLICATION + DELEM_PERIOD + super.getCacheKey();
     }
+
+    public String getRateLimitTimeUnit() {
+        return rateLimitTimeUnit;
+    }
+
+    public void setRateLimitTimeUnit(String rateLimitTimeUnit) {
+        this.rateLimitTimeUnit = rateLimitTimeUnit;
+    }
+
+    public int getRateLimitCount() {
+        return rateLimitCount;
+    }
+
+    public void setRateLimitCount(int rateLimitCount) {
+        this.rateLimitCount = rateLimitCount;
+    }
 }

components/apimgt/org.wso2.carbon.apimgt.common.analytics/src/main/java/org/wso2/carbon/apimgt/common/analytics/publishers/dto/enums/FaultSubCategories.java

@@ -43,6 +43,7 @@ public enum Throttling implements FaultSubCategory {
         BLOCKED,
         CUSTOM_POLICY_LIMIT_EXCEEDED,
         BURST_CONTROL_LIMIT_EXCEEDED,
+        APPLICATION_BURST_CONTROL_LIMIT_EXCEEDED,
         QUERY_TOO_DEEP,
         QUERY_TOO_COMPLEX,
         OTHER

components/apimgt/org.wso2.carbon.apimgt.gateway/src/main/java/org/wso2/carbon/apimgt/gateway/handlers/analytics/FaultCodeClassifier.java

@@ -109,6 +109,8 @@
             return FaultSubCategories.Throttling.CUSTOM_POLICY_LIMIT_EXCEEDED;
         case APIThrottleConstants.SUBSCRIPTION_BURST_THROTTLE_OUT_ERROR_CODE:
             return FaultSubCategories.Throttling.BURST_CONTROL_LIMIT_EXCEEDED;
+        case APIThrottleConstants.APPLICATION_BURST_THROTTLE_OUT_ERROR_CODE:
+            return FaultSubCategories.Throttling.APPLICATION_BURST_CONTROL_LIMIT_EXCEEDED;
         case APIThrottleConstants.GRAPHQL_QUERY_TOO_DEEP:
         case WebSocketApiConstants.FrameErrorConstants.GRAPHQL_QUERY_TOO_DEEP:
             return FaultSubCategories.Throttling.QUERY_TOO_DEEP;

components/apimgt/org.wso2.carbon.apimgt.gateway/src/main/java/org/wso2/carbon/apimgt/gateway/handlers/security/AuthenticationContext.java

@@ -42,6 +42,7 @@
     private String subscriber;
     private List<String> throttlingDataList;
     private int spikeArrestLimit;
+    private int applicationSpikesArrestLimit;
     private String subscriberTenantDomain;
     private String spikeArrestUnit;
     private boolean stopOnQuotaReach;
@@ -180,6 +181,23 @@
         this.consumerKey = consumerKey;
     }
 
+    public String getApplicationSpikesArrestUnit() {
+        return applicationSpikesArrestUnit;
+    }
+
+    public void setApplicationSpikesArrestUnit(String applicationSpikesArrestUnit) {
+        this.applicationSpikesArrestUnit = applicationSpikesArrestUnit;
+    }
+
+    private String applicationSpikesArrestUnit;
+
+    public int getApplicationSpikesArrestLimit(){
+        return applicationSpikesArrestLimit;
+    }
+    public void setApplicationSpikesArrestLimit(int applicationSpikesArrestLimit) {
+        this.applicationSpikesArrestLimit = applicationSpikesArrestLimit;
+    }
+
     public int getSpikeArrestLimit() {
         return spikeArrestLimit;
     }

components/apimgt/org.wso2.carbon.apimgt.gateway/src/main/java/org/wso2/carbon/apimgt/gateway/handlers/security/oauth/OAuthAuthenticator.java

@@ -352,6 +352,8 @@
             authContext.setSubscriberTenantDomain(info.getSubscriberTenantDomain());
             authContext.setSpikeArrestLimit(info.getSpikeArrestLimit());
             authContext.setSpikeArrestUnit(info.getSpikeArrestUnit());
+            authContext.setApplicationSpikesArrestLimit(info.getApplicationSpikeArrestLimit());
+            authContext.setApplicationSpikesArrestUnit(info.getApplicationSpikeArrestUnit());
             authContext.setStopOnQuotaReach(info.isStopOnQuotaReach());
             authContext.setIsContentAware(info.isContentAware());
             APISecurityUtils.setAuthenticationContext(synCtx, authContext, securityContextHeader);

components/apimgt/org.wso2.carbon.apimgt.gateway/src/main/java/org/wso2/carbon/apimgt/gateway/handlers/throttling/APIThrottleConstants.java

@@ -27,6 +27,7 @@ public class APIThrottleConstants {
     public static final int APPLICATION_THROTTLE_OUT_ERROR_CODE = 900803;
     public static final int SUBSCRIPTION_THROTTLE_OUT_ERROR_CODE = 900804;
     public static final int SUBSCRIPTION_BURST_THROTTLE_OUT_ERROR_CODE = 900807;
+    public static final int APPLICATION_BURST_THROTTLE_OUT_ERROR_CODE = 900809;
     public static final int BLOCKED_ERROR_CODE = 900805;
     public static final int CUSTOM_POLICY_THROTTLE_OUT_ERROR_CODE = 900806;
     public static final int CONNECTIONS_COUNT_THROTTLE_OUT_ERROR_CODE = 900808;
@@ -47,6 +48,8 @@ public class APIThrottleConstants {
     public static final String HARD_THROTTLING_CONFIGURATION = "hard_throttling_limits";
     public static final String PRODUCTION_HARD_LIMIT = "PRODUCTION_HARD_LIMIT";
     public static final String SUBSCRIPTION_BURST_LIMIT = "SUBSCRIPTION_BURST_LIMIT";
+    public static final String APPLICATION_BURST_LIMIT = "APPLICATION_BURST_LIMIT";
+    public static final String APPLICATION_BURST_LIMIT_EXCEEDED = "APPLICATION_BURST_LIMIT_EXCEED";
     public static final String SANDBOX_HARD_LIMIT = "SANDBOX_HARD_LIMIT";
     public static final String THROTTLED_OUT_REASON = APIConstants.THROTTLE_OUT_REASON_KEY;
     public static final String THROTTLED_NEXT_ACCESS_TIMESTAMP = "NEXT_ACCESS_TIME";