@Override public void setDriver(LensDriver driver) { this.driver = driver; log.debug("The Driver for this driver query hook is {}", driver.getFullyQualifiedName()); }
@Override public void setDriver(LensDriver driver) { this.driver = driver; log.debug("The Driver for this driver query hook is {}", driver.getFullyQualifiedName()); }
private LensDriver getDriverFromName(Collection<LensDriver> drivers) { for (LensDriver driver : drivers) { if (driverName.equals(driver.getFullyQualifiedName())) { return driver; } } return null; }
private LensDriver getDriverFromName(Collection<LensDriver> drivers) { for (LensDriver driver : drivers) { if (driverName.equals(driver.getFullyQualifiedName())) { return driver; } } return null; }
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) {
private void captureExceptionInformation(final DriverQueryContext driverQueryContext, final Exception e) { String expMsg = LensUtil.getCauseMessage(e); driverQueryContext.setDriverQueryCostEstimateError(e); failureCause = new StringBuilder("Driver :") .append(driver.getFullyQualifiedName()) .append(" Cause :") .append(expMsg) .toString(); log.error("Setting driver cost failed for driver {} Cause: {}", driver, failureCause, e); } }
@Override public void preRewrite(AbstractQueryContext ctx) throws LensException { log.debug("Pre rewrite for user {}, user query: {}, driver: {}", ctx.getSubmittedUser(), ctx.getUserQuery(), driver.getFullyQualifiedName()); }
@Override public void preRewrite(AbstractQueryContext ctx) throws LensException { log.debug("Pre rewrite for user {}, user query: {}, driver: {}", ctx.getSubmittedUser(), ctx.getUserQuery(), driver.getFullyQualifiedName()); }
@Override public void preEstimate(AbstractQueryContext ctx) throws LensException { log.debug("Pre estimate for user {}, user query: {}, driver: {}, driver query :{}", ctx.getSubmittedUser(), ctx.getUserQuery(), driver.getFullyQualifiedName(), ctx.getDriverQuery(driver)); }
@Override public void postRewrite(AbstractQueryContext ctx) throws LensException { log.debug("Post rewrite for user {}, user query: {}, driver: {}, driver query :{}", ctx.getSubmittedUser(), ctx.getUserQuery(), driver.getFullyQualifiedName(), ctx.getDriverQuery(driver)); }
@Override public void postRewrite(AbstractQueryContext ctx) throws LensException { log.debug("Post rewrite for user {}, user query: {}, driver: {}, driver query :{}", ctx.getSubmittedUser(), ctx.getUserQuery(), driver.getFullyQualifiedName(), ctx.getDriverQuery(driver)); }
@Override public void preEstimate(AbstractQueryContext ctx) throws LensException { log.debug("Pre estimate for user {}, user query: {}, driver: {}, driver query :{}", ctx.getSubmittedUser(), ctx.getUserQuery(), driver.getFullyQualifiedName(), ctx.getDriverQuery(driver)); }
@Override public void postEstimate(AbstractQueryContext ctx) throws LensException { log.debug("Post estimate for user {}, user query: {}, driver: {}, driver query :{}, query cost :{}", ctx.getSubmittedUser(), ctx.getUserQuery(), driver.getFullyQualifiedName(), ctx.getDriverQuery(driver), ctx.getDriverQueryCost(driver)); }
@Override public void postDriverSelection(AbstractQueryContext ctx) throws LensException { log.debug("Post driver selection for user {}, user query: {}, driver {}, driver query: {}", ctx.getSubmittedUser(), ctx.getUserQuery(), ctx.getSelectedDriver().getFullyQualifiedName(), ctx.getSelectedDriverQuery()); }
@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()); } }
@Test public void testLoadingMultipleDrivers() { Collection<LensDriver> drivers = queryService.getDrivers(); assertEquals(drivers.size(), 4); Set<String> driverNames = new HashSet<>(drivers.size()); for (LensDriver driver : drivers) { assertEquals(driver.getConf().get("lens.driver.test.drivername"), driver.getFullyQualifiedName()); driverNames.add(driver.getFullyQualifiedName()); } assertTrue(driverNames.containsAll(Arrays.asList("hive/hive1", "hive/hive2", "jdbc/jdbc1", "mock/fail1"))); }
@Override public void postEstimate(AbstractQueryContext ctx) throws LensException { log.debug("Post estimate for user {}, user query: {}, driver: {}, driver query :{}, query cost :{}", ctx.getSubmittedUser(), ctx.getUserQuery(), driver.getFullyQualifiedName(), ctx.getDriverQuery(driver), ctx.getDriverQueryCost(driver)); }
/** * To prepared query. * * @return the lens prepared query */ public LensPreparedQuery toPreparedQuery() { return new LensPreparedQuery(prepareHandle, userQuery, preparedTime, preparedUser, getDriverContext() .getSelectedDriver() != null ? getDriverContext().getSelectedDriver().getFullyQualifiedName() : null, getDriverContext().getSelectedDriverQuery(), lensConf); }
/** * To prepared query. * * @return the lens prepared query */ public LensPreparedQuery toPreparedQuery() { return new LensPreparedQuery(prepareHandle, userQuery, preparedTime, preparedUser, getDriverContext() .getSelectedDriver() != null ? getDriverContext().getSelectedDriver().getFullyQualifiedName() : null, getDriverContext().getSelectedDriverQuery(), lensConf); }
public void extractFailedAttempt(LensDriver selectedDriver) { failedAttempts.add(new FailedAttempt(selectedDriver.getFullyQualifiedName(), getDriverStatus().getProgress(), getDriverStatus().getProgressMessage(), getDriverStatus().getErrorMessage(), getDriverStatus().getDriverStartTime(), getDriverStatus().getDriverFinishTime())); getDriverStatus().clear(); }