Package org.apache.aurora.scheduler.storage.entities

Examples of org.apache.aurora.scheduler.storage.entities.IScheduledTask


      String job,
      int instanceId,
      UriInfo uriInfo,
      Optional<String> forwardRequest) {

    IScheduledTask task = Iterables.getOnlyElement(
        Storage.Util.consistentFetchTasks(storage,
            Query.instanceScoped(JobKeys.from(role, env, job), instanceId).active()),
        null);
    if (task == null) {
      return respond(NOT_FOUND, "No such live instance found.");
    }

    if (task.getStatus() != RUNNING) {
      return respond(NOT_FOUND, "The selected instance is currently in state " + task.getStatus());
    }

    IAssignedTask assignedTask = task.getAssignedTask();
    Optional<Integer> port = getRedirectPort(assignedTask);
    if (!port.isPresent()) {
      return respond(NOT_FOUND, "The task does not have a registered http port.");
    }
View Full Code Here


      try {
        return storage.write(new MutateWork.Quiet<Boolean>() {
          @Override
          public Boolean apply(MutableStoreProvider store) {
            LOG.fine("Attempting to schedule task " + taskId);
            final IScheduledTask task = Iterables.getOnlyElement(
                store.getTaskStore().fetchTasks(Query.taskScoped(taskId).byStatus(PENDING)),
                null);
            if (task == null) {
              LOG.warning("Failed to look up task " + taskId + ", it may have been deleted.");
            } else {
View Full Code Here

              } else {
                newState = PENDING;
                auditMessage = "Rescheduled";
              }

              IScheduledTask newTask = IScheduledTask.build(createTask(
                  upToDateTask.get().getAssignedTask().getInstanceId(),
                  upToDateTask.get().getAssignedTask().getTask())
                  .newBuilder()
                  .setFailureCount(upToDateTask.get().getFailureCount())
                  .setAncestorId(taskId));
View Full Code Here

        quotaManager.getQuotaInfo(ROLE));
  }

  @Test
  public void testGetQuotaInfoWithCronTasks() {
    IScheduledTask prodTask = createProdTask("pc", 6, 6, 6);
    IScheduledTask nonProdTask = createProdTask("npc", 7, 7, 7);
    IResourceAggregate quota = IResourceAggregate.build(new ResourceAggregate(4, 4, 4));

    expectQuota(quota);
    expectTasks(prodTask, nonProdTask);
    expectJobUpdates(taskConfig(1, 1, 1, true), taskConfig(1, 1, 1, true));
View Full Code Here

        quotaManager.getQuotaInfo(ROLE));
  }

  @Test
  public void testGetQuotaInfoPartialUpdate() {
    IScheduledTask prodTask = createProdTask("foo", 3, 3, 3);
    IScheduledTask updatingProdTask = createTask(JOB_NAME, "id1", 3, 3, 3, true, 1);
    IScheduledTask updatingFilteredProdTask = createTask(JOB_NAME, "id0", 3, 3, 3, true, 0);
    IScheduledTask nonProdTask = createTask("bar", "id1", 2, 2, 2, false, 0);
    IResourceAggregate quota = IResourceAggregate.build(new ResourceAggregate(4, 4, 4));

    expectQuota(quota);
    expectTasks(prodTask, updatingProdTask, updatingFilteredProdTask, nonProdTask);
    expectJobUpdates(taskConfig(1, 1, 1, true), taskConfig(1, 1, 1, true));
View Full Code Here

          } else {
            newState = PENDING;
            auditMessage = "Rescheduled";
          }

          IScheduledTask newTask = IScheduledTask.build(createTask(
              upToDateTask.get().getAssignedTask().getInstanceId(),
              upToDateTask.get().getAssignedTask().getTask())
              .newBuilder()
              .setFailureCount(upToDateTask.get().getFailureCount())
              .setAncestorId(taskId));
View Full Code Here

   * @param stateChange State change notification.
   */
  @Subscribe
  public synchronized void taskChangedState(TaskStateChange stateChange) {
    if (stateChange.getNewState() == PENDING) {
      IScheduledTask task = stateChange.getTask();
      GroupKey key = new GroupKey(task.getAssignedTask().getTask());
      TaskGroup newGroup = new TaskGroup(key, Tasks.id(task));
      TaskGroup existing = groups.putIfAbsent(key, newGroup);
      if (existing == null) {
        long penaltyMs;
        if (stateChange.isTransition()) {
View Full Code Here

      String job,
      int instanceId,
      UriInfo uriInfo,
      Optional<String> forwardRequest) {

    IScheduledTask task = Iterables.getOnlyElement(
        Storage.Util.consistentFetchTasks(storage,
            Query.instanceScoped(JobKeys.from(role, env, job), instanceId).active()),
        null);
    if (task == null) {
      return respond(NOT_FOUND, "No such live instance found.");
    }

    if (task.getStatus() != RUNNING) {
      return respond(NOT_FOUND, "The selected instance is currently in state " + task.getStatus());
    }

    IAssignedTask assignedTask = task.getAssignedTask();
    Optional<Integer> port = getRedirectPort(assignedTask);
    if (!port.isPresent()) {
      return respond(NOT_FOUND, "The task does not have a registered http port.");
    }
View Full Code Here

      try {
        return storage.write(new MutateWork.Quiet<Boolean>() {
          @Override
          public Boolean apply(MutableStoreProvider store) {
            LOG.fine("Attempting to schedule task " + taskId);
            final IScheduledTask task = Iterables.getOnlyElement(
                store.getTaskStore().fetchTasks(Query.taskScoped(taskId).byStatus(PENDING)),
                null);
            if (task == null) {
              LOG.warning("Failed to look up task " + taskId + ", it may have been deleted.");
            } else {
View Full Code Here

    }
  }

  @Subscribe
  public void taskChangedState(TaskStateChange stateChange) {
    IScheduledTask task = stateChange.getTask();
    Optional<ScheduleStatus> previousState = stateChange.getOldState();

    if (stateChange.isTransition() && !previousState.equals(Optional.of(ScheduleStatus.INIT))) {
      decrementCount(previousState.get());
    }
    incrementCount(task.getStatus());

    updateRackCounters(task, task.getStatus());
    updateJobCounters(task, task.getStatus());
  }
View Full Code Here

TOP

Related Classes of org.apache.aurora.scheduler.storage.entities.IScheduledTask

Copyright © 2018 www.massapicom. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.