java通过websocket调用python脚本实时输出信息_java通过websocket调用python服务-优快云博客

我们这边的系统有自动化发布和调用jenkins集成（调用python脚本）等等依托于调用python脚本的功能。但是脚本调用又是一个比较耗时的操作，前端又怎么实时看到脚本的进度是一个比较难以解决的麻烦事。

环境描述：

后端是java，springboot项目，前端请求java代码，java代码触发python脚本工作。

实现方式：

客户端想要实时看到服务端python进度的功能，无非两种解决方法。
一是客户端主动拉取进度。
二是服务端主动推送进度。

针对以上两种，我们这里用到了两种具体的实现方式。

客户端主动拉取进度

客户端触发脚本工作以后，还要主动轮训获取进度信息。
这时候，相当于有两个接口。

触发python脚本工作（python xxxx.py）的接口，同时将脚本里实时输出的最新进度存在redis里。
第二个接口实时去获取redis里最新的进度的信息。

服务端主动推送进度

通过websocket长连接的方式，实时将进度反馈给客户端。

实现代码

在这里以服务端主动推送进度，websocket的方式为例。
java项目，基于spingboot。

在pom.xml加入websocket的依赖

  <!--import websocket-->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-websocket</artifactId>
        </dependency>

全局配置WebSocket

@Configuration
public class WebSocketConfig {
   
   
    @Bean
    public ServerEndpointExporter serverEndpointExporter(){
   
   
        return new ServerEndpointExporter();
    }
}

写业务逻辑处理的websocket接口

@RestController
@ServerEndpoint("/websocket")
@Slf4j
public class WebSocketHandler {
   
   

    @Autowired
    private AsyncTaskExecutor taskExecutor;

    // concurrent包的线程安全Set，用来存放每个客户端对应的Session对象。
    private static final CopyOnWriteArraySet<Session> SessionSet = new CopyOnWriteArraySet<Session>();


    /**
     * 新的WebSocket请求开启
     */
    @OnOpen
    public void onOpen(Session session) {
   
   
        SessionSet.add(session);
        log.info("【WebSocket】建立WebSocket连接，【连接id】=" + session.getId());
    }


    /**
     * 接收WebSocket信息
     * 【扩展】 可以根据不同的类型分发到不同的线程做处理
     */
    @OnMessage
    public void onMessage(String message, Session session) {
   
   
        try {
   
   
        	// 启用一个线程去做python调用的工作
        	// message: 传递的参数
            taskExecutor.execute(new PythonDealThread(message, session));