spring websocket实现极简群聊

项目地址:https://gitee.com/nurxat18903/websocket1.0

前端使用sockjs.js和stomp.js
服务端spring websocket
websocket模型图如下:

服务器主要分三个节点:连接点,接收点,和发送点
1.连接点:使用websocket协议握手建立连接
2.接收点:接收来自客户端发出的消息
3.发送点:也就是应答中转代理,负责将来自接收点的信息转发给客户端

客户端主要有三步
1.连接服务器连接点:通过stomp协议包装sockjs协议后与服务器握手
2.订阅服务器发送点:表示订阅服务器消息,允许服务器发消息到客户端了
3.发送到服务器接收点:有消息要发送时,调用stomp的send()方法发送到服务器接收点

按照这个模型,服务端主要写配置类和controller
配置类:

package com.project.controller;



import org.springframework.context.annotation.Configuration;
import org.springframework.messaging.Message;
import org.springframework.messaging.MessageChannel;
import org.springframework.messaging.simp.config.ChannelRegistration;
import org.springframework.messaging.simp.config.MessageBrokerRegistry;
import org.springframework.messaging.simp.stomp.StompCommand;
import org.springframework.messaging.simp.stomp.StompHeaderAccessor;
import org.springframework.messaging.support.ChannelInterceptorAdapter;
import org.springframework.messaging.support.MessageHeaderAccessor;
import org.springframework.web.socket.config.annotation.AbstractWebSocketMessageBrokerConfigurer;
import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
import org.springframework.web.socket.config.annotation.StompEndpointRegistry;

import java.util.List;
import java.util.Map;

@Configuration
@EnableWebSocketMessageBroker
public class SpringWebSocketConfig extends AbstractWebSocketMessageBrokerConfigurer {
    //订阅端点(发送点),也就是服务器给客户端发消息端点
    @Override
    public void configureMessageBroker(MessageBrokerRegistry config) {
        config.enableSimpleBroker("/socketPublish");
    }


    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        /*
         * 路径"/webSocketEndPoint"被注册为STOMP端点，对外暴露，客户端通过该路径接入WebSocket服务
         */
        //(连接点)握手端点
        registry.addEndpoint("/socketPoint").setAllowedOrigins("*").withSockJS();
    }

}

controller类:

package com.project.controller;

import com.project.pojo.Response;
import org.springframework.messaging.handler.annotation.DestinationVariable;
import org.springframework.messaging.handler.annotation.MessageMapping;
import org.springframework.messaging.handler.annotation.SendTo;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestBody;

import java.util.Map;

@Controller
public class WebSocketController {
    //(接收点),客户端发来的消息这儿接收
    @MessageMapping("/socketReqest")
    //接收的消息转发给(发送点)订阅端点,由订阅端点批发给客户端
    @SendTo("/socketPublish")
    public Response getMessage(@RequestBody Map<String,String> map){
        System.out.println(map.get("msg"));
        Response response = new Response();
        response.setMsg(map.get("msg"));
        return response;
    }
}

详细请看源码…