SpringBoot集成websocket（3）|（websocket调用websocket采用回调方式实现数据互传）

Oak科技2023-08-08 18:47

SpringBoot集成websocket（3）|（websocket调用websocket采用回调方式实现数据互传）

### 文章目录

  • [SpringBoot集成websocket（3）|（websocket调用websocket采用回调方式实现数据互传）](#文章目录 SpringBoot集成websocket（3）|（websocket调用websocket采用回调方式实现数据互传） @[TOC] 前言 一、websocket服务端依赖引入 二、websocket服务代码实现 1.WebSocketConfig配置 2.WebSocketServer服务实现 3.ChatClient4Chat连接工具实现 3.WebSocketClient连接第三方客户端实现 总结)
  • [@[TOC]](#文章目录 SpringBoot集成websocket（3）|（websocket调用websocket采用回调方式实现数据互传） @[TOC] 前言 一、websocket服务端依赖引入 二、websocket服务代码实现 1.WebSocketConfig配置 2.WebSocketServer服务实现 3.ChatClient4Chat连接工具实现 3.WebSocketClient连接第三方客户端实现 总结)
  • [前言](#文章目录 SpringBoot集成websocket（3）|（websocket调用websocket采用回调方式实现数据互传） @[TOC] 前言 一、websocket服务端依赖引入 二、websocket服务代码实现 1.WebSocketConfig配置 2.WebSocketServer服务实现 3.ChatClient4Chat连接工具实现 3.WebSocketClient连接第三方客户端实现 总结)
  • [一、websocket服务端依赖引入](#文章目录 SpringBoot集成websocket（3）|（websocket调用websocket采用回调方式实现数据互传） @[TOC] 前言 一、websocket服务端依赖引入 二、websocket服务代码实现 1.WebSocketConfig配置 2.WebSocketServer服务实现 3.ChatClient4Chat连接工具实现 3.WebSocketClient连接第三方客户端实现 总结)
  • [二、websocket服务代码实现](#文章目录 SpringBoot集成websocket（3）|（websocket调用websocket采用回调方式实现数据互传） @[TOC] 前言 一、websocket服务端依赖引入 二、websocket服务代码实现 1.WebSocketConfig配置 2.WebSocketServer服务实现 3.ChatClient4Chat连接工具实现 3.WebSocketClient连接第三方客户端实现 总结)
  • [1.WebSocketConfig配置](#文章目录 SpringBoot集成websocket（3）|（websocket调用websocket采用回调方式实现数据互传） @[TOC] 前言 一、websocket服务端依赖引入 二、websocket服务代码实现 1.WebSocketConfig配置 2.WebSocketServer服务实现 3.ChatClient4Chat连接工具实现 3.WebSocketClient连接第三方客户端实现 总结)
  • [2.WebSocketServer服务实现](#文章目录 SpringBoot集成websocket（3）|（websocket调用websocket采用回调方式实现数据互传） @[TOC] 前言 一、websocket服务端依赖引入 二、websocket服务代码实现 1.WebSocketConfig配置 2.WebSocketServer服务实现 3.ChatClient4Chat连接工具实现 3.WebSocketClient连接第三方客户端实现 总结)
  • [3.ChatClient4Chat连接工具实现](#文章目录 SpringBoot集成websocket（3）|（websocket调用websocket采用回调方式实现数据互传） @[TOC] 前言 一、websocket服务端依赖引入 二、websocket服务代码实现 1.WebSocketConfig配置 2.WebSocketServer服务实现 3.ChatClient4Chat连接工具实现 3.WebSocketClient连接第三方客户端实现 总结)
  • [3.WebSocketClient连接第三方客户端实现](#文章目录 SpringBoot集成websocket（3）|（websocket调用websocket采用回调方式实现数据互传） @[TOC] 前言 一、websocket服务端依赖引入 二、websocket服务代码实现 1.WebSocketConfig配置 2.WebSocketServer服务实现 3.ChatClient4Chat连接工具实现 3.WebSocketClient连接第三方客户端实现 总结)
  • [总结](#文章目录 SpringBoot集成websocket（3）|（websocket调用websocket采用回调方式实现数据互传） @[TOC] 前言 一、websocket服务端依赖引入 二、websocket服务代码实现 1.WebSocketConfig配置 2.WebSocketServer服务实现 3.ChatClient4Chat连接工具实现 3.WebSocketClient连接第三方客户端实现 总结)

章节

第一章链接: SpringBoot集成websocket（1）|（websocket客户端实现）

第二章链接: SpringBoot集成websocket（2）|（websocket服务端实现以及websocket中转实现）

前言

本节主要介绍的是springboot实现websocket的客户端服务端，以及客户端与服务端的数据互传。以下为伪代码，业务逻辑删除导致不能直接拷贝运行，大家可以参考其中的思路实现。

一、websocket服务端依赖引入

c 复制代码 
   <dependency>
       <groupId>org.springframework.boot</groupId>
       <artifactId>spring-boot-starter-websocket</artifactId>
   </dependency>

二、websocket服务代码实现

1.WebSocketConfig配置

springboot接入websocket需要启用对应的配置

c 复制代码 
@Configuration
@EnableWebSocket
public class WebSocketConfig {
    @Bean
    public ServerEndpointExporter serverEndpointExporter(){
        return new ServerEndpointExporter();
    }
}

2.WebSocketServer服务实现

springboot提供对外的websocket接口实现

c 复制代码 
@Component
@Data
@Slf4j
@ServerEndpoint(value = "/v1/chat")
public class DocChatServer {

    public final static String CHAT_ERR_MSG_FORMAT = "{\"header\":{\"code\":10001,\"message\":\"参数格式不对\",\"sId\":\"%s\",\"status\":2}}";

    @Autowired
    private void setOriginMessageSender() {
    // 初始化注入bean 隐藏掉了
    }


    @OnOpen
    public void OnOpen(Session session) {
        log.debug("chat websocket open ");
    }


    @OnClose
    public void OnClose() {
        log.debug("chat websocket close ");
    }

    @OnMessage
    public void OnMessage(Session session, String message) {
        SearchDocParamVo param = null;
        log.debug("ApiRequest = {}", message);
        // 参数校验
        try {
            JSONObject jsonObject = JSONObject.parseObject(message);
            // todo  参数校验
        } catch (Exception e) {
            String errMsg = String.format(CHAT_ERR_MSG_FORMAT, session.getId());
            log.error("chat请求参数格式不会：{}，异常：{}", errMsg, e);
            send(session, errMsg);
            return;
        }
        // todo 业务处理
        List<ChatRecord.Source> sources = Lists.newArrayList();;
        String prompt = "";
        // 谈话接口
        queryChat(session, param, prompt, sources);

    }


    /**
     * 执行谈话
     *
     * @param session
     * @param param
     * @param prompt
     * @param sources
     */
    private void queryChat(Session session, SearchDocParamVo param, String prompt, List<ChatRecord.Source> sources) {
       // todo 业务处理 。。。
        // 消息发送
        try {
            boolean b = this.sendChatMessage(session, param, sources, texts);
            if (!b) {
                List<Text> textsTry = Lists.newArrayList();
                Text build1 = Text.builder()
                        .role("user")
                        .content("请更具自己的理解回答问题：" + param.getContent())
                        .build();
                textsTry.add(build1);
                this.sendChatMessage(session, param, sources, textsTry);
            }
        } catch (Exception e) {
            log.error("发送消息异常：{}", e.getMessage());
        }
    }

    /**
     * 收到谈话响应数据处理
     *
     * @param session
     * @param param
     * @param sources
     * @param texts
     * @return
     */
    private boolean sendChatMessage(Session session, SearchDocParamVo param, List<ChatRecord.Source> sources, List<Text> texts) {
        ChatClient4Chat planetClient4Chat = new ChatClient4Chat(websocketConfigConst);
        try {
            planetClient4Chat.send(param, texts, new ApiResponseObserver() {
                public void onReceive(String message) {
                    // 收到远程websocket服务响应的数据
                }

                public void onError(Throwable throwable) {
                    log.error("收到错误：{}", throwable);

                }

                public void onCompleted() {
                    log.error("收到结束");
                }
            });
            // 以下是业务逻辑   可忽略
            for (int i = 0; i < 100; i++) {
                if (planetClient4Chat.isHasCheck()) {
                    log.debug("has check");
                    return planetClient4Chat.isSuccess();
                } else {
                    Thread.sleep(500);
                }
            }

        } catch (Exception e) {
            log.error("发送消息异常：{}", e.getMessage());
        }
        return true;
    }


    public void send(Session session, String msg) {
        synchronized (session) {
            if (!session.isOpen()) {
                log.error("客户端连接关闭，数据不发送：{}", msg);
                return;
            }
            try {
                session.getBasicRemote().sendText(msg);
            } catch (IOException ex) {
                log.error("传递消息给客户端异常：{}", ex.getMessage());
            }
        }
    }

    public int getStatus(String message) {
        int status = -1;
        try {
			//todo 业务逻辑
            return choices.getStatus();
        } catch (Exception e) {
            log.error("数据中提取status异常：{}", e);
        }
        return status;
    }




    @OnError
    public void onerror(Session session, Throwable throwable) {
        log.error("chat连接异常关闭:远程主机强迫关闭了一个现有的连接：{}", throwable);
    }



}

3.ChatClient4Chat连接工具实现

springboot提供对中间衔接工具了，连接第三饭websocket接口

实现代码如下

c 复制代码 
@Slf4j
@Getter
@Setter
public class ChatClient4Chat {
    private static WebsocketConfigConst websocketConfigConst;
    private StringBuilder stringBuilder;
    private boolean hasCheck;
    private boolean success;
    private Queue<String> queue;
    private ChatChatServer sparkChatServer;

    ChatClient4Chat(WebsocketConfigConst websocketConfigConst) {
        this.websocketConfigConst = websocketConfigConst;
        this.stringBuilder = new StringBuilder();
        this.hasCheck = false;
        this.success = true;
        this.queue = new LinkedList<String>();

    }

    /**
     * 执行聊天
     *
     * @param param
     */
    public void send(SearchDocParamVo param, List<Text> texts, ApiResponseObserver apiResponseObserver) {
        // 获取连接
        ChatChatServer chatServer = (ChatChatServer ) getWebSocketClient(apiResponseObserver);
        if (chatServer != null && chatServer.isOpen()) {
            this.sparkChatServer = chatServer;
            // 消息发送
            try {
                chatServer.send(SparkHand.initParam(param, texts, websocketConfigConst.type, websocketConfigConst.appid, websocketConfigConst.token));
            } catch (Exception e) {
                log.error("发送消息异常：{}", e.getMessage());
            }
        } else {
            log.error("接口连接未打开");
        }
    }

    public void close() {
        // 获取连接
        if (sparkChatServer != null && sparkChatServer.isOpen()) {
            sparkChatServer.close();
        } else {
            log.error("接口连接未打开，关闭异常");
        }
    }


    private void waitConnect() {
        try {
            Thread.sleep(50);
        } catch (InterruptedException e) {
            log.error("等待连接异常");
        }
    }

    private WebSocketClient getWebSocketClient(ApiResponseObserver apiResponseObserver) {
        WebSocketClient client = new SparkChatServer(websocketConfigConst.chaturl, apiResponseObserver);
        client.connect();
        waitConnect();
        return client;
    }
}

3.WebSocketClient连接第三方客户端实现

springboot提供对第三方websocket连接的客户端

实现代码如下

c 复制代码 
@Slf4j
public class SparkChatServer extends WebSocketClient {
    private ApiResponseObserver apiResponseObserver;

    public SparkChatServer(URI serverUri, ApiResponseObserver apiResponseObserver) {
        super(serverUri);
        this.apiResponseObserver = apiResponseObserver;
    }

    @Override
    public void onOpen(ServerHandshake serverHandshake) {
        log.debug("chat 服务连接成功");
    }

    @Override
    public void onMessage(String message) {
        log.debug("收到chat数据{}", message);
        apiResponseObserver.onReceive(message);

    }

    @Override
    public void onClose(int i, String s, boolean b) {
        log.debug("退出chat连接");
    }

    @Override
    public void onError(Exception e) {
        log.error("chat连接出现异常：{}", e);

    }
}

总结

本文主要介绍websocket客户端、服务端的实现，同时通过连接工具中转websocket请求参数，捉到实时同步，以及数据收集。代码为伪代码，删除了实际使用当中的业务逻辑，介绍的是实现实录，大家可以参考。

第一章链接: SpringBoot集成websocket（1）|（websocket客户端实现）

第二章链接: SpringBoot集成websocket（2）|（websocket服务端实现以及websocket中转实现）

相关推荐
37手游后端团队1 分钟前
AI祛魅：从“计算24点”看大模型的局限性与真实能力
人工智能·后端·openai
武子康5 分钟前
大数据-11-MapReduce JOIN 操作的Java实现 Driver Mapper Reducer具体实现逻辑 模拟SQL进行联表操作
大数据·后端
知其然亦知其所以然5 分钟前
只会写 Mapper 就想进大厂？MyBatis 原理你真懂了吗？
java·后端·面试
礼貌而已9 分钟前
Vue3项目与桌面端（C++）通过Websocket 对接接口方案实现
开发语言·c++·websocket
rit843249911 分钟前
Odoo 17 在线聊天报错 “Couldn‘t bind the websocket...“ 的解决方案
网络·websocket·网络协议
bing_15813 分钟前
Spring Boot 项目中Http 请求如何对响应体进行压缩
spring boot·后端·http
猩猩程序员17 分钟前
巅峰对决：JetBrains 深度剖析 Rust vs Go 之战
后端
Cache技术分享19 分钟前
104. Java 继承 - 实例方法的覆盖
前端·后端
Lx35221 分钟前
SELECT * 的性能陷阱与列筛选优化
后端·sql·mysql
程序员岳焱22 分钟前
Java 注释规范的 6 个 “信息缺失坑” 与完整注释方案
后端
热门推荐
01Coze扣子平台完整体验和实践（附国内和国际版对比）02【图像处理与机器视觉】XJTU期末考点03KGG转MP3工具|非KGM文件|解密音频04坚果投影仪J10如何用苹果Siri开关机并和米家联动05海康Visionmaster-常见问题排查方法-启动阶段06YOLOv8入门 | 重要性能衡量指标、训练结果评价及分析及影响mAP的因素【发论文关注的指标】07【云原生】动态资源分配（DRA）深度洞察报告08从零安装 LLaMA-Factory 微调 Qwen 大模型成功及所有的坑09【SpeedAI科研小助手】2分钟极速解决知网维普重复率、AIGC率过高，一键全文降！文件格式不变，公式都保留的！10Anaconda下的pkgs占用空间13G，如何安全的清理（已解决）