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

QueryContext.getSelectedDriver介绍

暂无

代码示例

代码示例来源：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: org.apache.lens/lens-server-api

public String getResultSetParentDir() {
 if (getSelectedDriver() != null && getSelectedDriverConf().get(LensConfConstants.RESULT_SET_PARENT_DIR) != null) {
  log.info("Fetching Parent Dir from driver conf:- "
      + getSelectedDriverConf().get(LensConfConstants.RESULT_SET_PARENT_DIR));
  return getSelectedDriverConf().get(LensConfConstants.RESULT_SET_PARENT_DIR);
 }
 return conf.get(LensConfConstants.RESULT_SET_PARENT_DIR, LensConfConstants.RESULT_SET_PARENT_DIR_DEFAULT);
}

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

/**
 * Gets the driver resultset.
 *
 * @param queryHandle the query handle
 * @return the driver resultset
 * @throws LensException the lens exception
 */
LensResultSet getDriverResultset(QueryHandle queryHandle) throws LensException {
 return allQueries.get(queryHandle).getSelectedDriver().fetchResultSet(allQueries.get(queryHandle));
}

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

@Override
 public void preLaunch(QueryContext ctx) throws LensException {
  log.debug("Pre launch for user {}, user query: {}, driver {}, driver query: {}", ctx.getSubmittedUser(),
   ctx.getUserQuery(), ctx.getSelectedDriver().getFullyQualifiedName(), ctx.getSelectedDriverQuery());
 }
}

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

@Override
 public void preLaunch(QueryContext ctx) throws LensException {
  log.debug("Pre launch for user {}, user query: {}, driver {}, driver query: {}", ctx.getSubmittedUser(),
   ctx.getUserQuery(), ctx.getSelectedDriver().getFullyQualifiedName(), ctx.getSelectedDriverQuery());
 }
}

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

private String canLaunchWithMaxConcurrentConstraint(QueryContext candidateQuery, int concurrentLaunched) {
 if (concurrentLaunched >= maxConcurrentQueries) {
  return concurrentLaunched + "/" + maxConcurrentQueries + " queries running on "
   + candidateQuery.getSelectedDriver().getFullyQualifiedName();
 }
 return null;
}
private String canLaunchWithQueueConstraint(QueryContext candidateQuery, Set<QueryContext> launchedQueries) {

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

public String getResultSetParentDir() {
 if (getSelectedDriver() != null && getSelectedDriverConf().get(LensConfConstants.RESULT_SET_PARENT_DIR) != null) {
  log.info("Fetching Parent Dir from driver conf:- "
      + getSelectedDriverConf().get(LensConfConstants.RESULT_SET_PARENT_DIR));
  return getSelectedDriverConf().get(LensConfConstants.RESULT_SET_PARENT_DIR);
 }
 return conf.get(LensConfConstants.RESULT_SET_PARENT_DIR, LensConfConstants.RESULT_SET_PARENT_DIR_DEFAULT);
}

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

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

代码示例来源：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

private String canLaunchWithMaxConcurrentConstraint(QueryContext candidateQuery, int concurrentLaunched) {
 if (concurrentLaunched >= maxConcurrentQueries) {
  return concurrentLaunched + "/" + maxConcurrentQueries + " queries running on "
   + candidateQuery.getSelectedDriver().getFullyQualifiedName();
 }
 return null;
}
private String canLaunchWithQueueConstraint(QueryContext candidateQuery, Set<QueryContext> launchedQueries) {

代码示例来源：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: apache/lens

public ImmutableSet<QueryLaunchingConstraint> getSelectedDriverQueryConstraints() {
 return getSelectedDriver().getQueryConstraints();
}

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

public void extractFailedAttempt() {
 extractFailedAttempt(getSelectedDriver());
}

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

QueryLauncher(QueryContext query) {
 this.query = query;
 log.info("Query launcher created for query {} on driver {}", query.getQueryHandle(),
  query.getSelectedDriver().getFullyQualifiedName());
 query.setLaunchTime(System.currentTimeMillis());
}

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

/**
 *
 * @param query
 * @return
 */
@Override
public boolean remove(QueryContext query) {
 this.queriesByDriver.remove(query.getSelectedDriver(), query);
 return this.queries.remove(query);
}

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

/**
 *
 * @param query
 * @return
 * @throws IllegalStateException if selected driver or selected driver query cost is not set for the query
 */
@Override
public boolean add(QueryContext query) {
 checkState(query);
 this.queriesByDriver.put(query.getSelectedDriver(), query);
 return this.queries.add(query);
}

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

@Test(expectedExceptions = IllegalStateException.class)
 public void testCheckStateMustRecognizeIllegalStateWhenQueryCostIsNotSet() {

  QueryContext mockQuery = mock(QueryContext.class);
  /* Selected Driver is set, however since selected driver query cost is not set. This should result in
  IllegalStateException. */
  when(mockQuery.getSelectedDriver()).thenReturn(mock(LensDriver.class));

  QueryCollection mockQueries = mock(QueryCollection.class);
  DefaultEstimatedQueryCollection queries = new DefaultEstimatedQueryCollection(mockQueries);
  queries.checkState(mockQuery);
 }
}

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

@Test(dataProvider = "dpTestAllowsLaunchOfQuery")
public void testAllowsLaunchOfQuery(final int currentDriverLaunchedQueries, final boolean expectedCanLaunch) {
 QueryContext mockCandidateQuery = mock(QueryContext.class);
 EstimatedImmutableQueryCollection mockLaunchedQueries = mock(EstimatedImmutableQueryCollection.class);
 LensDriver mockDriver = mock(LensDriver.class);
 when(mockCandidateQuery.getSelectedDriver()).thenReturn(mockDriver);
 when(mockLaunchedQueries.getQueriesCount(mockDriver)).thenReturn(currentDriverLaunchedQueries);
 String actualCanLaunch = constraint.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: apache/lens

private BackOffRetryHandler<QueryContext> getDriverRetryPolicy(QueryContext ctx) {
 if (ctx.getDriverRetryPolicy() == null) {
  String errorMessage = ctx.getDriverStatus().getErrorMessage() != null ? ctx.getDriverStatus().getErrorMessage()
   : ctx.getStatus().getErrorMessage();
  ctx.setDriverRetryPolicy(ctx.getSelectedDriver().getRetryPolicyDecider()
   .decidePolicy(errorMessage));
 }
 return ctx.getDriverRetryPolicy();
}