docker里Java服务执行ping命令模拟流式输出

文章目录

    • 业务场景
    • 处理解决
      • 实现ping功能并实时返回输出
      • 实现长ping和中断请求
      • docker容器找不到ping命令处理

业务场景

  • 我们某市的客户,一直使用CS版本的信控平台,直接安装客户Windows server服务器上,主要对信号机设备进行在线管理、方案配时、管控等
  • 其中有一项功能,在网络波动情况,对信号机管控失败,判断信号机是否在线。大致方法是直接调用Windows的dos窗口,发送 ping ip的命令,显示网络情况
  • 我们新的信控平台使用Spring Cloud微服务架构,使用Spring Boot构建Java服务,使用google的jib插件打成docker镜像包
  • 我们使用docker虚拟化部署,使用docker-compose统一管理所有服务,部署在Linux服务器里
  • 客户很喜欢之前的功能,需要我们在新平台里实现这个功能,调用dos窗口,ping网络
  • 而我们新平台是B/S架构,浏览器是很难调用Windows组件去弹出窗口实现ping功能,而且我们也没有限制一定使用的是Windows电脑访问,有网有浏览器就行
  • ping功能无论Windows还是Linux,都是有的,至于界面展现,只能自己实现了
    docker里Java服务执行ping命令模拟流式输出_第1张图片

处理解决

实现ping功能并实时返回输出

  • 代码实现,有两个核心功能点
  • 一是根据不同的操作系统,执行对应的系统命令,进行结果接收与解析
    • 对于第一个问题,Java有现成的类库,使用Runtime.getRuntime().exec(ping命令)即可
    • 对于Windows服务器,需要使用GB2312解析命令执行结果
    • 对于Linux 服务器,需要使用UTF_8解析命令执行结果
    • 对于ServletOutputStream.println输出中文字符串报错Not an ISO 8859-1 character问题,可以使用PrintWriter.println输出代替
    • 也可以在ServletOutputStream.println输出时输出字符数组(string.getBytes()
  • 二是流式输出到请求端,模拟再现一秒一次的逐步展示的效果
    • 对于第二个问题,核心是命令执行的结果输出流,要实时的返回给请求端,请求端能接收到
    • 主要是获取流,然后按行读取,按行flush()即可返回给请求端
    • 对于请求端实时渲染,需要在代码的response里指定ContentTypetext/event-stream,这样flush刷新的返回流,才能实时被前端浏览器接收到(ChatGPT流式输出也是使用的这种content-type
    • 一开始是考虑使用multipart,完全不行,流flush后,浏览器无法获取,只能在流输出完成后,浏览器才能获取到
  • 具体代码如下:
    /**
     * 获取信号机的网络状态
     * @param ip
     * @param count
     * @param response
     */
    @GetMapping("/ping/start")
    public void ping(String ip, Integer count, HttpServletResponse response) {
        logger.info("ping 信号机【{}】 开始 ......", ip);
        response.setStatus(HttpServletResponse.SC_OK);
        response.setContentType("text/event-stream");
        response.setCharacterEncoding("UTF-8");
        String line = null;
        Process pro = null;
        BufferedReader buf = null;
        try {
            if (null == count) {
                count = 4;
            }
            String osName = System.getProperty("os.name");
            if (osName.toLowerCase().contains("windows")){
                pro = Runtime.getRuntime().exec("ping -n " + count + " " + ip);
                buf = new BufferedReader(new InputStreamReader(pro.getInputStream(), "GB2312"));
            } else if (osName.toLowerCase().contains("linux")){
                pro = Runtime.getRuntime().exec("ping -c " + count + " " + ip);
                buf = new BufferedReader(new InputStreamReader(pro.getInputStream(), StandardCharsets.UTF_8));
            }
            PrintWriter out = response.getWriter();
            while (null != buf && (line = buf.readLine()) != null){
                out.println(line);
                out.flush();
            }
            logger.info("执行ping请求结束!");
            out.close();
        } catch (Exception e){
            logger.error("执行ping命令出现异常");
            e.printStackTrace();
        }finally {
            if (null != pro){
                pro.destroy();
            }
        }
    }

实现长ping和中断请求

  • 主要是在请求时传输一个唯一命令id,缓存到内存里
  • 当命令执行完成,或者接收到打断请求时,调用destroy()打断循环,结束请求
  • 当然,可以尝试使用kill -2去模拟CTRL + C的打断,可以使用Runtime.getRuntime().exec(中断命令)打断试下,我的代码已经满足自己的需求了,就没再尝试,有兴趣的小伙伴可以试一下
  • 具体代码如下:

package com.newatc.api.rest;


import com.newatc.api.signalcontrol.dto.PingRequestVO;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpServletResponse;
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.nio.charset.StandardCharsets;
import java.util.HashMap;
import java.util.Map;

/**
 * 调用系统命令并返回
 *
 * @author yanyulin
 * @date 2024-1-16 15:11:55
 */
@RestController
@RequestMapping(value = "/api/syscmd")
public class SysCmdController {

    private static final Logger logger = LoggerFactory.getLogger(SysCmdController.class);

    /**
     * 命令id-执行过程map
     */
    public static final Map<String, Boolean> COMMAND_REQUEST_MAP = new HashMap<>();
    /**
     * 开始信号机的网络状态诊断
     */
    @PostMapping("/ping/start")
    public void ping(@RequestBody PingRequestVO pingRequest, HttpServletResponse response) {
        String ip = pingRequest.getIp();
        String cmdId = pingRequest.getCmdId();
        Integer count = pingRequest.getCount();
        logger.info("ping 信号机【{}】 开始 ......", ip);
        response.setStatus(HttpServletResponse.SC_OK);
        response.setContentType("text/event-stream");
        response.setCharacterEncoding("UTF-8");
        String line = null;
        Process pro = null;
        BufferedReader buf = null;
        try {
            if (null == count) {
                count = 4;
            }
            if (count > 50) {
                count = 50;
            }
            String osName = System.getProperty("os.name");
            if (osName.toLowerCase().contains("windows")){
                pro = Runtime.getRuntime().exec("ping -n " + count + " " + ip);
                buf = new BufferedReader(new InputStreamReader(pro.getInputStream(), "GB2312"));
            } else if (osName.toLowerCase().contains("linux")){
                pro = Runtime.getRuntime().exec("ping -c " + count + " " + ip);
                buf = new BufferedReader(new InputStreamReader(pro.getInputStream(), StandardCharsets.UTF_8));
            }
            COMMAND_REQUEST_MAP.put(cmdId, true);
            PrintWriter out = response.getWriter();
            while (null != buf && (line = buf.readLine()) != null){
                out.println(line);
                out.flush();
                if (!COMMAND_REQUEST_MAP.get(cmdId)) {
                    pro.destroy();
                }
            }
            logger.info("执行ping请求结束!");
            out.close();
        } catch (Exception e){
            logger.error("执行ping命令出现异常");
            e.printStackTrace();
        }finally {
            if (null != pro){
                pro.destroy();
            }
            COMMAND_REQUEST_MAP.remove(cmdId);
        }
    }

    /**
     * 打断命令执行状态
     */
    @PostMapping("/ping/stop")
    public void ping(@RequestBody PingRequestVO requestVO) {
        COMMAND_REQUEST_MAP.put(requestVO.getCmdId(), false);
    }
}

docker容器找不到ping命令处理

  • 我们打包导出的docker镜像,无法使用ping命令,报错,找不到这个命令bash: ping:command not found
  • 我们使用的是极简镜像eclipse-temurin:11-jre-focal,这个版本里的ubuntu没有安装不需要的命令
  • 具体可以参考我的这篇博文:《自制Java镜像发布到dockerhub公网使用》
  • 也可以直接使用我发布到公网的包含ping命令的jre11镜像文件1363241277/jre11:11-jre-focal
  • 主要思路,就是打包使用的原始Java镜像里,要已经安装ping等需要的命令

你可能感兴趣的:(java,Linux,docker,java,容器)