Examples with Client org.apache.zeppelin.interpreter.thrift.RemoteInterpreterService.Client used on opensource projects

Example 6 with Client

use of org.apache.zeppelin.interpreter.thrift.RemoteInterpreterService.Client in project zeppelin by apache.

the class RemoteAngularObjectRegistry method removeAndNotifyRemoteProcess.

/**
   * When ZeppelinServer side code want to remove angularObject from the registry,
   * this method should be used instead of remove()
   * @param name
   * @param noteId
   * @param paragraphId
   * @return
   */
public AngularObject removeAndNotifyRemoteProcess(String name, String noteId, String paragraphId) {
    RemoteInterpreterProcess remoteInterpreterProcess = getRemoteInterpreterProcess();
    if (remoteInterpreterProcess == null || !remoteInterpreterProcess.isRunning()) {
        return super.remove(name, noteId, paragraphId);
    }
    Client client = null;
    boolean broken = false;
    try {
        client = remoteInterpreterProcess.getClient();
        client.angularObjectRemove(name, noteId, paragraphId);
        return super.remove(name, noteId, paragraphId);
    } catch (TException e) {
        broken = true;
        logger.error("Error", e);
    } catch (Exception e) {
        logger.error("Error", e);
    } finally {
        if (client != null) {
            remoteInterpreterProcess.releaseClient(client, broken);
        }
    }
    return null;
}

Example 7 with Client

use of org.apache.zeppelin.interpreter.thrift.RemoteInterpreterService.Client in project zeppelin by apache.

the class RemoteAngularObjectRegistry method addAndNotifyRemoteProcess.

/**
   * When ZeppelinServer side code want to add angularObject to the registry,
   * this method should be used instead of add()
   * @param name
   * @param o
   * @param noteId
   * @return
   */
public AngularObject addAndNotifyRemoteProcess(String name, Object o, String noteId, String paragraphId) {
    Gson gson = new Gson();
    RemoteInterpreterProcess remoteInterpreterProcess = getRemoteInterpreterProcess();
    if (!remoteInterpreterProcess.isRunning()) {
        return super.add(name, o, noteId, paragraphId, true);
    }
    Client client = null;
    boolean broken = false;
    try {
        client = remoteInterpreterProcess.getClient();
        client.angularObjectAdd(name, noteId, paragraphId, gson.toJson(o));
        return super.add(name, o, noteId, paragraphId, true);
    } catch (TException e) {
        broken = true;
        logger.error("Error", e);
    } catch (Exception e) {
        logger.error("Error", e);
    } finally {
        if (client != null) {
            remoteInterpreterProcess.releaseClient(client, broken);
        }
    }
    return null;
}

Example 8 with Client

use of org.apache.zeppelin.interpreter.thrift.RemoteInterpreterService.Client in project zeppelin by apache.

the class RemoteInterpreter method interpret.

@Override
public InterpreterResult interpret(String st, InterpreterContext context) {
    if (logger.isDebugEnabled()) {
        logger.debug("st:\n{}", st);
    }
    FormType form = getFormType();
    RemoteInterpreterProcess interpreterProcess = getInterpreterProcess();
    Client client = null;
    try {
        client = interpreterProcess.getClient();
    } catch (Exception e1) {
        throw new InterpreterException(e1);
    }
    InterpreterContextRunnerPool interpreterContextRunnerPool = interpreterProcess.getInterpreterContextRunnerPool();
    List<InterpreterContextRunner> runners = context.getRunners();
    if (runners != null && runners.size() != 0) {
        // assume all runners in this InterpreterContext have the same note id
        String noteId = runners.get(0).getNoteId();
        interpreterContextRunnerPool.clear(noteId);
        interpreterContextRunnerPool.addAll(noteId, runners);
    }
    boolean broken = false;
    try {
        final GUI currentGUI = context.getGui();
        RemoteInterpreterResult remoteResult = client.interpret(sessionKey, className, st, convert(context));
        Map<String, Object> remoteConfig = (Map<String, Object>) gson.fromJson(remoteResult.getConfig(), new TypeToken<Map<String, Object>>() {
        }.getType());
        context.getConfig().clear();
        context.getConfig().putAll(remoteConfig);
        if (form == FormType.NATIVE) {
            GUI remoteGui = gson.fromJson(remoteResult.getGui(), GUI.class);
            currentGUI.clear();
            currentGUI.setParams(remoteGui.getParams());
            currentGUI.setForms(remoteGui.getForms());
        } else if (form == FormType.SIMPLE) {
            final Map<String, Input> currentForms = currentGUI.getForms();
            final Map<String, Object> currentParams = currentGUI.getParams();
            final GUI remoteGUI = gson.fromJson(remoteResult.getGui(), GUI.class);
            final Map<String, Input> remoteForms = remoteGUI.getForms();
            final Map<String, Object> remoteParams = remoteGUI.getParams();
            currentForms.putAll(remoteForms);
            currentParams.putAll(remoteParams);
        }
        InterpreterResult result = convert(remoteResult);
        return result;
    } catch (TException e) {
        broken = true;
        throw new InterpreterException(e);
    } finally {
        interpreterProcess.releaseClient(client, broken);
    }
}

Example 9 with Client

use of org.apache.zeppelin.interpreter.thrift.RemoteInterpreterService.Client in project zeppelin by apache.

the class RemoteInterpreter method completion.

@Override
public List<InterpreterCompletion> completion(String buf, int cursor) {
    RemoteInterpreterProcess interpreterProcess = getInterpreterProcess();
    Client client = null;
    try {
        client = interpreterProcess.getClient();
    } catch (Exception e1) {
        throw new InterpreterException(e1);
    }
    boolean broken = false;
    try {
        List completion = client.completion(sessionKey, className, buf, cursor);
        return completion;
    } catch (TException e) {
        broken = true;
        throw new InterpreterException(e);
    } finally {
        interpreterProcess.releaseClient(client, broken);
    }
}

Example 10 with Client

use of org.apache.zeppelin.interpreter.thrift.RemoteInterpreterService.Client in project zeppelin by apache.

the class RemoteInterpreter method getFormType.

@Override
public FormType getFormType() {
    init();
    if (formType != null) {
        return formType;
    }
    RemoteInterpreterProcess interpreterProcess = getInterpreterProcess();
    Client client = null;
    try {
        client = interpreterProcess.getClient();
    } catch (Exception e1) {
        throw new InterpreterException(e1);
    }
    boolean broken = false;
    try {
        formType = FormType.valueOf(client.getFormType(sessionKey, className));
        return formType;
    } catch (TException e) {
        broken = true;
        throw new InterpreterException(e);
    } finally {
        interpreterProcess.releaseClient(client, broken);
    }
}