本文整理了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
暂无
代码示例来源: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);
内容来源于网络,如有侵权,请联系作者删除!