Merge pull request #667 from conductor-oss/rate_limit_policy

Added rate limit policy

Author: v1r3n

common/src/main/java/com/netflix/conductor/common/metadata/workflow/RateLimitConfig.java

@@ -12,12 +12,23 @@
  */
 package com.netflix.conductor.common.metadata.workflow;
 
+import com.netflix.conductor.annotations.protogen.ProtoEnum;
 import com.netflix.conductor.annotations.protogen.ProtoField;
 import com.netflix.conductor.annotations.protogen.ProtoMessage;
 
 /** Rate limit configuration for workflows */
 @ProtoMessage
 public class RateLimitConfig {
+
+    /** Rate limit policy defining how to handle requests exceeding the limit */
+    @ProtoEnum
+    public enum RateLimitPolicy {
+        /** Queue the request until capacity is available */
+        QUEUE,
+        /** Reject the request immediately */
+        REJECT
+    }
+
     /**
      * Key that defines the rate limit. Rate limit key is a combination of workflow payload such as
      * name, or correlationId etc.
@@ -29,6 +40,10 @@ public class RateLimitConfig {
     @ProtoField(id = 2)
     private int concurrentExecLimit;
 
+    /** Policy to apply when rate limit is exceeded */
+    @ProtoField(id = 3)
+    private RateLimitPolicy policy = RateLimitPolicy.QUEUE;
+
     public String getRateLimitKey() {
         return rateLimitKey;
     }
@@ -44,4 +59,12 @@ public int getConcurrentExecLimit() {
     public void setConcurrentExecLimit(int concurrentExecLimit) {
         this.concurrentExecLimit = concurrentExecLimit;
     }
+
+    public RateLimitPolicy getPolicy() {
+        return policy;
+    }
+
+    public void setPolicy(RateLimitPolicy policy) {
+        this.policy = policy;
+    }
 }

grpc/src/main/proto/model/ratelimitconfig.proto

@@ -7,6 +7,11 @@ option java_outer_classname = "RateLimitConfigPb";
 option go_package = "github.com/netflix/conductor/client/gogrpc/conductor/model";
 
 message RateLimitConfig {
+    enum RateLimitPolicy {
+        QUEUE = 0;
+        REJECT = 1;
+    }
     string rate_limit_key = 1;
     int32 concurrent_exec_limit = 2;
+    RateLimitConfig.RateLimitPolicy policy = 3;
 }