feat: 执行引擎任务调度配额限制 TencentBlueKing#261

wangyu096 · Jun 27, 2024 · 47520e5 · 47520e5
1 parent 88794d6
commit 47520e5
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/...in/java/com/tencent/bk/job/execute/engine/quota/limit/RunningJobResourceQuotaManager.java b/...in/java/com/tencent/bk/job/execute/engine/quota/limit/RunningJobResourceQuotaManager.java
@@ -121,6 +121,9 @@ public RunningJobResourceQuotaManager(StringRedisTemplate redisTemplate,
     }
 
     public void addJob(String appCode, ResourceScope resourceScope, long jobInstanceId) {
+        if (!runningJobResourceQuotaStore.isQuotaLimitEnabled()) {
+            return;
+        }
         long startTime = System.currentTimeMillis();
         RedisScript<Void> script = RedisScript.of(ADD_JOB_LUA_SCRIPT, Void.class);
 
@@ -144,6 +147,9 @@ private String convertAppCode(String appCode) {
     }
 
     public void removeJob(String appCode, ResourceScope resourceScope, long jobInstanceId) {
+        if (!runningJobResourceQuotaStore.isQuotaLimitEnabled()) {
+            return;
+        }
         long startTime = System.currentTimeMillis();
         RedisScript<Void> script = RedisScript.of(REMOVE_JOB_LUA_SCRIPT, Void.class);
 
@@ -169,6 +175,9 @@ public void removeJob(String appCode, ResourceScope resourceScope, long jobInsta
      */
     public ResourceQuotaCheckResultEnum checkResourceQuotaLimit(String appCode,
                                                                 ResourceScope resourceScope) {
+        if (!runningJobResourceQuotaStore.isQuotaLimitEnabled()) {
+            return ResourceQuotaCheckResultEnum.NO_LIMIT;
+        }
         long startTime = System.currentTimeMillis();
         RedisScript<String> script = RedisScript.of(CHECK_QUOTA_LUA_SCRIPT, String.class);