Skip to content

Commit

Permalink
Added exposure of job controller API (hapifhir#678)
Browse files Browse the repository at this point in the history
  • Loading branch information
jkiddo committed Apr 23, 2024
1 parent 4e2faee commit 31e64f6
Showing 1 changed file with 42 additions and 0 deletions.
42 changes: 42 additions & 0 deletions src/main/java/ca/uhn/fhir/jpa/starter/web/JobController.java
Original file line number Diff line number Diff line change
@@ -0,0 +1,42 @@
package ca.uhn.fhir.jpa.starter.web;

import ca.uhn.fhir.batch2.api.IJobCoordinator;
import ca.uhn.fhir.batch2.api.JobOperationResultJson;
import ca.uhn.fhir.batch2.model.JobInstance;
import ca.uhn.fhir.batch2.model.StatusEnum;
import ca.uhn.fhir.batch2.models.JobInstanceFetchRequest;
import jakarta.validation.constraints.Min;
import org.springframework.data.domain.Sort;
import org.springframework.http.MediaType;
import org.springframework.web.bind.annotation.*;

import java.util.List;

@RestController
@RequestMapping("control")
public class JobController {
private final IJobCoordinator theJobCoordinator;

public JobController(IJobCoordinator theJobCoordinator) {
this.theJobCoordinator = theJobCoordinator;
}

@RequestMapping(value = JobController.JOBS, method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_VALUE)
public List<JobInstance> getAllJobs(@RequestParam(name = "pageStart") @Min(0) int pageStart, @RequestParam(name = "batchSize") int batchSize, @RequestParam(name = "jobStatus", required = false) StatusEnum jobStatus) {
JobInstanceFetchRequest jobInstanceFetchRequest = new JobInstanceFetchRequest();
jobInstanceFetchRequest.setPageStart(pageStart);
jobInstanceFetchRequest.setBatchSize(batchSize);
jobInstanceFetchRequest.setJobStatus(jobStatus != null ? jobStatus.toString() : "");
jobInstanceFetchRequest.setSort(Sort.by(Sort.Direction.DESC, JobController.MY_CREATE_TIME));

return theJobCoordinator.fetchAllJobInstances(jobInstanceFetchRequest).getContent();
}

@RequestMapping(value = JobController.JOBS, method = RequestMethod.DELETE, produces = MediaType.APPLICATION_JSON_VALUE)
public JobOperationResultJson cancelInstance(@RequestParam(name = "instanceId") String instanceId) {
return theJobCoordinator.cancelInstance(instanceId);
}

public static final String JOBS = "jobs";
public static final String MY_CREATE_TIME = "myCreateTime";
}

0 comments on commit 31e64f6

Please sign in to comment.