Allow fairness weight overrides via UpdateTaskQueueConfig API (#626)

<!-- Describe what has changed in this PR -->
**What changed?**
+ Added two update fairness weight maps to the UpdateTaskQueueConfig api
request.
+ Added a fairness_weights map in the taskQueueConfig message to enable
persistence and return response.

<!-- Tell your future self why have you made these changes -->
**Why?**
+ Allow override of fairness weights for individual fairness keys via
UpdateTaskQueueConfig api.

<!-- Are there any breaking changes on binary or code level? -->
**Breaking changes**
+ No.

<!-- If this breaks the Server, please provide the Server PR to merge
right after this PR was merged. -->
**Server PR**

---------

Co-authored-by: Stephan Behnke <[email protected]>
Co-authored-by: Stephan Behnke <[email protected]>

Author: sivagirish81

openapi/openapiv2.json

@@ -9065,6 +9065,21 @@
         "updateFairnessKeyRateLimitDefault": {
           "$ref": "#/definitions/UpdateTaskQueueConfigRequestRateLimitUpdate",
           "description": "Update to the default fairness key rate limit.\nIf not set, this configuration is unchanged.\nIf the `rate_limit` field in the `RateLimitUpdate` is missing, remove the existing rate limit."
+        },
+        "setFairnessWeightOverrides": {
+          "type": "object",
+          "additionalProperties": {
+            "type": "number",
+            "format": "float"
+          },
+          "description": "If set, overrides the fairness weight for each specified fairness key.\nFairness keys not listed in this map will keep their existing overrides (if any)."
+        },
+        "unsetFairnessWeightOverrides": {
+          "type": "array",
+          "items": {
+            "type": "string"
+          },
+          "description": "If set, removes any existing fairness weight overrides for each specified fairness key.\nFairness weights for corresponding keys fall back to the values set during task creation (if any),\nor to the default weight of 1.0."
         }
       }
     },
@@ -15292,6 +15307,14 @@
         "fairnessKeysRateLimitDefault": {
           "$ref": "#/definitions/v1RateLimitConfig",
           "description": "If set, each individual fairness key will be limited to this rate, scaled by the weight of the fairness key."
+        },
+        "fairnessWeightOverrides": {
+          "type": "object",
+          "additionalProperties": {
+            "type": "number",
+            "format": "float"
+          },
+          "description": "If set, overrides the fairness weights for the corresponding fairness keys."
         }
       }
     },

openapi/openapiv3.yaml

@@ -12339,6 +12339,12 @@ components:
           allOf:
             - $ref: '#/components/schemas/RateLimitConfig'
           description: If set, each individual fairness key will be limited to this rate, scaled by the weight of the fairness key.
+        fairnessWeightOverrides:
+          type: object
+          additionalProperties:
+            type: number
+            format: float
+          description: If set, overrides the fairness weights for the corresponding fairness keys.
     TaskQueueReachability:
       type: object
       properties:
@@ -12919,6 +12925,22 @@ components:
             Update to the default fairness key rate limit.
              If not set, this configuration is unchanged.
              If the `rate_limit` field in the `RateLimitUpdate` is missing, remove the existing rate limit.
+        setFairnessWeightOverrides:
+          type: object
+          additionalProperties:
+            type: number
+            format: float
+          description: |-
+            If set, overrides the fairness weight for each specified fairness key.
+             Fairness keys not listed in this map will keep their existing overrides (if any).
+        unsetFairnessWeightOverrides:
+          type: array
+          items:
+            type: string
+          description: |-
+            If set, removes any existing fairness weight overrides for each specified fairness key.
+             Fairness weights for corresponding keys fall back to the values set during task creation (if any),
+             or to the default weight of 1.0.
     UpdateTaskQueueConfigRequest_RateLimitUpdate:
       type: object
       properties:

temporal/api/taskqueue/v1/message.proto

@@ -343,4 +343,6 @@ message TaskQueueConfig {
     RateLimitConfig queue_rate_limit = 1;
     // If set, each individual fairness key will be limited to this rate, scaled by the weight of the fairness key.
     RateLimitConfig fairness_keys_rate_limit_default = 2;
+    // If set, overrides the fairness weights for the corresponding fairness keys.
+    map<string, float> fairness_weight_overrides = 3;
 }

temporal/api/workflowservice/v1/request_response.proto

@@ -2528,6 +2528,13 @@ message UpdateTaskQueueConfigRequest {
     // If not set, this configuration is unchanged.
     // If the `rate_limit` field in the `RateLimitUpdate` is missing, remove the existing rate limit.
     RateLimitUpdate update_fairness_key_rate_limit_default = 6;
+    // If set, overrides the fairness weight for each specified fairness key.
+    // Fairness keys not listed in this map will keep their existing overrides (if any).
+    map<string, float> set_fairness_weight_overrides = 7;
+    // If set, removes any existing fairness weight overrides for each specified fairness key.
+    // Fairness weights for corresponding keys fall back to the values set during task creation (if any),
+    // or to the default weight of 1.0.
+    repeated string unset_fairness_weight_overrides = 8;
 }
 
 message UpdateTaskQueueConfigResponse {