Package org.apache.aurora.gen

Examples of org.apache.aurora.gen.TaskConfig

1637
1638
1639
1640
1641
1642
1643
1644
1645
1646
1647
  }

  @Test
  public void testAddInstancesInvalidConfig() throws Exception {
    AddInstancesConfig config = createInstanceConfig(defaultTask(true));
    TaskConfig taskConfig = config.getTaskConfig().setExecutorConfig(null);
    config.setTaskConfig(taskConfig);
    expectAuth(ROLE, true);

    control.replay();
View Full Code Here
1839
1840
1841
1842
1843
1844
1845
1846
1847
1848
1849
          anyObject(AuditCheck.class))).andReturn(context);
    }
  }

  private static TaskConfig defaultTask(boolean production) {
    return new TaskConfig()
        .setOwner(new Identity(ROLE, USER))
        .setEnvironment(DEFAULT_ENVIRONMENT)
        .setJobName(JOB_NAME)
        .setContactEmail("testing@twitter.com")
        .setExecutorConfig(new ExecutorConfig("aurora", "data"))
View Full Code Here
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
    clock = new FakeClock();
  }

  @Test
  public void testLoadTasksFromStorage() throws Exception {
    final TaskConfig storedTask = defaultTask();

    storage.write(new Storage.MutateWork.NoResult.Quiet() {
      @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)))));
      }
    });

    backfill();

    // Since task fields are backfilled with defaults, additional flags should be filled.
    ITaskConfig expected = ITaskConfig.build(new TaskConfig(storedTask)
        .setProduction(false)
        .setMaxTaskFailures(1)
        .setExecutorConfig(EXECUTOR_CONFIG)
        .setConstraints(ImmutableSet.of(ConfigurationManager.hostLimitConstraint(1))));
View Full Code Here
150
151
152
153
154
155
156
157
158
159
160
    JobConfiguration job = new JobConfiguration()
        .setOwner(OWNER)
        .setKey(jobKey.newBuilder())
        .setInstanceCount(numTasks)
        .setTaskConfig(new TaskConfig(task)
            .setOwner(OWNER)
            .setEnvironment(jobKey.getEnvironment())
            .setJobName(jobKey.getName()));
    return SanitizedConfiguration.fromUnsanitized(IJobConfiguration.build(job));
  }
View Full Code Here
158
159
160
161
162
163
164
165
166
167
168
            .setJobName(jobKey.getName()));
    return SanitizedConfiguration.fromUnsanitized(IJobConfiguration.build(job));
  }

  private static TaskConfig defaultTask() {
    return new TaskConfig()
        .setOwner(OWNER)
        .setJobName(JOB_NAME)
        .setEnvironment(ENV)
        .setNumCpus(1.0)
        .setRamMb(ONE_GB)
View Full Code Here
405
406
407
408
409
410
411
412
413
414
415
    return IScheduledTask.build(new ScheduledTask()
        .setStatus(ScheduleStatus.PENDING)
        .setAssignedTask(new AssignedTask()
            .setInstanceId(0)
            .setTaskId(id)
            .setTask(new TaskConfig()
                .setJobName(jobName)
                .setEnvironment(env)
                .setOwner(new Identity(role, role))
                .setExecutorConfig(new ExecutorConfig().setData("executor config")))));
  }
View Full Code Here
65
66
67
68
69
70
71
72
73
74
75
76
77
      TaskStore.Mutable taskStore,
      Clock clock) {

    if (Tasks.isActive(task.getStatus())) {
      // Perform a sanity check on the number of active shards.
      TaskConfig config = task.getAssignedTask().getTask();
      Query.Builder query = Query.instanceScoped(
          JobKeys.from(config.getOwner().getRole(), config.getEnvironment(), config.getJobName()),
          task.getAssignedTask().getInstanceId())
          .active();
      Set<String> activeTasksInShard = FluentIterable.from(taskStore.fetchTasks(query))
          .transform(Tasks.SCHEDULED_TO_ID)
          .toSet();
View Full Code Here
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
              IJobKey jobKey,
              Collection<IScheduledTask> tasks) {

            // Pick the latest transitioned task for each immediate job since the job can be in the
            // middle of an update or some shards have been selectively created.
            TaskConfig mostRecentTaskConfig =
                Tasks.getLatestActiveTask(tasks).getAssignedTask().getTask().newBuilder();

            return IJobConfiguration.build(new JobConfiguration()
                .setKey(jobKey.newBuilder())
                .setOwner(mostRecentTaskConfig.getOwner())
                .setTaskConfig(mostRecentTaskConfig)
                .setInstanceCount(tasks.size()));
          }
        }));
View Full Code Here
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
      long diskMb,
      boolean production,
      ScheduleStatus status,
      Optional<String> dedicated) {

    TaskConfig task = new TaskConfig()
        .setOwner(new Identity().setRole(role))
        .setEnvironment("test")
        .setJobName(job)
        .setNumCpus(numCpus)
        .setRamMb(ramMb)
        .setDiskMb(diskMb)
        .setProduction(production);

    if (dedicated.isPresent()) {
      task.addToConstraints(new Constraint(
          ConfigurationManager.DEDICATED_ATTRIBUTE,
          TaskConstraint.value(new ValueConstraint(false, ImmutableSet.of(dedicated.get())))));
    }

    return IScheduledTask.build(new ScheduledTask()
View Full Code Here
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
              IJobKey jobKey,
              Collection<IScheduledTask> tasks) {

            // Pick the latest transitioned task for each immediate job since the job can be in the
            // middle of an update or some shards have been selectively created.
            TaskConfig mostRecentTaskConfig =
                Tasks.getLatestActiveTask(tasks).getAssignedTask().getTask().newBuilder();

            return IJobConfiguration.build(new JobConfiguration()
                .setKey(jobKey.newBuilder())
                .setOwner(mostRecentTaskConfig.getOwner())
                .setTaskConfig(mostRecentTaskConfig)
                .setInstanceCount(tasks.size()));
          }
        }));
View Full Code Here
TOP

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

  • org.apache.aurora.scheduler.app.SchedulerIT
  • org.apache.aurora.scheduler.async.GcExecutorLauncherTest
  • org.apache.aurora.scheduler.async.HistoryPrunerTest
  • org.apache.aurora.scheduler.async.PreemptorImplTest
  • org.apache.aurora.scheduler.async.RescheduleCalculatorImplTest
  • org.apache.aurora.scheduler.async.TaskGroupsTest
  • org.apache.aurora.scheduler.async.TaskHistoryPrunerTest
  • org.apache.aurora.scheduler.async.TaskSchedulerImplTest
  • org.apache.aurora.scheduler.async.TaskSchedulerTest
  • org.apache.aurora.scheduler.async.TaskTimeoutTest
    • 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.