use of com.thinkbiganalytics.kylo.spark.exceptions.LivyException in project kylo by Teradata.
the class DefaultLivyClient method getSessions.
@Override
public SessionsGetResponse getSessions(JerseyRestClient client) {
try {
SessionsGetResponse sessions = client.get(SESSIONS_URL, SessionsGetResponse.class);
livyServer.setLivyServerStatus(LivyServerStatus.alive);
logger.debug("sessions={}", sessions);
return sessions;
} catch (WebApplicationException wae) {
if (wae.getResponse().getStatusInfo().getFamily() != Response.Status.Family.SUCCESSFUL) {
livyServer.setLivyServerStatus(LivyServerStatus.http_error);
}
throw wae;
} catch (LivyException le) {
livyServer.setLivyServerStatus(LivyServerStatus.http_error);
throw le;
}
}
use of com.thinkbiganalytics.kylo.spark.exceptions.LivyException in project kylo by Teradata.
the class DefaultLivyClient method getStatement.
@Override
public Statement getStatement(JerseyRestClient client, SparkLivyProcess sparkLivyProcess, Integer statementId) {
Integer sessionId = sparkLivyProcess.getSessionId();
try {
Statement statement = client.get(STATEMENT_URL(sessionId, statementId), Statement.class);
livyServer.setLivyServerStatus(LivyServerStatus.alive);
logger.debug("statement={}", statement);
return statement;
} catch (WebApplicationException wae) {
if (wae.getResponse().getStatus() != 404 || wae.getResponse().getStatusInfo().getFamily() != Response.Status.Family.SUCCESSFUL) {
livyServer.setLivyServerStatus(LivyServerStatus.http_error);
}
throw wae;
} catch (LivyException le) {
livyServer.setLivyServerStatus(LivyServerStatus.http_error);
throw le;
}
}
use of com.thinkbiganalytics.kylo.spark.exceptions.LivyException in project kylo by Teradata.
the class DefaultLivyClient method postSessions.
@Override
public Session postSessions(JerseyRestClient client, SessionsPost sessionsPost) {
try {
Session session = client.post(SESSIONS_URL, sessionsPost, Session.class);
livyServer.setLivyServerStatus(LivyServerStatus.alive);
livyServer.setSessionIdHighWaterMark(session.getId());
livyServer.setLivySessionState(session.getId(), session.getState());
logger.debug("session={}", session);
return session;
} catch (WebApplicationException wae) {
if (wae.getResponse().getStatusInfo().getFamily() != Response.Status.Family.SUCCESSFUL) {
livyServer.setLivyServerStatus(LivyServerStatus.http_error);
}
throw wae;
} catch (LivyException le) {
livyServer.setLivyServerStatus(LivyServerStatus.http_error);
throw le;
}
}
use of com.thinkbiganalytics.kylo.spark.exceptions.LivyException in project kylo by Teradata.
the class DefaultLivyClient method postStatement.
@Override
public Statement postStatement(JerseyRestClient client, SparkLivyProcess sparkLivyProcess, StatementsPost sp) {
Integer sessionId = sparkLivyProcess.getSessionId();
try {
Statement statement = client.post(STATEMENTS_URL(sessionId), sp, Statement.class);
livyServer.setLivyServerStatus(LivyServerStatus.alive);
logger.debug("statement={}", statement);
return statement;
} catch (WebApplicationException wae) {
if (wae.getResponse().getStatus() != 404 || wae.getResponse().getStatusInfo().getFamily() != Response.Status.Family.SUCCESSFUL) {
livyServer.setLivyServerStatus(LivyServerStatus.http_error);
}
throw wae;
} catch (LivyException le) {
livyServer.setLivyServerStatus(LivyServerStatus.http_error);
throw le;
}
}
use of com.thinkbiganalytics.kylo.spark.exceptions.LivyException in project kylo by Teradata.
the class SparkLivyProcessManager method startLivySession.
private Session startLivySession(SparkLivyProcess sparkLivyProcess) {
// it was determined we needed a
sparkLivyProcess.newSession();
JerseyRestClient jerseyClient = getClient(sparkLivyProcess);
Map<String, String> sparkProps = livyProperties.getSparkProperties();
SessionsPost.Builder builder = new SessionsPost.Builder().kind(// "shared" most likely
livyProperties.getLivySessionKind().toString()).conf(// "spark.driver.extraJavaOptions", "-agentlib:jdwp=transport=dt_socket,server=y,suspend=y,address=8990"
sparkProps);
logger.debug("LivyProperties={}", livyProperties);
if (livyProperties.getProxyUser()) {
String user = processCache.inverse().get(sparkLivyProcess);
if (user != null) {
builder.proxyUser(user);
}
}
SessionsPost sessionsPost = builder.build();
logger.info("sessionsPost={}", sessionsPost);
Session currentSession;
try {
currentSession = livyClient.postSessions(jerseyClient, sessionsPost);
if (currentSession == null) {
logger.error("Server not reachable", new LivyServerNotReachableException("Empty result from LivyClient.postSessions"));
throw new LivyUserException("livy.server_not_found");
}
} catch (LivyException le) {
throw le;
} catch (Exception e) {
// resets the latch
sparkLivyProcess.startFailed();
this.processStopped(sparkLivyProcess);
// NOTE: you can get "javax.ws.rs.ProcessingException: java.io.IOException: Error writing to server" on Ubuntu see: https://stackoverflow.com/a/39718929/154461
throw new LivyException(e);
}
sparkLivyProcess.setSessionId(currentSession.getId());
// notifies listeners
this.processStarted(sparkLivyProcess);
// begin monitoring this session if configured to do so..
heartbeatMonitor.monitorSession(sparkLivyProcess);
return currentSession;
}
Aggregations