Springboot整合websocket代码实现

1.启动类加注解

2.编写websocket配置类

package com.els.config;

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.server.standard.ServerEndpointExporter;

/**
 * @Author: pengsiwen
 * @Date: 2024-07-13
 * @Description
 */
@Configuration
public class WebSocketConfig {
    public WebSocketConfig() {
    }

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

3.业务实现

package com.els.config;

import cn.hutool.core.lang.UUID;
import com.alibaba.fastjson.JSON;
import com.els.common.util.S2cloudUtil;
import com.els.common.util.SpringUtil;
import com.els.modules.system.model.User;
import com.els.modules.system.service.UserService;
import com.els.modules.tender.common.S2FormEnum;
import com.els.modules.tender.model.TenderOnlineWebSocketVO;
import com.els.modules.tender.model.TenderOpenInfoRecords;
import com.els.modules.tender.service.TenderOpenInfoRecordsService;
import com.els.modules.tender.utils.FastJsonUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;

import javax.websocket.OnClose;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.stream.Collectors;

/**
 * @Author: pengsiwen
 * @Date: 2024-07-11
 * @Description
 */
@Slf4j
@Component
@ServerEndpoint("/websocket/tender/online/{subpackageId}/{userId}")
public class TenderOnlineWebSocket {

    private Session session;

    private String subpackageId;

    private String userId;

    private String id;

    private static Map<String, List<TenderOnlineWebSocket>> clientMap = new ConcurrentHashMap<>();

    @OnOpen
    public void onOpen(Session session, @PathParam(value = "subpackageId") String subpackageId, @PathParam(value = "userId") String userId) {
        try {
            this.session = session;
            this.subpackageId = subpackageId;
            this.userId = userId;
            this.id = UUID.randomUUID().toString();
            List<TenderOnlineWebSocket> clients = clientMap.get(subpackageId) == null ? new ArrayList<>() : clientMap.get(subpackageId);
            log.info("账号{},建立连接", userId);
            clients.add(this);
            clientMap.put(subpackageId, clients);
        } catch (Exception e) {
            log.error("【TenderOnlineWebSocket消息】建立连接异常", e);
        }
    }

    @OnClose
    public void onClose() {
        try {
            List<TenderOnlineWebSocket> clients = clientMap.get(this.subpackageId);
            if (clients != null) {
                List<TenderOnlineWebSocket> result = clients.stream().filter(socket -> !this.id.equals(socket.id)).collect(Collectors.toList());
                clientMap.put(this.subpackageId, result);
            }
            log.info("【TenderOnlineWebSocket消息】连接断开");
        } catch (Exception e) {
            log.error("【TenderOnlineWebSocket消息】断开连接异常", e);
        }
    }

    @OnMessage
    public void onMessage(String message) {
        try {
            if ("heartCheck".equals(message)) {
                log.info("收到{}心跳信息！", session.getRequestURI().toString());
                session.getAsyncRemote().sendText(message);
                return;
            }
            log.info("收到消息：{}", message);
            TenderOnlineWebSocketVO tenderOnlineWebSocketVO = null;
            try {
                tenderOnlineWebSocketVO = FastJsonUtils.toBean(message, TenderOnlineWebSocketVO.class);
            } catch (Exception e) {
                log.error("消息格式不对：{},消息内容：{}", e.getMessage(), message);
                session.getAsyncRemote().sendText(message);
                return;
            }
            //获取当前登录人
            log.info("当前人员ID：{}", userId);
            UserService userService = SpringUtil.getBean(UserService.class);
            User currentUserInfo = userService.getById(userId);
            // 查询当前账号是否已签到
            TenderOpenInfoRecordsService tenderOpenInfoRecordsService = SpringUtil.getBean(TenderOpenInfoRecordsService.class);
            List<TenderOpenInfoRecords> projectOpenInfoRecordsList =
                    tenderOpenInfoRecordsService.queryOpenInfoRecordsSignMessageBySubpackageId(subpackageId, currentUserInfo.getSid());
            if (!projectOpenInfoRecordsList.isEmpty()) {
                // 不需重复记录签到信息
                TenderOnlineWebSocketVO finalTenderOnlineWebSocketVO = tenderOnlineWebSocketVO;
                long count = projectOpenInfoRecordsList.parallelStream()
                        .filter(projectOpenInfoRecords -> finalTenderOnlineWebSocketVO.getContent().equals(projectOpenInfoRecords.getContent()))
                        .count();
                if (count > 0) {
                    session.getAsyncRemote().sendText("heartCheck");
                    return;
                }
            }
            // 保存日志
            TenderOpenInfoRecords tenderOpenInfoRecords = new TenderOpenInfoRecords();
            S2cloudUtil.copySysField(currentUserInfo, tenderOpenInfoRecords);
            tenderOpenInfoRecords.setSheetCode(S2FormEnum.TENDER_OPEN_INFO_RECORDS.getFormCode());
            tenderOpenInfoRecords.setElsAccount(currentUserInfo.getSid());
            tenderOpenInfoRecords.setTenderProject(tenderOnlineWebSocketVO.getTenderProject());
            tenderOpenInfoRecords.setTenderProjectId(tenderOnlineWebSocketVO.getTenderProjectId());
            tenderOpenInfoRecords.setSubpackageId(tenderOnlineWebSocketVO.getSubpackageId());
            tenderOpenInfoRecords.setSubpackage(tenderOnlineWebSocketVO.getSubpackage());
            tenderOpenInfoRecords.setContent(tenderOnlineWebSocketVO.getContent());
            tenderOpenInfoRecords.setCheckType(tenderOnlineWebSocketVO.getCheckType());
            tenderOpenInfoRecords.setCheckTypeId(tenderOnlineWebSocketVO.getCheckTypeId());
            tenderOpenInfoRecords.setProcessType(tenderOnlineWebSocketVO.getProcessType());
            tenderOpenInfoRecords.setProcessTypeId(tenderOnlineWebSocketVO.getProcessTypeId());
            tenderOpenInfoRecordsService.save(tenderOpenInfoRecords);
            // 发送消息
            sendQuoteMessage(subpackageId, JSON.toJSONString(tenderOpenInfoRecords));
        } catch (Exception e) {
            log.error("消息异常：{}", e);
        }
    }

    public void sendQuoteMessage(String subpackageId, String messageContent) {
        List<TenderOnlineWebSocket> clients = clientMap.get(subpackageId);
        if (clients != null && !clients.isEmpty()) {
            for (TenderOnlineWebSocket socket : clients) {
                if (socket.session != null && socket.session.isOpen()) {
                    log.info("给{}-发送消息-{}", socket.userId, messageContent);
                    socket.session.getAsyncRemote().sendText(messageContent);
                }
            }
        }
    }
}

消息实体类

package com.els.modules.tender.model;

import lombok.Data;

import java.io.Serializable;

/**
 * Copyright (C) 2022 All Rights Reserved.
 *
 * @description:
 * @author: yn
 * @create: 2022-04-26
 **/
@Data
public class TenderOnlineWebSocketVO implements Serializable {
    //审查方式
    private String checkType;

    private String checkTypeId;

    //招标流程
    private String processType;

    private String processTypeId;

    //当前哪一步
    private String currentStep;

    private String currentStepId;

    //分包名称
    private String subpackage;

    private String subpackageId;

    //项目名称
    private String tenderProject;

    private String tenderProjectId;

    //消息内容
    private String content;
}