in src/main/java/com/uber/cadence/internal/testservice/TestWorkflowStoreImpl.java [173:236]
public long save(RequestContext ctx)
throws InternalServiceError, EntityNotExistsError, BadRequestError {
long result;
lock.lock();
boolean historiesEmpty = histories.isEmpty();
try {
ExecutionId executionId = ctx.getExecutionId();
HistoryStore history = histories.get(executionId);
List<HistoryEvent> events = ctx.getEvents();
if (history == null) {
if (events.isEmpty()
|| events.get(0).getEventType() != EventType.WorkflowExecutionStarted) {
throw new IllegalStateException("No history found for " + executionId);
}
history = new HistoryStore(executionId, lock);
histories.put(executionId, history);
}
history.checkNextEventId(ctx.getInitialEventId());
history.addAllLocked(events, ctx.currentTimeInNanoseconds());
result = history.getNextEventIdLocked();
timerService.updateLocks(ctx.getTimerLocks(), "TestWorkflowStoreImpl save");
ctx.fireCallbacks(history.getEventsLocked().size());
} finally {
if (historiesEmpty && !histories.isEmpty()) {
timerService.unlockTimeSkipping(
"TestWorkflowStoreImpl save"); // Initially locked in the constructor
}
lock.unlock();
}
// Push tasks to the queues out of locks
DecisionTask decisionTask = ctx.getDecisionTask();
if (decisionTask != null) {
StickyExecutionAttributes attributes =
ctx.getWorkflowMutableState().getStickyExecutionAttributes();
TaskListId id =
new TaskListId(
decisionTask.getTaskListId().getDomain(),
attributes == null
? decisionTask.getTaskListId().getTaskListName()
: attributes.getWorkerTaskList().getName());
BlockingQueue<PollForDecisionTaskResponse> decisionsQueue = getDecisionTaskListQueue(id);
decisionsQueue.add(decisionTask.getTask());
}
List<ActivityTask> activityTasks = ctx.getActivityTasks();
if (activityTasks != null) {
for (ActivityTask activityTask : activityTasks) {
BlockingQueue<PollForActivityTaskResponse> activitiesQueue =
getActivityTaskListQueue(activityTask.getTaskListId());
activitiesQueue.add(activityTask.getTask());
}
}
List<Timer> timers = ctx.getTimers();
if (timers != null) {
for (Timer t : timers) {
timerService.schedule(
Duration.ofSeconds(t.getDelaySeconds()), t.getCallback(), t.getTaskInfo());
}
}
return result;
}