org.apache.lens.server.api.query.QueryContext.getPriority()方法的使用及代码示例

x33g5p2x  于2022-01-28 转载在 其他  
字(7.7k)|赞(0)|评价(0)|浏览(94)

本文整理了Java中org.apache.lens.server.api.query.QueryContext.getPriority方法的一些代码示例,展示了QueryContext.getPriority的具体用法。这些代码示例主要来源于Github/Stackoverflow/Maven等平台,是从一些精选项目中提取出来的代码,具有较强的参考意义,能在一定程度帮忙到你。QueryContext.getPriority方法的具体详情如下:
包路径:org.apache.lens.server.api.query.QueryContext
类名称:QueryContext
方法名:getPriority

QueryContext.getPriority介绍

暂无

代码示例

代码示例来源:origin: apache/lens

@Override
 public int compare(final QueryContext o1, final QueryContext o2) {
  return o1.getPriority().compareTo(o2.getPriority());
 }
}

代码示例来源:origin: org.apache.lens/lens-server-api

@Override
 public int compare(final QueryContext o1, final QueryContext o2) {
  return o1.getPriority().compareTo(o2.getPriority());
 }
}

代码示例来源:origin: apache/lens

private String canLaunchWithPriorityConstraint(QueryContext candidateQuery, Set<QueryContext> launchedQueries) {
 if (maxConcurrentQueriesPerPriority == null) {
  return null;
 }
 Priority priority = candidateQuery.getPriority();
 Integer limit = maxConcurrentQueriesPerPriority.get(priority);
 if (limit == null) {
  return null;
 }
 int launchedOnPriority = 0;
 for (QueryContext context : launchedQueries) {
  if (context.getPriority().equals(priority)) {
   launchedOnPriority++;
  }
 }
 if (launchedOnPriority >= limit) {
  return launchedOnPriority + "/" + limit + " queries running with priority " + priority;
 }
 return null;
}

代码示例来源:origin: org.apache.lens/lens-server-api

private String canLaunchWithPriorityConstraint(QueryContext candidateQuery, Set<QueryContext> launchedQueries) {
 if (maxConcurrentQueriesPerPriority == null) {
  return null;
 }
 Priority priority = candidateQuery.getPriority();
 Integer limit = maxConcurrentQueriesPerPriority.get(priority);
 if (limit == null) {
  return null;
 }
 int launchedOnPriority = 0;
 for (QueryContext context : launchedQueries) {
  if (context.getPriority().equals(priority)) {
   launchedOnPriority++;
  }
 }
 if (launchedOnPriority >= limit) {
  return launchedOnPriority + "/" + limit + " queries running with priority " + priority;
 }
 return null;
}

代码示例来源:origin: apache/lens

@Test
public void testCompareOnQueryPriority() {
 QueryContext query1 = mock(QueryContext.class);
 when(query1.getPriority()).thenReturn(Priority.HIGH); // Ordinal = 1
 QueryContext query2 = mock(QueryContext.class);
 when(query2.getPriority()).thenReturn(Priority.LOW); // Ordinal = 3
 assertEquals(priorityAndFifoComparator.compare(query1, query2), -2);
}

代码示例来源:origin: apache/lens

public static QueryContext createQueryInstanceWithQueryHandleAndPriorityStubbing(String handlePrefix, int index,
                                       Priority priority) {
 QueryContext mockQuery = mock(QueryContext.class);
 when(mockQuery.getQueryHandle()).thenReturn(QueryHandle.fromString(handlePrefix + index));
 when(mockQuery.getPriority()).thenReturn(priority);
 return mockQuery;
}

代码示例来源:origin: apache/lens

@Test(dataProvider = "dpTestPerPriorityConstraints")
public void testPerPriorityConstraints(final Priority[] launchedPriorities, final Priority candidatePriority,
 final boolean expectedCanLaunch) {
 EstimatedImmutableQueryCollection mockLaunchedQueries = mock(EstimatedImmutableQueryCollection.class);
 LensDriver mockDriver = mock(LensDriver.class);
 Set<QueryContext> launchedQueries = new HashSet<>();
 for (Priority priority : launchedPriorities) {
  QueryContext context = mock(QueryContext.class);
  when(context.getPriority()).thenReturn(priority);
  launchedQueries.add(context);
 }
 when(mockLaunchedQueries.getQueries(mockDriver)).thenReturn(launchedQueries);
 when(mockLaunchedQueries.getQueriesCount(mockDriver)).thenReturn(launchedQueries.size());
 QueryContext mockCandidateQuery = mock(QueryContext.class);
 when(mockCandidateQuery.getPriority()).thenReturn(candidatePriority);
 when(mockCandidateQuery.getSelectedDriver()).thenReturn(mockDriver);
 String actualCanLaunch = perPriorityConstraint.allowsLaunchOf(mockCandidateQuery, mockLaunchedQueries);
 if (expectedCanLaunch) {
  assertNull(actualCanLaunch);
 } else {
  assertNotNull(actualCanLaunch);
 }
}

代码示例来源:origin: apache/lens

@Test(dataProvider = "dpSubmitTimeCompare")
 public void testCompareOnQuerySubmitTime(final long submitTimeQuery1, final long submitTimeQuery2,
   final int expectedResult) {

  QueryContext query1 = mock(QueryContext.class);
  when(query1.getPriority()).thenReturn(Priority.HIGH);

  QueryCost s1 = new StaticQueryCost(0.0);
  QueryCost s2 = new StaticQueryCost(0.0);

  when(query1.getSelectedDriverQueryCost()).thenReturn(s1);

  QueryContext query2 = mock(QueryContext.class);
  when(query2.getPriority()).thenReturn(Priority.HIGH);
  when(query2.getSelectedDriverQueryCost()).thenReturn(s2);

  when(query1.getSubmissionTime()).thenReturn(submitTimeQuery1);
  when(query2.getSubmissionTime()).thenReturn(submitTimeQuery2);

  // Cost and Priority both are same, hence the comparison should happen
  // on query submission time
  assertEquals(priorityAndFifoComparator.compare(query1, query2), expectedResult);
  assertEquals(costAndFifoComparator.compare(query1, query2), expectedResult);

 }
}

代码示例来源:origin: apache/lens

@Test(dataProvider = "dpTestPerQueuePerPriorityConstraints")
 public void testPerQueuePerPriorityConstraints(final QueuePriority[] launchedQueuePriorities,
  final String candidateQueue, final Priority candidatePriority, final boolean expectedCanLaunch) {
  EstimatedImmutableQueryCollection mockLaunchedQueries = mock(EstimatedImmutableQueryCollection.class);
  LensDriver mockDriver = mock(LensDriver.class);
  Set<QueryContext> launchedQueries = new HashSet<>();
  for (QueuePriority queuePriority : launchedQueuePriorities) {
   QueryContext context = mock(QueryContext.class);
   when(context.getQueue()).thenReturn(queuePriority.getQueue());
   when(context.getPriority()).thenReturn(queuePriority.getPriority());
   launchedQueries.add(context);
  }
  when(mockLaunchedQueries.getQueries(mockDriver)).thenReturn(launchedQueries);
  when(mockLaunchedQueries.getQueriesCount(mockDriver)).thenReturn(launchedQueries.size());

  QueryContext mockCandidateQuery = mock(QueryContext.class);
  when(mockCandidateQuery.getQueue()).thenReturn(candidateQueue);
  when(mockCandidateQuery.getPriority()).thenReturn(candidatePriority);
  when(mockCandidateQuery.getSelectedDriver()).thenReturn(mockDriver);
  String actualCanLaunch = perQueueAndPerPriorityConstraint.allowsLaunchOf(mockCandidateQuery, mockLaunchedQueries);

  if (expectedCanLaunch) {
   assertNull(actualCanLaunch);
  } else {
   assertNotNull(actualCanLaunch);
  }
 }
}

代码示例来源:origin: apache/lens

@Test
public void testPriorityOnMockQuery() throws Exception {
 String query = "select mock, fail from " + TEST_TABLE;
 QueryContext ctx = queryService.createContext(query, null, new LensConf(), new Configuration(), 5000L);
 ctx.setLensSessionIdentifier(lensSessionId.getPublicId().toString());
 queryService.acquire(lensSessionId);
 try {
  queryService.rewriteAndSelect(ctx);
 } finally {
  queryService.release(lensSessionId);
 }
 assertNotNull(ctx.getSelectedDriver());
 assertEquals(ctx.getPriority(), Priority.NORMAL);
}

代码示例来源:origin: org.apache.lens/lens-server-api

/**
 * To lens query.
 *
 * @return the lens query
 */
public LensQuery toLensQuery() {
 return new LensQuery(queryHandle, userQuery, super.getSubmittedUser(), getPriority(), isPersistent,
  getSelectedDriver() != null ? getSelectedDriver().getFullyQualifiedName() : null,
  getSelectedDriverQuery(),
  status,
  resultSetPath, driverOpHandle, lensConf, submissionTime, launchTime, driverStatus.getDriverStartTime(),
  driverStatus.getDriverFinishTime(), endTime, closedTime, queryName, getFailedAttempts());
}

代码示例来源:origin: apache/lens

/**
 * To lens query.
 *
 * @return the lens query
 */
public LensQuery toLensQuery() {
 return new LensQuery(queryHandle, userQuery, super.getSubmittedUser(), getPriority(), isPersistent,
  getSelectedDriver() != null ? getSelectedDriver().getFullyQualifiedName() : null,
  getSelectedDriverQuery(),
  status,
  resultSetPath, driverOpHandle, lensConf, submissionTime, launchTime, driverStatus.getDriverStartTime(),
  driverStatus.getDriverFinishTime(), endTime, closedTime, queryName, getFailedAttempts());
}

代码示例来源:origin: org.apache.lens/lens-server-api

if (null != ctx.getPriority()) {
 this.priority = ctx.getPriority().toString();

代码示例来源:origin: apache/lens

if (null != ctx.getPriority()) {
 this.priority = ctx.getPriority().toString();

相关文章

微信公众号

最新文章

更多