@Override public CreateSessionResponse createSession(RpcController controller, CreateSessionRequest request) throws ServiceException { try { // create a new session with base database name. If no database name is give, we use default database. String databaseName = request.hasBaseDatabaseName() ? request.getBaseDatabaseName() : DEFAULT_DATABASE_NAME; if (!context.getCatalog().existDatabase(databaseName)) { LOG.info("Session creation is canceled due to absent base database \"" + databaseName + "\"."); throw new UndefinedDatabaseException(databaseName); } String sessionId = context.getSessionManager().createSession(request.getUsername(), databaseName); CreateSessionResponse.Builder builder = CreateSessionResponse.newBuilder(); builder.setState(OK); builder.setSessionId(TajoIdProtos.SessionIdProto.newBuilder().setId(sessionId).build()); builder.setSessionVars(ProtoUtil.convertFromMap(context.getSessionManager().getAllVariables(sessionId))); return builder.build(); } catch (Throwable t) { printStackTraceIfError(LOG, t); CreateSessionResponse.Builder builder = CreateSessionResponse.newBuilder(); builder.setState(returnError(t)); return builder.build(); } }
@Override public CreateSessionResponse createSession(RpcController controller, CreateSessionRequest request) throws ServiceException { try { // create a new session with base database name. If no database name is give, we use default database. String databaseName = request.hasBaseDatabaseName() ? request.getBaseDatabaseName() : DEFAULT_DATABASE_NAME; if (!context.getCatalog().existDatabase(databaseName)) { LOG.info("Session creation is canceled due to absent base database \"" + databaseName + "\"."); throw new UndefinedDatabaseException(databaseName); } String sessionId = context.getSessionManager().createSession(request.getUsername(), databaseName); CreateSessionResponse.Builder builder = CreateSessionResponse.newBuilder(); builder.setState(OK); builder.setSessionId(TajoIdProtos.SessionIdProto.newBuilder().setId(sessionId).build()); builder.setSessionVars(ProtoUtil.convertFromMap(context.getSessionManager().getAllVariables(sessionId))); return builder.build(); } catch (Throwable t) { printStackTraceIfError(LOG, t); CreateSessionResponse.Builder builder = CreateSessionResponse.newBuilder(); builder.setState(returnError(t)); return builder.build(); } }
private QueryIdRequest buildQueryIdRequest(QueryId queryId) { return QueryIdRequest.newBuilder() .setSessionId(SessionIdProto.newBuilder().setId(getSessionId())) .setQueryId(queryId.getProto()) .build(); }
private QueryIdRequest buildQueryIdRequest(QueryId queryId) { return QueryIdRequest.newBuilder() .setSessionId(SessionIdProto.newBuilder().setId(getSessionId())) .setQueryId(queryId.getProto()) .build(); }