/** * Sets the database to connect on lens server * * @param dbName database to connect to */ public void setLensDatabase(String dbName) { this.set(DBNAME_KEY, dbName); }
@Test public void testWaitForQueryToCompleteWithAndWithoutRetryOnTimeOut() throws LensAPIException { LensClientConfig config = createLensClientConfigWithServerUrl(); config.set(ENABLE_SLEEP_FOR_GET_QUERY_OP, "true"); config.setInt(LensClientConfig.READ_TIMEOUT_MILLIS, 3000); try (LensClient lensClient = new LensClient(config)) { assertTrue(lensClient.setDatabase(TEST_DB)); //Test waitForQueryToComplete without retry on timeout QueryHandle handle = lensClient.executeQueryAsynch("cube select id,name from test_dim", "test3", new LensConf()); try { lensClient.getStatement().waitForQueryToComplete(handle, false); fail("SocketTimeoutException was expected"); } catch (Exception e) { if (!isExceptionDueToSocketTimeout(e)) { fail("SocketTimeoutException was excepted as part of Read Timeout"); } } //Test waitForQueryToComplete with Retry on timeout handle = lensClient.executeQueryAsynch("cube select id,name from test_dim", "test3", new LensConf()); lensClient.getStatement().waitForQueryToComplete(handle); LensQuery query = lensClient.getQueryDetails(handle); assertTrue(query.getStatus().successful()); } }
@BeforeTest public void setUp() throws Exception { super.setUp(); Hive hive = Hive.get(new HiveConf()); Database db = new Database(); db.setName(TEST_DB); hive.createDatabase(db, true); LensClientConfig lensClientConfig = new LensClientConfig(); lensClientConfig.setLensDatabase(TEST_DB); lensClientConfig.set(LensConfConstants.SERVER_BASE_URL, "http://localhost:" + getTestPort() + "/lensapi"); LensClient client = new LensClient(lensClientConfig); mlClient = new LensMLClient(client); }
/** * Sets the database to connect on lens server * * @param dbName database to connect to */ public void setLensDatabase(String dbName) { this.set(DBNAME_KEY, dbName); }
@BeforeTest public void setUp() throws Exception { super.setUp(); Hive hive = Hive.get(new HiveConf()); Database db = new Database(); db.setName(TEST_DB); hive.createDatabase(db, true); LensClientConfig lensClientConfig = new LensClientConfig(); lensClientConfig.setLensDatabase(TEST_DB); lensClientConfig.set(LensConfConstants.SERVER_BASE_URL, "http://localhost:" + getTestPort() + "/lensapi"); LensClient client = new LensClient(lensClientConfig); mlClient = new LensMLClient(client); }
public void setUser(String user) { this.set(USER_NAME, user); }
public void setBaseUrl(String baseUrl) { this.set(SERVER_BASE_URL, baseUrl); }
public void setUser(String user) { this.set(USER_NAME, user); }
public void setBaseUrl(String baseUrl) { this.set(SERVER_BASE_URL, baseUrl); }
/** * Sets the param. * * @param keyval the keyval * @return the string */ @CliCommand(value = "set", help = "Assign <value> to session parameter specified with <key> on lens server") public String setParam(@CliOption(key = {""}, mandatory = true, help = "<key>=<value>") String keyval) { String[] pair = keyval.split("="); if (pair.length != 2) { return "Error: Pass parameter as <key>=<value>"; } if (pair[0].startsWith(LensCliConfigConstants.LENS_CLI_PREFIX) || pair[0].startsWith(LensClientConfig.CLIENT_PFX)) { getClient().getConf().set(pair[0], pair[1]); return "Client side Set " + pair[0] + "=" + pair[1]; } else { APIResult result = getClient().setConnectionParam(pair[0], pair[1]); return result.getMessage(); } }
/** * Sets the param. * * @param keyval the keyval * @return the string */ @CliCommand(value = "set", help = "Assign <value> to session parameter specified with <key> on lens server") public String setParam(@CliOption(key = {""}, mandatory = true, help = "<key>=<value>") String keyval) { String[] pair = keyval.split("="); if (pair.length != 2) { return "Error: Pass parameter as <key>=<value>"; } if (pair[0].startsWith(LensCliConfigConstants.LENS_CLI_PREFIX) || pair[0].startsWith(LensClientConfig.CLIENT_PFX)) { getClient().getConf().set(pair[0], pair[1]); return "Client side Set " + pair[0] + "=" + pair[1]; } else { APIResult result = getClient().setConnectionParam(pair[0], pair[1]); return result.getMessage(); } }
private LensClientConfig createLensClientConfigWithServerUrl() { LensClientConfig config = new LensClientConfig(); config.set("lens.server.base.url", "http://localhost:" + getTestPort() + "/lensapi"); return config; } }