Package org.apache.aurora.gen

Examples of org.apache.aurora.gen.TaskEvent


              storeProvider.getUnsafeTaskStore().mutateTasks(query, new TaskMutation() {
                @Override
                public IScheduledTask apply(IScheduledTask task) {
                  ScheduledTask mutableTask = task.newBuilder();
                  mutableTask.setStatus(targetState.get());
                  mutableTask.addToTaskEvents(new TaskEvent()
                      .setTimestamp(clock.nowMillis())
                      .setStatus(targetState.get())
                      .setMessage(transitionMessage.orNull())
                      .setScheduler(LOCAL_HOST_SUPPLIER.get()));
                  return IScheduledTask.build(mutableTask);
View Full Code Here


        String newestTask = Iterables.getLast(Sets.newTreeSet(activeTasksInShard));
        if (Tasks.id(task).equals(newestTask)) {
          LOG.info("Retaining task " + Tasks.id(task));
        } else {
          task.setStatus(ScheduleStatus.KILLED);
          task.addToTaskEvents(new TaskEvent(clock.nowMillis(), ScheduleStatus.KILLED)
              .setMessage("Killed duplicate shard."));
          // TODO(wfarner); Circle back if this is necessary.  Currently there's a race
          // condition between the time the scheduler is actually available without hitting
          // IllegalStateException (see DriverImpl).
          // driver.killTask(Tasks.id(task));
View Full Code Here

          taskStore.mutateTasks(query, new TaskMutation() {
            @Override
            public IScheduledTask apply(IScheduledTask task) {
              ScheduledTask mutableTask = task.newBuilder();
              mutableTask.setStatus(targetState.get());
              mutableTask.addToTaskEvents(new TaskEvent()
                  .setTimestamp(clock.nowMillis())
                  .setStatus(targetState.get())
                  .setMessage(transitionMessage.orNull())
                  .setScheduler(LOCAL_HOST_SUPPLIER.get()));
              return IScheduledTask.build(mutableTask);
View Full Code Here

  }

  private static ScheduledTask makeTask(String id, ScheduleStatus status) {
    ScheduledTask scheduledTask = new ScheduledTask()
        .setStatus(status)
        .setTaskEvents(ImmutableList.of(new TaskEvent(100, status)))
        .setAssignedTask(new AssignedTask()
            .setTaskId(id)
            .setTask(new TaskConfig()
                .setJob(new JobKey("role-" + id, "test", "job-" + id))
                .setJobName("job-" + id)
View Full Code Here

  }

  static List<ITaskEvent> makeEvents(Map<Long, ScheduleStatus> events) {
    ImmutableList.Builder<ITaskEvent> taskEvents = ImmutableList.builder();
    for (Map.Entry<Long, ScheduleStatus> entry : events.entrySet()) {
      taskEvents.add(ITaskEvent.build(new TaskEvent(entry.getKey(), entry.getValue())));
    }

    return taskEvents.build();
  }
View Full Code Here

  }

  @Test
  public void testRewriteSandboxDeleted() throws Exception {
    final TaskConfig storedTask = defaultTask();
    final TaskEvent expectedEvent = new TaskEvent(100, FINISHED);

    storage.write(new Storage.MutateWork.NoResult.Quiet() {
      @Override
      protected void execute(Storage.MutableStoreProvider storeProvider) {
        storeProvider.getUnsafeTaskStore().saveTasks(ImmutableSet.of(
            IScheduledTask.build(new ScheduledTask()
                .setStatus(SANDBOX_DELETED)
                .setTaskEvents(ImmutableList.of(expectedEvent, new TaskEvent(200, SANDBOX_DELETED)))
                .setAssignedTask(new AssignedTask()
                    .setTaskId(TASK_ID)
                    .setInstanceId(0)
                    .setTask(storedTask)))));
      }
View Full Code Here

      @Override
      protected void execute(Storage.MutableStoreProvider storeProvider) {
        storeProvider.getUnsafeTaskStore().saveTasks(ImmutableSet.of(
            IScheduledTask.build(new ScheduledTask()
                .setStatus(PENDING)
                .setTaskEvents(ImmutableList.of(new TaskEvent(100, PENDING)))
                .setAssignedTask(new AssignedTask()
                    .setTaskId(TASK_ID)
                    .setInstanceId(0)
                    .setTask(storedTask)))));
      }
View Full Code Here

    private Result changeStatusAndEvaluate(ScheduleStatus status) {
      ScheduledTask builder = task.get().newBuilder();
      if (builder.getStatus() != status) {
        // Only add a task event if this is a state change.
        builder.addToTaskEvents(new TaskEvent().setTimestamp(clock.nowMillis()).setStatus(status));
      }
      builder.setStatus(status);

      task = Optional.of(IScheduledTask.build(builder));
      return updater.evaluate(task);
View Full Code Here

  }

  IScheduledTask makeTask(ITaskConfig config, ScheduleStatus status) {
    List<TaskEvent> events = Lists.newArrayList();
    if (status != PENDING) {
      events.add(new TaskEvent().setTimestamp(clock.nowMillis()).setStatus(PENDING));
    }
    if (Tasks.isTerminated(status) || status == KILLING) {
      events.add(new TaskEvent().setTimestamp(clock.nowMillis()).setStatus(ASSIGNED));
      events.add(new TaskEvent().setTimestamp(clock.nowMillis()).setStatus(RUNNING));
    }

    events.add(new TaskEvent().setTimestamp(clock.nowMillis()).setStatus(status));

    return IScheduledTask.build(
        new ScheduledTask()
            .setStatus(status)
            .setTaskEvents(ImmutableList.copyOf(events))
View Full Code Here

  private ScheduledTask makeTask(String role, String job, String taskId, int priority) {
    return makeTask(role, job, taskId, priority, "dev", false);
  }

  private void addEvent(ScheduledTask task, ScheduleStatus status) {
    task.addToTaskEvents(new TaskEvent(clock.nowMillis(), status));
  }
View Full Code Here

TOP

Related Classes of org.apache.aurora.gen.TaskEvent

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.