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

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

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

QueryContext.getSelectedDriverQueryCost介绍

暂无

代码示例

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

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

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

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

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

@VisibleForTesting
void checkState(final QueryContext query) {
 Preconditions.checkState(query.getSelectedDriver() != null);
 Preconditions.checkState(query.getSelectedDriverQueryCost() != null);
}

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

private void checkEstimatedQueriesState(final QueryContext query) throws LensException {
 if (query.getSelectedDriver() == null || query.getSelectedDriverQueryCost() == null) {
  throw new LensException("selected driver: " + query.getSelectedDriver() + " OR selected driver query cost: "
   + query.getSelectedDriverQueryCost() + " is null. Query doesn't appear to be an estimated query.");
 }
}

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

private QueryCost getTotalQueryCost(final Collection<QueryContext> queries) {
 if (queries.isEmpty()) {
  return new StaticQueryCost(0);
 }
 Double totalQueryCost = 0d;
 for (QueryContext query : queries) {
  Double queryCost = query.getSelectedDriverQueryCost().getEstimatedResourceUsage();
  totalQueryCost = totalQueryCost + queryCost;
 }
 log.debug("Total Query Cost:{}", totalQueryCost);
 return new StaticQueryCost(totalQueryCost);
}

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

@Test(dataProvider = "dpQueryCostCompare")
public void testCompareOnQueryCost(final int resultOfQueryCostCompare, final int expectedResult) {
 QueryContext query1 = mock(QueryContext.class);
 QueryCost qcO1 = mock(QueryCost.class);
 when(query1.getSelectedDriverQueryCost()).thenReturn(qcO1);
 QueryContext query2 = mock(QueryContext.class);
 QueryCost qcO2 = mock(QueryCost.class);
 when(query2.getSelectedDriverQueryCost()).thenReturn(qcO2);
 when(qcO1.compareTo(qcO2)).thenReturn(resultOfQueryCostCompare);
 assertEquals(costAndFifoComparator.compare(query1, query2), expectedResult);
}

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

@Test(expectedExceptions = IllegalStateException.class)
public void testCheckStateMustRecognizeIllegalStateWhenSelectedDriverIsNotSet() throws LensException {
 QueryContext mockQuery = mock(QueryContext.class);
 /* Setting selected driver cost, however since selected driver is not set. This should result in
 IllegalStateException */
 when(mockQuery.getSelectedDriverQueryCost()).thenReturn(mock(QueryCost.class));
 QueryCollection mockQueries = mock(QueryCollection.class);
 DefaultEstimatedQueryCollection queries = new DefaultEstimatedQueryCollection(mockQueries);
 queries.checkState(mockQuery);
}

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

public static QueryContext createQueryInstanceWithQueryHandleAndCostStubbing(String handlePrefix, int index,
                                       double queryCost) {
 QueryContext mockQuery = mock(QueryContext.class);
 when(mockQuery.getQueryHandle()).thenReturn(QueryHandle.fromString(handlePrefix + index));
 when(mockQuery.getSelectedDriverQueryCost()).thenReturn(new FactPartitionBasedQueryCost(queryCost));
 return mockQuery;
}

代码示例来源: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
public void testAddAndRemoveAndGetQueriesMethod() throws LensException {
 QueryContext mockQuery = mock(QueryContext.class);
 LensDriver mockSelectedDriver = mock(LensDriver.class);
 QueryCost mockQueryCost = mock(QueryCost.class);
 when(mockQuery.getSelectedDriver()).thenReturn(mockSelectedDriver);
 when(mockQuery.getSelectedDriverQueryCost()).thenReturn(mockQueryCost);
 QueryCollection mockQueries = mock(QueryCollection.class);
 EstimatedQueryCollection queries = new DefaultEstimatedQueryCollection(mockQueries);
 queries.add(mockQuery);
 assertEquals(queries.getQueriesCount(mockSelectedDriver), 1);
 verify(mockQueries, times(1)).add(mockQuery);
 queries.remove(mockQuery);
 assertEquals(queries.getQueriesCount(mockSelectedDriver), 0);
 verify(mockQueries, times(1)).remove(mockQuery);
}

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

@Test
public void testGetTotalQueryCostForUserWithMoreThanOneLaunchedQueries() throws LensException {
 QueryCollection mockQueries = mock(QueryCollection.class);
 Set<QueryContext> mockQueriesSet = createQueriesSetWithUserStubbing(2, MOCK_USER);
 when(mockQueries.getQueries(MOCK_USER)).thenReturn(mockQueriesSet);
 final QueryContext query0 = Iterables.get(mockQueriesSet, 0);
 final QueryContext query1 = Iterables.get(mockQueriesSet, 1);
 final QueryCost sCost0 = new StaticQueryCost(1.0);
 final QueryCost sCost1 = new StaticQueryCost(1.0);
 final QueryCost expectedCost = new StaticQueryCost(2.0);
 when(query0.getSelectedDriverQueryCost()).thenReturn(sCost0);
 when(query1.getSelectedDriverQueryCost()).thenReturn(sCost1);
 QueryCost actualQueryCost = new DefaultEstimatedQueryCollection(mockQueries).getTotalQueryCost(MOCK_USER);
 assertEquals(actualQueryCost, expectedCost);
}

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

QueryContext query = mock(QueryContext.class);
 when(query.getSelectedDriver()).thenReturn(mockDriver);
 when(query.getSelectedDriverQueryCost()).thenReturn(COST);
 col.add(query);
 QueryContext query = mock(QueryContext.class);
 when(query.getSelectedDriver()).thenReturn(mockDriver2);
 when(query.getSelectedDriverQueryCost()).thenReturn(COST);
 col.add(query);
when(mockCandidateQuery.getSelectedDriverQueryCost()).thenReturn(COST);
String actualCanLaunch = constraint.allowsLaunchOf(mockCandidateQuery, col);

相关文章

微信公众号

最新文章

更多