@Override
public Collection<UpdateWorkerTask> getUpdatingUpdateTasks(
final long guestId, final Connector connector) {
List<UpdateWorkerTask> tasks =
JPAUtils.find(
em,
UpdateWorkerTask.class,
"updateWorkerTasks.isInProgressOrScheduledBefore",
System.currentTimeMillis(),
guestId,
connector.getName());
HashMap<Integer, UpdateWorkerTask> seen = new HashMap<Integer, UpdateWorkerTask>();
for (UpdateWorkerTask task : tasks) {
if (hasStalled(task)) {
task.status = Status.STALLED;
em.merge(task);
} else {
if (seen.containsKey(task.objectTypes)) {
if (seen.get(task.objectTypes).timeScheduled < task.timeScheduled)
seen.put(task.objectTypes, task);
} else {
seen.put(task.objectTypes, task);
}
}
}
return seen.values();
}
@Transactional(readOnly = false)
@Override
public ScheduleResult reScheduleUpdateTask(
UpdateWorkerTask updt,
long time,
boolean incrementRetries,
UpdateWorkerTask.AuditTrailEntry auditTrailEntry) {
if (isShuttingDown) {
StringBuilder sb =
new StringBuilder(
"module=updateQueue component=updateAllConnectors" + " action=updateConnector")
.append(" message=\"Service is shutting down... Refusing updates\"");
logger.warn(sb.toString());
return new ScheduleResult(
updt.connectorName,
updt.getObjectTypes(),
ScheduleResult.ResultType.SYSTEM_IS_SHUTTING_DOWN,
time);
}
if (!incrementRetries) {
UpdateWorkerTask failed = new UpdateWorkerTask(updt);
failed.retries = updt.retries;
failed.connectorName = updt.connectorName;
failed.status = Status.FAILED;
failed.guestId = updt.guestId;
failed.timeScheduled = updt.timeScheduled;
em.persist(failed);
updt.retries = 0;
} else updt.retries += 1;
updt.addAuditTrailEntry(auditTrailEntry);
updt.status = Status.SCHEDULED;
updt.timeScheduled = time;
em.merge(updt);
return new ScheduleResult(
updt.connectorName,
updt.getObjectTypes(),
ScheduleResult.ResultType.SCHEDULED_UPDATE_DEFERRED,
time);
}
@Override
@Transactional(readOnly = false)
public void setUpdateWorkerTaskStatus(long updateWorkerTaskId, Status status)
throws RuntimeException {
UpdateWorkerTask updt = em.find(UpdateWorkerTask.class, updateWorkerTaskId);
if (updt == null) {
RuntimeException exception =
new RuntimeException(
"null UpdateWorkerTask trying to set its status: " + updateWorkerTaskId);
logger.error(
"module=updateQueue component=connectorUpdateService action=setUpdateWorkerTaskStatus");
throw exception;
}
updt.status = status;
}
@Override
@Transactional(readOnly = false)
public List<UpdateWorkerTask> getScheduledOrInProgressUpdateTasks(
long guestId, Connector connector) {
List<UpdateWorkerTask> updateWorkerTask =
JPAUtils.find(
em,
UpdateWorkerTask.class,
"updateWorkerTasks.isScheduledOrInProgress",
guestId,
connector.getName());
for (UpdateWorkerTask workerTask : updateWorkerTask) {
if (hasStalled(workerTask)) {
workerTask.status = Status.STALLED;
em.merge(workerTask);
}
}
return updateWorkerTask;
}
@Override
@Transactional(readOnly = false)
public UpdateWorkerTask getScheduledUpdateTask(
long guestId, String connectorName, int objectTypes) {
UpdateWorkerTask updateWorkerTask =
JPAUtils.findUnique(
em,
UpdateWorkerTask.class,
"updateWorkerTasks.withObjectTypes.isScheduled",
Status.SCHEDULED,
Status.IN_PROGRESS,
guestId,
objectTypes,
connectorName);
if (updateWorkerTask != null && hasStalled(updateWorkerTask)) {
updateWorkerTask.status = Status.STALLED;
em.merge(updateWorkerTask);
return null;
}
return updateWorkerTask;
}
@Override
@Transactional(readOnly = false)
public ScheduleResult scheduleUpdate(
long guestId,
String connectorName,
int objectTypes,
UpdateType updateType,
long timeScheduled,
String... jsonParams) {
if (isShuttingDown) {
StringBuilder sb =
new StringBuilder(
"module=updateQueue component=updateAllConnectors" + " action=scheduleUpdate")
.append(" message=\"Service is shutting down... Refusing updates\"");
logger.warn(sb.toString());
return new ScheduleResult(
connectorName,
objectTypes,
ScheduleResult.ResultType.SYSTEM_IS_SHUTTING_DOWN,
System.currentTimeMillis());
}
UpdateWorkerTask updateScheduled = getScheduledUpdateTask(guestId, connectorName, objectTypes);
ScheduleResult scheduleResult = null;
if (updateScheduled == null) {
UpdateWorkerTask updateWorkerTask = new UpdateWorkerTask();
updateWorkerTask.guestId = guestId;
updateWorkerTask.connectorName = connectorName;
updateWorkerTask.objectTypes = objectTypes;
updateWorkerTask.updateType = updateType;
updateWorkerTask.status = Status.SCHEDULED;
updateWorkerTask.timeScheduled = timeScheduled;
if (jsonParams != null && jsonParams.length > 0) updateWorkerTask.jsonParams = jsonParams[0];
em.persist(updateWorkerTask);
long now = System.currentTimeMillis();
scheduleResult =
new ScheduleResult(
connectorName,
objectTypes,
timeScheduled <= now
? ScheduleResult.ResultType.SCHEDULED_UPDATE_IMMEDIATE
: ScheduleResult.ResultType.SCHEDULED_UPDATE_DEFERRED,
timeScheduled);
} else {
scheduleResult =
new ScheduleResult(
connectorName,
objectTypes,
ScheduleResult.ResultType.ALREADY_SCHEDULED,
updateScheduled.timeScheduled);
}
StringBuilder sb =
new StringBuilder(
"module=updateQueue component=connectorUpdateService action=scheduleUpdate")
.append(" guestId=")
.append(guestId)
.append(" connectorName=")
.append(connectorName)
.append(" objectTypes=")
.append(objectTypes)
.append(" resultType=")
.append(scheduleResult.type.toString());
logger.info(sb.toString());
return scheduleResult;
}