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