fix: add protection to download createdtask

aneoconsulting · Sep 12, 2024 · 20ce11b · 20ce11b
1 parent 3718678
commit 20ce11b
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 6 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -124,7 +124,9 @@ jobs:
         run: |
          mkdir /tmp/sockets
          chmod 777 /tmp/sockets
-         docker run --rm -d --name htcmock -u $(id -u) -e ComputePlane__WorkerChannel__Address=/cache/worker.sock   -e ComputePlane__AgentChannel__Address=/cache/agent.sock -v /tmp/sockets:/cache -v  /tmp/Results:/tmp/Results dockerhubaneo/armonik_core_htcmock_test_worker:0.25.0
+         mkdir /tmp/test
+         chmod 777 /tmp/test
+         docker run --rm -d --name htcmock -u $(id -u) -e ComputePlane__WorkerChannel__Address=/cache/worker.sock   -e ComputePlane__AgentChannel__Address=/cache/agent.sock -v /tmp/sockets:/cache -v  /tmp/test/Results:/tmp/test/Results dockerhubaneo/armonik_core_htcmock_test_worker:0.25.0
 
       - name: Run TaskReRunner
         working-directory: src/TaskReRunner

diff --git a/src/TaskDumper/Program.cs b/src/TaskDumper/Program.cs
@@ -329,12 +329,19 @@ await File.WriteAllTextAsync(folder + "Subtasks.json",
     await foreach (var result in resultsCreated)
     {
       results[result.ResultId] = result;
+      var createdResponse = resultClient.GetResult(new GetResultRequest
+                                                   {
+                                                     ResultId = taskResponse.Task.PayloadId,
+                                                   });
       // Put subtask results in var folder + "Results"
-      await File.WriteAllBytesAsync(Path.Combine(folder + "Results",
-                                                 result.ResultId),
-                                    await resultClient.DownloadResultData(taskResponse.Task.SessionId,
-                                                                          result.ResultId,
-                                                                          CancellationToken.None) ?? Encoding.ASCII.GetBytes(""));
+      if (createdResponse.Result.Status == ResultStatus.Completed)
+      {
+        await File.WriteAllBytesAsync(Path.Combine(folder + "Results",
+                                                   result.ResultId),
+                                      await resultClient.DownloadResultData(taskResponse.Task.SessionId,
+                                                                            result.ResultId,
+                                                                            CancellationToken.None) ?? Encoding.ASCII.GetBytes(""));
+      }
     }
 
     await File.WriteAllTextAsync(folder + "CreatedResults.json",