Package org.apache.aurora.gen

Examples of org.apache.aurora.gen.TaskQuery

168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
   * @param query Query to extract the keys from.
   * @return A present if keys can be extracted, absent otherwise.
   */
  public static Optional<Set<IJobKey>> from(Query.Builder query) {
    if (Query.isJobScoped(query)) {
      TaskQuery taskQuery = query.get();
      ImmutableSet.Builder<IJobKey> builder = ImmutableSet.builder();

      if (taskQuery.isSetJobName()) {
        builder.add(from(
            taskQuery.getOwner().getRole(),
            taskQuery.getEnvironment(),
            taskQuery.getJobName()));
      }

      if (taskQuery.isSetJobKeys()) {
        builder.addAll(IJobKey.setFromBuilders(taskQuery.getJobKeys()));
      }
      return Optional.of(assertValid(builder.build()));
    } else {
      return Optional.absent();
    }
View Full Code Here
145
146
147
148
149
150
151
   */
  public static final class Builder implements Supplier<TaskQuery> {
    private final TaskQuery query;

    Builder() {
      this.query = new TaskQuery();
    }
View Full Code Here
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
   * @param query Query to extract the keys from.
   * @return A present if keys can be extracted, absent otherwise.
   */
  public static Optional<Set<IJobKey>> from(Query.Builder query) {
    if (Query.isJobScoped(query)) {
      TaskQuery taskQuery = query.get();
      ImmutableSet.Builder<IJobKey> builder = ImmutableSet.builder();

      if (taskQuery.isSetJobName()) {
        builder.add(from(
            taskQuery.getRole(),
            taskQuery.getEnvironment(),
            taskQuery.getJobName()));
      }

      if (taskQuery.isSetJobKeys()) {
        builder.addAll(IJobKey.setFromBuilders(taskQuery.getJobKeys()));
      }
      return Optional.of(assertValid(builder.build()));
    } else {
      return Optional.absent();
    }
View Full Code Here
50
51
52
53
54
55
56
57
   *
   * @param taskQuery Query to test.
   * @return {@code true} if the query specifies at least one job key, otherwise {@code false}.
   */
  public static boolean isJobScoped(Builder taskQuery) {
    TaskQuery q = taskQuery.get();
    return q.isSetRole() && q.isSetEnvironment() && q.isSetJobName() || q.isSetJobKeys();
  }
View Full Code Here
827
828
829
830
831
832
833
834
835
836
837
    assertResponse(AUTH_FAILED, thrift.killTasks(query.get(), DEFAULT_LOCK, SESSION));
  }

  @Test
  public void testKillTasksInvalidJobName() throws Exception {
    TaskQuery query = new TaskQuery()
        .setOwner(ROLE_IDENTITY)
        .setJobName("");

    control.replay();
View Full Code Here
870
871
872
873
874
875
876
877
878
879
880
  @Test
  public void testKillCronAuthenticatesQueryJobKeyRole() throws Exception {
    expectAuth(ROOT, false);
    IJobKey key = JobKeys.from("role", "env", "job");

    Query.Builder query = Query.arbitrary(new TaskQuery().setJobKeys(
        ImmutableSet.of(key.newBuilder())));

    storageUtil.expectTaskFetch(query.active());
    expectAuth(ImmutableSet.of("role"), true);
    expect(cronJobManager.deleteJob(key)).andReturn(true);
View Full Code Here
1703
1704
1705
1706
1707
1708
1709
1710
1711
1712
        ImmutableSet.<JobConfiguration>builder().addAll(crons).add(immediateJob).build();
    assertEquals(allJobs, thrift.getJobs(null).getResult().getGetJobsResult().getConfigs());
  }

  private TaskQuery setupPaginatedQuery(Iterable<IScheduledTask> tasks, int offset, int limit) {
    TaskQuery query = new TaskQuery().setOffset(offset).setLimit(limit);
    Builder builder = Query.arbitrary(query);
    storageUtil.expectTaskFetch(builder, ImmutableSet.copyOf(tasks));
    return query;
  }
View Full Code Here
1722
1723
1724
1725
1726
1727
1728
1729
1730
1731
1732
1733
1734
  @Test
  public void testGetTasksStatusPagination() throws Exception {
    Iterable<IScheduledTask> tasks = makeDefaultScheduledTasks(10);

    TaskQuery page1Query = setupPaginatedQuery(tasks, 0, 4);
    TaskQuery page2Query = setupPaginatedQuery(tasks, 4, 4);
    TaskQuery page3Query = setupPaginatedQuery(tasks, 8, 4);

    control.replay();

    Response page1Response = assertOkResponse(thrift.getTasksStatus(page1Query));
    Response page2Response = assertOkResponse(thrift.getTasksStatus(page2Query));
View Full Code Here
1753
1754
1755
1756
1757
1758
1759
1760
    storageUtil.expectTaskFetch(query, ImmutableSet.copyOf(tasks));

    control.replay();

    ImmutableList<ScheduledTask> expected = IScheduledTask.toBuildersList(tasks);
    Response response = assertOkResponse(thrift.getTasksStatus(new TaskQuery()));
    assertEquals(expected, response.getResult().getScheduleStatusResult().getTasks());
  }
View Full Code Here
1768
1769
1770
1771
1772
1773
1774
1775
    ImmutableList<ScheduledTask> expected = IScheduledTask.toBuildersList(makeDefaultScheduledTasks(
        10,
        defaultTask(true).setExecutorConfig(null)));

    Response response = assertOkResponse(thrift.getTasksWithoutConfigs(new TaskQuery()));
    assertEquals(expected, response.getResult().getScheduleStatusResult().getTasks());
  }
View Full Code Here
TOP

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

  • org.apache.aurora.scheduler.base.JobKeys
  • org.apache.aurora.scheduler.base.Query
  • org.apache.aurora.scheduler.base.Query$Builder
  • org.apache.aurora.scheduler.http.api.ApiBetaTest
  • org.apache.aurora.scheduler.storage.mem.MemTaskStoreTest
  • org.apache.aurora.scheduler.thrift.aop.FeatureToggleInterceptorTest
  • org.apache.aurora.scheduler.thrift.SchedulerThriftInterfaceTest
    • 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.