コード例 #1
0
 @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();
 }
コード例 #2
0
 @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);
 }
コード例 #3
0
 @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;
 }
コード例 #4
0
 @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;
 }
コード例 #5
0
 @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;
 }
コード例 #6
0
 @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;
 }