上个小节我们已经可以成功付款，并且后端项目的Web方法接收到了付款成功的通知，罚款记录的状态从“未付款”变成了“已付款”。这节课我们要给SpringBoot配置上WebSocket，这样将来才可以推送付款成功的结果给前端页面。

一、缓存WebSocket连接

WebSocket类似于Servlet，需要我们声明一些生命周期函数，而且这些函数还都不是我们能主动调用的。

@ServerEndpoint(value = "/mysocket")
@Component
public class MyWebSockertService{
    @OnOpen
    public void onOpen(Session session) {
        //创建WebSocket连接时候执行
    }

    @OnClose
    public void onClose(Session session) {
        //管理WebSocket连接时候执行
    }

    @OnMessage
    public void onMessage(String message, Session session) {
       //接收消息时候执行
    }

    @OnError
    public void onError(Session session, Throwable error) {
        //发生错误时候执行
    }
    
}

向客户端发送消息，需要使用Session对象。但是这些生命周期函数都由于客户端某种操作，而触发执行的。如果客户端不触发操作，那么后端是无法主动给客户端发送消息的。所以我们要把Session对象缓存起来。需要的时候，我们提取缓存的Session，主动向客户端发送消息。

因为后端的WebSocket服务类是多例的，所以我们想要全局共享缓存，要么用Redis，要么声明静态的HashMap对象。如果选用Redis，那么保存Session对象要用到序列化，会消耗一定的时间，所以不建议使用。如果全局共享使用HashMap，又会存在并发读写的问题，最终我们选择ConcurrentHashMap类。

二、配置WebSocket

在com.example.emos.api.config包中创建WebSocketConfig.java类，用于给SpringBoot添加WebSocket功能。

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

创建com.example.emos.api.websocket包，定义WebSocketService.java类，用于负责处理WebSocket连接。这个类是多例的，每次有客户端用WebSocket连接这个类，Spring就会创建一个新的WebSocketService对象，所以大家不用担心线程安全的问题。

因为WebSocket是长连接协议，它完全异于Http协议，所以我们不能用处理Http请求和响应的想法去看待WebSocket连接。例如我们在Http协议上很轻松可以传递Cookie数据，但是WebSocket不支持Cookie，所以我们要自己把Token字符串上传给服务端，然后服务端不能用StpUtil.getUserIdAsInt()获取到UserId，我们要自己从Token字符串中提取UserId出来。

在WebSocket中，我们要约定跟客户端传递数据的格式。为了能让数据看起来格式规整，我采用传递JSON字符串的方式。

@Slf4j
@ServerEndpoint(value = "/socket")
@Component
public class WebSocketService {

    //用于保存WebSocket连接对象
    public static ConcurrentHashMap<String, Session> sessionMap = new ConcurrentHashMap<>();

    /**
     * 连接建立成功调用的方法
     */
    @OnOpen
    public void onOpen(Session session) {
    }

    /**
     * 连接关闭调用的方法
     */
    @OnClose
    public void onClose(Session session) {
        Map map = session.getUserProperties();
        if (map.containsKey("userId")) {
            String userId = MapUtil.getStr(map, "userId");
            sessionMap.remove(userId);
        }
    }

    /**
     * 接收消息
     *
     * @param message
     * @param session
     */
    @OnMessage
    public void onMessage(String message, Session session) {
        //把字符串转换成JSON
        JSONObject json = JSONUtil.parseObj(message);
        String opt = json.getStr("opt");
        if("ping".equals(opt)){
            return;
        }
        //从JSON中取出Token
        String token = json.getStr("token");
        //从Token取出userId
        String userId = StpUtil.stpLogic.getLoginIdByToken(token).toString();
        
        //取出Session绑定的属性
        Map map = session.getUserProperties();
        //如果没有userId属性，就给Session绑定userId属性，关闭连接的时候会用到
        if (!map.containsKey("userId")) {
            map.put("userId", userId);
        }
        //把Session缓存起来
        if (sessionMap.containsKey(userId)) {
            //替换缓存中的Session
            sessionMap.replace(userId, session);
        } else {
            //向缓存添加Session
            sessionMap.put(userId, session);
        }
        sendInfo("ok",userId);
        
    }

    @OnError
    public void onError(Session session, Throwable error) {
        log.error("发生错误", error);
    }
    
    /**
     * 发送消息给客户端
     */
    public static void sendInfo(String message, String userId) {
        if (StrUtil.isNotBlank(userId) && sessionMap.containsKey(userId)) {
            //从缓存中查找到Session对象
            Session session = sessionMap.get(userId);
            //发送消息
            sendMessage(message, session);
        }
    }

    /**
     * 封装发送消息给客户端
     */
    private static void sendMessage(String message, Session session) {
        try {
            session.getBasicRemote().sendText(message);
        } catch (Exception e) {
            log.error("执行异常", e);
        }
    }
}