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

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

105
106
107
108
109
110
111
112
113
114
115
  @Override
  public void updateJob(final SanitizedCronJob config) throws CronException {
    requireNonNull(config);
    checkNoRunOverlap(config);

    final IJobKey jobKey = config.getSanitizedConfig().getJobConfig().getKey();
    storage.write(new MutateWork.NoResult<CronException>() {
      @Override
      public void execute(Storage.MutableStoreProvider storeProvider) throws CronException {
        checkCronExists(jobKey, storeProvider.getJobStore());
View Full Code Here
124
125
126
127
128
129
130
131
132
133
134
  @Override
  public void createJob(final SanitizedCronJob cronJob) throws CronException {
    requireNonNull(cronJob);
    checkNoRunOverlap(cronJob);

    final IJobKey jobKey = cronJob.getSanitizedConfig().getJobConfig().getKey();
    storage.write(new MutateWork.NoResult<CronException>() {
      @Override
      protected void execute(Storage.MutableStoreProvider storeProvider) throws CronException {
        checkNotExists(jobKey, storeProvider.getJobStore());
View Full Code Here
336
337
338
339
340
341
342
343
344
345
346
            return null;
          }
        }
    );

    IJobKey jobKey = Iterables.getOnlyElement(
        FluentIterable.from(tasksInJob).transform(Tasks.SCHEDULED_TO_JOB_KEY).toSet());
    storageUtil.expectTaskFetch(HistoryPruner.jobHistoryQuery(jobKey), tasksInJob);
    if (pruned.length > 0) {
      stateManager.deleteTasks(Tasks.ids(pruned));
    }
View Full Code Here
331
332
333
334
335
336
337
338
339
340
341
342
  }

  @Test
  public void testServiceTasksRescheduled() throws Exception {
    int numServiceTasks = 5;
    IJobKey adhocKey = KEY_A;
    IJobKey serviceKey = IJobKey.build(
        adhocKey.newBuilder().setName(adhocKey.getName() + "service"));

    expectTaskNotThrottled().times(numServiceTasks);
    expectNoCronJob(adhocKey);
    expectNoCronJob(serviceKey);
View Full Code Here
121
122
123
124
125
126
127
128
129
130
131
    lockManager.validateIfLocked(LOCK_KEY, Optional.<ILock>absent());
  }

  @Test
  public void testValidateLockNotStoredHeld() throws Exception {
    IJobKey jobKey = JobKeys.from("r", "e", "n");
    expectLockException(jobKey);
    ILock lock = lockManager.acquireLock(LOCK_KEY, USER);
    ILockKey key = ILockKey.build(LockKey.job(jobKey.newBuilder()));
    lockManager.validateIfLocked(key, Optional.of(lock));
  }
View Full Code Here
48
49
50
51
52
53
54
55
  @Override
  public void saveAcceptedJob(String managerId, IJobConfiguration jobConfig) {
    requireNonNull(managerId);
    requireNonNull(jobConfig);

    IJobKey key = JobKeys.assertValid(jobConfig.getKey());
    managers.getUnchecked(managerId).jobs.put(key, jobConfig);
  }
View Full Code Here
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
    }.run();
  }

  @Test
  public void testRemoveJob() throws Exception {
    final IJobKey jobKey = JobKeys.from("role", "env", "name");
    new MutationFixture() {
      @Override
      protected void setupExpectations() throws Exception {
        storageUtil.expectWriteOperation();
        storageUtil.jobStore.removeJob(jobKey);
        streamMatcher.expectTransaction(
            Op.removeJob(new RemoveJob().setJobKey(jobKey.newBuilder())))
            .andReturn(position);
      }

      @Override
      protected void performMutations(MutableStoreProvider storeProvider) {
View Full Code Here
1429
1430
1431
1432
1433
1434
1435
1436
1437
1438
1439
1440
1441
1442
1443
1444
1445
1446
1447
1448
1449
1450
1451
1452
1453
1454
1455
1456
1457
1458
1459
1460
1461
1462
1463
1464
1465
1466
1467
    assertResponse(INVALID_REQUEST, thrift.getPendingReason(query.get()));
  }

  @Test
  public void testGetConfigSummary() throws Exception {
    IJobKey key = JobKeys.from("test", "test", "test");

    TaskConfig firstGroupTask = defaultTask(true);
    TaskConfig secondGroupTask = defaultTask(true).setNumCpus(2);

    IScheduledTask first1 = IScheduledTask.build(new ScheduledTask()
        .setAssignedTask(new AssignedTask().setTask(firstGroupTask).setInstanceId(0)));

    IScheduledTask first2 = IScheduledTask.build(new ScheduledTask()
        .setAssignedTask(new AssignedTask().setTask(firstGroupTask).setInstanceId(1)));

    IScheduledTask second = IScheduledTask.build(new ScheduledTask()
        .setAssignedTask(new AssignedTask().setTask(secondGroupTask).setInstanceId(2)));

    storageUtil.expectTaskFetch(Query.jobScoped(key).active(), first1, first2, second);

    ConfigGroup group1 = new ConfigGroup()
        .setConfig(firstGroupTask)
        .setInstanceIds(Sets.newHashSet(0, 1));
    ConfigGroup group2 = new ConfigGroup()
        .setConfig(secondGroupTask)
        .setInstanceIds(Sets.newHashSet(2));

    ConfigSummary summary = new ConfigSummary()
        .setKey(key.newBuilder())
        .setGroups(Sets.newHashSet(group1, group2));

    ConfigSummaryResult expected = new ConfigSummaryResult().setSummary(summary);

    control.replay();

    Response response = assertOkResponse(thrift.getConfigSummary(key.newBuilder()));
    assertEquals(expected, response.getResult().getConfigSummaryResult());
  }
View Full Code Here
1466
1467
1468
1469
1470
1471
1472
1473
1474
1475
1476
1477
1478
1479
1480
1481
1482
1483
1484
1485
    assertEquals(expected, response.getResult().getConfigSummaryResult());
  }

  @Test
  public void testEmptyConfigSummary() throws Exception {
    IJobKey key = JobKeys.from("test", "test", "test");

    storageUtil.expectTaskFetch(Query.jobScoped(key).active(), ImmutableSet.<IScheduledTask>of());

    ConfigSummary summary = new ConfigSummary()
        .setKey(key.newBuilder())
        .setGroups(Sets.<ConfigGroup>newHashSet());

    ConfigSummaryResult expected = new ConfigSummaryResult().setSummary(summary);

    control.replay();

    Response response = assertOkResponse(thrift.getConfigSummary(key.newBuilder()));
    assertEquals(expected, response.getResult().getConfigSummaryResult());
  }
View Full Code Here
104
105
106
107
108
109
110
111
112
113
114
  public Response dump(
      @PathParam("role") final String role,
      @PathParam("environment") final String environment,
      @PathParam("job") final String job) {

    final IJobKey jobKey = JobKeys.from(role, environment, job);
    return dumpEntity("Cron job " + JobKeys.canonicalString(jobKey),
        new Work.Quiet<Optional<? extends TBase<?, ?>>>() {
          @Override
          public Optional<JobConfiguration> apply(StoreProvider storeProvider) {
            return storeProvider.getJobStore().fetchJob(cronJobManager.getManagerKey(), jobKey)
View Full Code Here
TOP

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

  • org.apache.aurora.scheduler.async.HistoryPrunerTest
  • org.apache.aurora.scheduler.async.TaskHistoryPrunerTest
  • org.apache.aurora.scheduler.base.JobKeys
  • org.apache.aurora.scheduler.cron.quartz.CronJobManagerImpl
  • org.apache.aurora.scheduler.http.StructDump
  • org.apache.aurora.scheduler.state.BaseSchedulerCoreImplTest
  • org.apache.aurora.scheduler.state.LockManagerImplTest
  • org.apache.aurora.scheduler.storage.db.DBJobUpdateStoreTest
  • org.apache.aurora.scheduler.storage.log.LogStorageTest
  • org.apache.aurora.scheduler.storage.mem.MemJobStore
    • 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.