|
32 | 32 | import io.opentelemetry.api.trace.Tracer; |
33 | 33 |
|
34 | 34 | import javax.annotation.Nullable; |
| 35 | + |
35 | 36 | import java.io.FileInputStream; |
36 | 37 | import java.io.IOException; |
37 | 38 | import java.io.InputStream; |
38 | 39 | import java.time.Duration; |
39 | 40 | import java.time.Instant; |
40 | | -import java.util.ArrayList; |
41 | | -import java.util.List; |
42 | 41 | import java.util.Optional; |
43 | 42 | import java.util.UUID; |
44 | 43 | import java.util.concurrent.TimeUnit; |
@@ -307,49 +306,6 @@ public void terminate(String instanceId, @Nullable Object output) { |
307 | 306 | this.sidecarClient.terminateInstance(builder.build()); |
308 | 307 | } |
309 | 308 |
|
310 | | - @Override |
311 | | - public OrchestrationStatusQueryResult queryInstances(OrchestrationStatusQuery query) { |
312 | | - OrchestratorService.InstanceQuery.Builder instanceQueryBuilder = OrchestratorService.InstanceQuery.newBuilder(); |
313 | | - Optional.ofNullable(query.getCreatedTimeFrom()).ifPresent(createdTimeFrom -> |
314 | | - instanceQueryBuilder.setCreatedTimeFrom(DataConverter.getTimestampFromInstant(createdTimeFrom))); |
315 | | - Optional.ofNullable(query.getCreatedTimeTo()).ifPresent(createdTimeTo -> |
316 | | - instanceQueryBuilder.setCreatedTimeTo(DataConverter.getTimestampFromInstant(createdTimeTo))); |
317 | | - Optional.ofNullable(query.getContinuationToken()).ifPresent(token -> |
318 | | - instanceQueryBuilder.setContinuationToken(StringValue.of(token))); |
319 | | - Optional.ofNullable(query.getInstanceIdPrefix()).ifPresent(prefix -> |
320 | | - instanceQueryBuilder.setInstanceIdPrefix(StringValue.of(prefix))); |
321 | | - instanceQueryBuilder.setFetchInputsAndOutputs(query.isFetchInputsAndOutputs()); |
322 | | - instanceQueryBuilder.setMaxInstanceCount(query.getMaxInstanceCount()); |
323 | | - query.getRuntimeStatusList().forEach(runtimeStatus -> |
324 | | - Optional.ofNullable(runtimeStatus).ifPresent(status -> |
325 | | - instanceQueryBuilder.addRuntimeStatus(OrchestrationRuntimeStatus.toProtobuf(status)))); |
326 | | - query.getTaskHubNames().forEach(taskHubName -> Optional.ofNullable(taskHubName).ifPresent(name -> |
327 | | - instanceQueryBuilder.addTaskHubNames(StringValue.of(name)))); |
328 | | - OrchestratorService.QueryInstancesResponse queryInstancesResponse = this.sidecarClient |
329 | | - .queryInstances(OrchestratorService.QueryInstancesRequest.newBuilder().setQuery(instanceQueryBuilder).build()); |
330 | | - return toQueryResult(queryInstancesResponse, query.isFetchInputsAndOutputs()); |
331 | | - } |
332 | | - |
333 | | - private OrchestrationStatusQueryResult toQueryResult( |
334 | | - OrchestratorService.QueryInstancesResponse queryInstancesResponse, boolean fetchInputsAndOutputs) { |
335 | | - List<OrchestrationMetadata> metadataList = new ArrayList<>(); |
336 | | - queryInstancesResponse.getOrchestrationStateList().forEach(state -> { |
337 | | - metadataList.add(new OrchestrationMetadata(state, this.dataConverter, fetchInputsAndOutputs)); |
338 | | - }); |
339 | | - return new OrchestrationStatusQueryResult(metadataList, queryInstancesResponse.getContinuationToken().getValue()); |
340 | | - } |
341 | | - |
342 | | - @Override |
343 | | - public void createTaskHub(boolean recreateIfExists) { |
344 | | - this.sidecarClient.createTaskHub(OrchestratorService.CreateTaskHubRequest.newBuilder() |
345 | | - .setRecreateIfExists(recreateIfExists).build()); |
346 | | - } |
347 | | - |
348 | | - @Override |
349 | | - public void deleteTaskHub() { |
350 | | - this.sidecarClient.deleteTaskHub(OrchestratorService.DeleteTaskHubRequest.newBuilder().build()); |
351 | | - } |
352 | | - |
353 | 309 | @Override |
354 | 310 | public PurgeResult purgeInstance(String instanceId) { |
355 | 311 | OrchestratorService.PurgeInstancesRequest request = OrchestratorService.PurgeInstancesRequest.newBuilder() |
|
0 commit comments