feat: Add result creation in Worker TaskHandler

aneoconsulting · Apr 6, 2023 · 38c913e · 38c913e
1 parent db6ef03
commit 38c913e
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/packages/csharp/ArmoniK.Api.Worker/Worker/ITaskHandler.cs b/packages/csharp/ArmoniK.Api.Worker/Worker/ITaskHandler.cs
@@ -111,4 +111,13 @@ Task<CreateTaskReply> CreateTasksAsync(IEnumerable<TaskRequest> tasks,
   /// <returns></returns>
   Task SendResult(string key,
                   byte[] data);
+
+  /// <summary>
+  ///   Create results metadata
+  /// </summary>
+  /// <param name="results">The collection of results to be created</param>
+  /// <returns>
+  ///   The result creation response
+  /// </returns>
+  Task<CreateResultsMetaDataResponse> CreateResultsMetaDataAsync(IEnumerable<CreateResultsMetaDataRequest.Types.ResultCreate> results);
 }
diff --git a/packages/csharp/ArmoniK.Api.Worker/Worker/TaskHandler.cs b/packages/csharp/ArmoniK.Api.Worker/Worker/TaskHandler.cs
@@ -135,6 +135,20 @@ public Task<byte[]> RequestCommonData(string key)
   public Task<byte[]> RequestDirectData(string key)
     => throw new NotImplementedException();
 
+  /// <inheritdoc />
+  public async Task<CreateResultsMetaDataResponse> CreateResultsMetaDataAsync(IEnumerable<CreateResultsMetaDataRequest.Types.ResultCreate> results)
+    => await client_.CreateResultsMetaDataAsync(new CreateResultsMetaDataRequest
+                                                {
+                                                  CommunicationToken = Token,
+                                                  Results =
+                                                  {
+                                                    results,
+                                                  },
+                                                  SessionId = sessionId_,
+                                                })
+                    .ConfigureAwait(false);
+
+
   /// <inheritdoc />
   public async Task SendResult(string key,
                                byte[] data)