Examples with ConsoleCommandOp io.jpom.socket.ConsoleCommandOp used on opensource projects

Example 1 with ConsoleCommandOp

use of io.jpom.socket.ConsoleCommandOp in project Jpom by dromara.

the class DockerLogHandler method handleTextMessage.

@Override
protected String handleTextMessage(Map<String, Object> attributes, WebSocketSession session, JSONObject json, ConsoleCommandOp consoleCommandOp) throws IOException {
    DockerInfoModel dockerInfoModel = (DockerInfoModel) attributes.get("dataItem");
    if (consoleCommandOp == ConsoleCommandOp.heart) {
        return null;
    }
    if (consoleCommandOp == ConsoleCommandOp.showlog) {
        super.logOpt(this.getClass(), attributes, json);
        String containerId = json.getString("containerId");
        Map<String, Object> map = dockerInfoModel.toParameter();
        map.put("containerId", containerId);
        int tail = json.getIntValue("tail");
        if (tail > 0) {
            map.put("tail", tail);
        }
        Consumer<String> consumer = s -> {
            try {
                SocketSessionUtil.send(session, s);
            } catch (IOException e) {
                log.error("发消息异常", e);
            }
        };
        map.put("charset", CharsetUtil.CHARSET_UTF_8);
        map.put("consumer", consumer);
        IPlugin plugin = PluginFactory.getPlugin(DockerInfoService.DOCKER_PLUGIN_NAME);
        try {
            plugin.execute("logContainer", map);
        } catch (Exception e) {
            log.error("拉取 容器日志异常", e);
            return "执行异常:" + e.getMessage();
        }
    } else {
        return null;
    }
    return null;
}