Package ca.uhn.fhir.jpa.batch2

Class JpaJobPersistenceImpl

java.lang.Object

ca.uhn.fhir.jpa.batch2.JpaJobPersistenceImpl

All Implemented Interfaces:

IJobPersistence, IWorkChunkPersistence

public class JpaJobPersistenceImpl
extends Object
implements IJobPersistence

Nested Class Summary

Nested classes/interfaces inherited from interface ca.uhn.fhir.batch2.api.IJobPersistence

IJobPersistence.CreateResult, IJobPersistence.JobInstanceUpdateCallback

Field Summary

Fields

Modifier and Type	Field	Description
static final String	CREATE_TIME

Constructor Summary

Constructors

Constructor	Description
JpaJobPersistenceImpl(IBatch2JobInstanceRepository theJobInstanceRepository, IBatch2WorkChunkRepository theWorkChunkRepository, IBatch2WorkChunkMetadataViewRepository theWorkChunkMetadataViewRepo, IHapiTransactionService theTransactionService, jakarta.persistence.EntityManager theEntityManager, ca.uhn.fhir.interceptor.api.IInterceptorBroadcaster theInterceptorBroadcaster)	Constructor

Method Summary

Modifier and Type	Method	Description
boolean	advanceJobStepAndUpdateChunkStatus(String theJobInstanceId, String theNextStepId, boolean theIsReductionStep)
JobOperationResultJson	cancelInstance(String theInstanceId)
WorkChunk	createWorkChunk(WorkChunk theWorkChunk)
void	deleteChunksAndMarkInstanceAsChunksPurged(String theInstanceId)
void	deleteInstanceAndChunks(String theInstanceId)
void	enqueueWorkChunkForProcessing(String theChunkId, Consumer<Integer> theCallback)
org.springframework.data.domain.Page<WorkChunkMetadata>	fetchAllWorkChunkMetadataForJobInStates(org.springframework.data.domain.Pageable thePageable, String theInstanceId, Set<WorkChunkStatusEnum> theStates)
Stream<WorkChunk>	fetchAllWorkChunksForStepStream(String theInstanceId, String theStepId)
Iterator<WorkChunk>	fetchAllWorkChunksIterator(String theInstanceId, boolean theWithData)	Note: Not @Transactional because the transaction happens in a lambda that's called outside of this method's scope
BatchInstanceStatusDTO	fetchBatchInstanceStatus(String theInstanceId)
Optional<JobInstance>	fetchInstance(String theInstanceId)
List<JobInstance>	fetchInstances(int thePageSize, int thePageIndex)
List<JobInstance>	fetchInstances(FetchJobInstancesRequest theRequest, int thePage, int theBatchSize)
List<JobInstance>	fetchInstances(String theJobDefinitionId, Set<StatusEnum> theStatuses, Date theCutoff, org.springframework.data.domain.Pageable thePageable)
List<JobInstance>	fetchInstancesByJobDefinitionId(String theJobDefinitionId, int thePageSize, int thePageIndex)
List<JobInstance>	fetchInstancesByJobDefinitionIdAndStatus(String theJobDefinitionId, Set<StatusEnum> theRequestedStatuses, int thePageSize, int thePageIndex)
org.springframework.data.domain.Page<JobInstance>	fetchJobInstances(JobInstanceFetchRequest theRequest)
List<JobInstance>	fetchRecentInstances(int thePageSize, int thePageIndex)
List<BatchWorkChunkStatusDTO>	fetchWorkChunkStatusForInstance(String theInstanceId)
Set<WorkChunkStatusEnum>	getDistinctWorkChunkStatesForJobAndStep(String theInstanceId, String theCurrentStepId)
boolean	markInstanceAsStatusWhenStatusIn(String theInstanceId, StatusEnum theStatusEnum, Set<StatusEnum> thePriorStates)
void	markWorkChunksWithStatusAndWipeData(String theInstanceId, List<String> theChunkIds, WorkChunkStatusEnum theStatus, String theErrorMessage)
void	onWorkChunkCompletion(WorkChunkCompletionEvent theEvent)
String	onWorkChunkCreate(WorkChunkCreateEvent theBatchWorkChunk)
Optional<WorkChunk>	onWorkChunkDequeue(String theChunkId)
WorkChunkStatusEnum	onWorkChunkError(WorkChunkErrorEvent theParameters)
void	onWorkChunkFailed(String theChunkId, String theErrorMessage)
void	onWorkChunkPollDelay(String theChunkId, Date theDeadline)
String	storeNewInstance(JobInstance theInstance)
boolean	updateInstance(String theInstanceId, IJobPersistence.JobInstanceUpdateCallback theModifier)
void	updateInstanceUpdateTime(String theInstanceId)
int	updatePollWaitingChunksForJobIfReady(String theInstanceId)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface ca.uhn.fhir.batch2.api.IJobPersistence

onChunkDequeued, onCreateWithFirstChunk

Field Details

CREATE_TIME

public static final String CREATE_TIME

See Also:

• Constant Field Values

Constructor Details

JpaJobPersistenceImpl

public JpaJobPersistenceImpl(IBatch2JobInstanceRepository theJobInstanceRepository,
 IBatch2WorkChunkRepository theWorkChunkRepository,
 IBatch2WorkChunkMetadataViewRepository theWorkChunkMetadataViewRepo,
 IHapiTransactionService theTransactionService,
 jakarta.persistence.EntityManager theEntityManager,
 ca.uhn.fhir.interceptor.api.IInterceptorBroadcaster theInterceptorBroadcaster)

Constructor

Method Details

onWorkChunkCreate

@Transactional(propagation=REQUIRED)
public String onWorkChunkCreate(WorkChunkCreateEvent theBatchWorkChunk)

Specified by:

onWorkChunkCreate in interface IWorkChunkPersistence

onWorkChunkDequeue

@Transactional(propagation=REQUIRED)
public Optional<WorkChunk> onWorkChunkDequeue(String theChunkId)

Specified by:

onWorkChunkDequeue in interface IWorkChunkPersistence

storeNewInstance

@Transactional(propagation=REQUIRED)
public String storeNewInstance(JobInstance theInstance)

Specified by:

storeNewInstance in interface IJobPersistence

fetchInstances

@Transactional(propagation=REQUIRES_NEW)
public List<JobInstance> fetchInstances(String theJobDefinitionId,
 Set<StatusEnum> theStatuses,
 Date theCutoff,
 org.springframework.data.domain.Pageable thePageable)

Specified by:

fetchInstances in interface IJobPersistence

fetchInstancesByJobDefinitionIdAndStatus

@Transactional(propagation=REQUIRES_NEW)
public List<JobInstance> fetchInstancesByJobDefinitionIdAndStatus(String theJobDefinitionId,
 Set<StatusEnum> theRequestedStatuses,
 int thePageSize,
 int thePageIndex)

Specified by:

fetchInstancesByJobDefinitionIdAndStatus in interface IJobPersistence

fetchInstancesByJobDefinitionId

@Transactional(propagation=REQUIRES_NEW)
public List<JobInstance> fetchInstancesByJobDefinitionId(String theJobDefinitionId,
 int thePageSize,
 int thePageIndex)

Specified by:

fetchInstancesByJobDefinitionId in interface IJobPersistence

fetchJobInstances

@Transactional(propagation=REQUIRES_NEW)
public org.springframework.data.domain.Page<JobInstance> fetchJobInstances(JobInstanceFetchRequest theRequest)

Specified by:

fetchJobInstances in interface IJobPersistence

fetchInstance

@Nonnull
public Optional<JobInstance> fetchInstance(String theInstanceId)

Specified by:

fetchInstance in interface IJobPersistence

fetchWorkChunkStatusForInstance

@Nonnull
public List<BatchWorkChunkStatusDTO> fetchWorkChunkStatusForInstance(String theInstanceId)

Specified by:

fetchWorkChunkStatusForInstance in interface IJobPersistence

fetchBatchInstanceStatus

@Nonnull
public BatchInstanceStatusDTO fetchBatchInstanceStatus(String theInstanceId)

Specified by:

fetchBatchInstanceStatus in interface IJobPersistence

fetchInstances

@Transactional(propagation=REQUIRES_NEW)
public List<JobInstance> fetchInstances(FetchJobInstancesRequest theRequest,
 int thePage,
 int theBatchSize)

Specified by:

fetchInstances in interface IJobPersistence

fetchInstances

@Transactional(propagation=REQUIRES_NEW)
public List<JobInstance> fetchInstances(int thePageSize,
 int thePageIndex)

Specified by:

fetchInstances in interface IJobPersistence

enqueueWorkChunkForProcessing

public void enqueueWorkChunkForProcessing(String theChunkId,
 Consumer<Integer> theCallback)

Specified by:

enqueueWorkChunkForProcessing in interface IJobPersistence

updatePollWaitingChunksForJobIfReady

public int updatePollWaitingChunksForJobIfReady(String theInstanceId)

Specified by:

updatePollWaitingChunksForJobIfReady in interface IJobPersistence

fetchRecentInstances

@Transactional(propagation=REQUIRES_NEW)
public List<JobInstance> fetchRecentInstances(int thePageSize,
 int thePageIndex)

Specified by:

fetchRecentInstances in interface IJobPersistence

onWorkChunkError

public WorkChunkStatusEnum onWorkChunkError(WorkChunkErrorEvent theParameters)

Specified by:

onWorkChunkError in interface IWorkChunkPersistence

onWorkChunkPollDelay

public void onWorkChunkPollDelay(String theChunkId,
 Date theDeadline)

Specified by:

onWorkChunkPollDelay in interface IWorkChunkPersistence

onWorkChunkFailed

public void onWorkChunkFailed(String theChunkId,
 String theErrorMessage)

Specified by:

onWorkChunkFailed in interface IWorkChunkPersistence

onWorkChunkCompletion

public void onWorkChunkCompletion(WorkChunkCompletionEvent theEvent)

Specified by:

onWorkChunkCompletion in interface IWorkChunkPersistence

markWorkChunksWithStatusAndWipeData

public void markWorkChunksWithStatusAndWipeData(String theInstanceId,
 List<String> theChunkIds,
 WorkChunkStatusEnum theStatus,
 String theErrorMessage)

Specified by:

markWorkChunksWithStatusAndWipeData in interface IWorkChunkPersistence

getDistinctWorkChunkStatesForJobAndStep

public Set<WorkChunkStatusEnum> getDistinctWorkChunkStatesForJobAndStep(String theInstanceId,
 String theCurrentStepId)

Specified by:

getDistinctWorkChunkStatesForJobAndStep in interface IJobPersistence

updateInstanceUpdateTime

public void updateInstanceUpdateTime(String theInstanceId)

Specified by:

updateInstanceUpdateTime in interface IJobPersistence

createWorkChunk

public WorkChunk createWorkChunk(WorkChunk theWorkChunk)

Specified by:

createWorkChunk in interface IJobPersistence

fetchAllWorkChunksIterator

public Iterator<WorkChunk> fetchAllWorkChunksIterator(String theInstanceId,
 boolean theWithData)

Note: Not @Transactional because the transaction happens in a lambda that's called outside of this method's scope

Specified by:

fetchAllWorkChunksIterator in interface IJobPersistence

Specified by:

fetchAllWorkChunksIterator in interface IWorkChunkPersistence

fetchAllWorkChunksForStepStream

public Stream<WorkChunk> fetchAllWorkChunksForStepStream(String theInstanceId,
 String theStepId)

Specified by:

fetchAllWorkChunksForStepStream in interface IJobPersistence

Specified by:

fetchAllWorkChunksForStepStream in interface IWorkChunkPersistence

fetchAllWorkChunkMetadataForJobInStates

public org.springframework.data.domain.Page<WorkChunkMetadata> fetchAllWorkChunkMetadataForJobInStates(org.springframework.data.domain.Pageable thePageable,
 String theInstanceId,
 Set<WorkChunkStatusEnum> theStates)

Specified by:

fetchAllWorkChunkMetadataForJobInStates in interface IJobPersistence

updateInstance

public boolean updateInstance(String theInstanceId,
 IJobPersistence.JobInstanceUpdateCallback theModifier)

Specified by:

updateInstance in interface IJobPersistence

deleteInstanceAndChunks

@Transactional(propagation=REQUIRES_NEW)
public void deleteInstanceAndChunks(String theInstanceId)

Specified by:

deleteInstanceAndChunks in interface IJobPersistence

deleteChunksAndMarkInstanceAsChunksPurged

@Transactional(propagation=REQUIRES_NEW)
public void deleteChunksAndMarkInstanceAsChunksPurged(String theInstanceId)

Specified by:

deleteChunksAndMarkInstanceAsChunksPurged in interface IJobPersistence

markInstanceAsStatusWhenStatusIn

public boolean markInstanceAsStatusWhenStatusIn(String theInstanceId,
 StatusEnum theStatusEnum,
 Set<StatusEnum> thePriorStates)

Specified by:

markInstanceAsStatusWhenStatusIn in interface IJobPersistence

cancelInstance

@Transactional(propagation=REQUIRES_NEW)
public JobOperationResultJson cancelInstance(String theInstanceId)

Specified by:

cancelInstance in interface IJobPersistence

advanceJobStepAndUpdateChunkStatus

@Transactional(propagation=REQUIRES_NEW)
public boolean advanceJobStepAndUpdateChunkStatus(String theJobInstanceId,
 String theNextStepId,
 boolean theIsReductionStep)

Specified by:

advanceJobStepAndUpdateChunkStatus in interface IJobPersistence