Check arguments in StageExecutionId, TaskId, StageId,TableWriteInfo

prestodb · Dec 15, 2023 · 93d7a57 · 93d7a57
1 parent 87688fb
commit 93d7a57
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/presto-main/src/main/java/com/facebook/presto/execution/StageExecutionId.java b/presto-main/src/main/java/com/facebook/presto/execution/StageExecutionId.java
@@ -50,6 +50,7 @@ public static StageExecutionId valueOf(List<String> ids)
     public StageExecutionId(StageId stageId, int id)
     {
         this.stageId = requireNonNull(stageId, "stageId is null");
+        checkArgument(id >= 0, "id is negative: %s", id);
         this.id = id;
     }
 

diff --git a/presto-main/src/main/java/com/facebook/presto/execution/StageId.java b/presto-main/src/main/java/com/facebook/presto/execution/StageId.java
@@ -48,6 +48,7 @@ public static StageId valueOf(List<String> ids)
     public StageId(QueryId queryId, int id)
     {
         this.queryId = requireNonNull(queryId, "queryId is null");
+        checkArgument(id >= 0, "id is negative: %s", id);
         this.id = id;
     }
 

diff --git a/presto-main/src/main/java/com/facebook/presto/execution/TaskId.java b/presto-main/src/main/java/com/facebook/presto/execution/TaskId.java
@@ -52,7 +52,7 @@ public TaskId(StageExecutionId stageExecutionId, int id, int attemptNumber)
     {
         this.stageExecutionId = requireNonNull(stageExecutionId, "stageExecutionId");
         this.attemptNumber = attemptNumber;
-        checkArgument(id >= 0, "id is negative");
+        checkArgument(id >= 0, "id is negative: %s", id);
         this.id = id;
     }