springboot websocket

本文使用stomp
STOMP(面向简单文本的消息传递协议)最初是为脚本语言(例如Ruby,Python和Perl)创建的,以连接到企业消息代理。它旨在解决常用消息传递模式的最小子集。STOMP可以在任何可靠的双向流网络协议上使用,例如TCP和WebSocket。尽管STOMP是面向文本的协议,但是消息有效负载可以是文本或二进制。
客户端可以使用SEND或SUBSCRIBE命令来发送或订阅消息,以及destination描述消息的内容和应由谁接收的标头。这启用了一种简单的发布-订阅机制,您可以使用该机制通过代理将消息发送到其他连接的客户端,或者将消息发送到服务器以请求执行某些工作。
当您使用Spring的STOMP支持时,Spring WebSocket应用程序将充当客户端的STOMP代理。消息被路由到@Controller消息处理方法或简单的内存中代理,该代理跟踪订阅并向订阅的用户广播消息。您还可以将Spring配置为与专用的STOMP代理(例如RabbitMQ,ActiveMQ等)一起使用,以实际广播消息。在那种情况下,Spring维护与代理的TCP连接,将消息中继到该代理,并将消息从该代理向下传递到已连接的WebSocket客户端。因此,Spring Web应用程序可以依靠基于HTTP的统一安全性,通用验证以及用于消息处理的熟悉的编程模型。
–摘自官网

  1. 依赖
    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-websocket</artifactId>
        </dependency>
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
        </dependency>
    </dependencies>
  1. 定义一个连接校验,用于记录用户的信息
/**
 * 连接时校验用户信息,并返回重写的Principal
 */
@Component
public class MyHandshakeHandler extends DefaultHandshakeHandler {

    @Override
    protected Principal determineUser(ServerHttpRequest request, WebSocketHandler wsHandler, Map<String, Object> attributes) {
        if (!(request instanceof ServletServerHttpRequest)) {
            return null;
        }
        ServletServerHttpRequest req = (ServletServerHttpRequest) request;
        //获取请求参数中携带的uid
        String uid = req.getServletRequest().getParameter("uid");
        if(uid == null){
            throw new RuntimeException("未登录");
        }
        return new MyPrincipal(uid);
    }
}
  1. 定义一个记录用户登陆退出,方便观察
/**
 * 用户登录退出操作
 */
@Component
public class MyWebSocketHandler implements WebSocketHandlerDecoratorFactory {

    @Override
    public WebSocketHandler decorate(WebSocketHandler handler) {
        return new WebSocketHandlerDecorator(handler) {
            //用户登录
            @Override
            public void afterConnectionEstablished(WebSocketSession session) throws Exception {
                String uid = session.getPrincipal().getName();
                System.out.println(uid + "登陆");
                super.afterConnectionEstablished(session);
            }

            //用户退出
            @Override
            public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {
                String uid = session.getPrincipal().getName();
                System.out.println(uid + "退出");

                super.afterConnectionClosed(session, closeStatus);
            }
        };
    }
}
  1. websocket配置
@Configuration
//开启消息代理,默认使用内置消息代理,也可以选择配置RabbitMQ等
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
    @Autowired
    private MyWebSocketHandler myWebSocketHandler;
    @Autowired
    private MyHandshakeHandler myHandshakeHandler;

    @Override
    public void configureMessageBroker(MessageBrokerRegistry registry) {
        //启用/user /topic两个消息前缀,消息发送的前缀,也是前端订阅的前缀
        registry.enableSimpleBroker("/user", "/topic");
        //当使用convertAndSendToUser发送消息时,前端订阅用/user开头。即一对一发送消息,使用/user为前缀订阅
        registry.setUserDestinationPrefix("/user");
        //前端向服务端发送消息的前缀
        registry.setApplicationDestinationPrefixes("/im/");
    }

    @Override
    public void registerStompEndpoints(StompEndpointRegistry stompEndpointRegistry) {
        //客户端和服务端进行连接的endpoint
        //如果使用移动端开发app,需要/im/conn/websocket连接
        stompEndpointRegistry.addEndpoint("/im/conn")
                .setHandshakeHandler(myHandshakeHandler)//设置连接校验
                .setAllowedOrigins("*")//跨域
                .withSockJS();//开启sockjs
    }

    @Override
    public void configureWebSocketTransport(WebSocketTransportRegistration registry) {
        //注册登陆退出
        registry.addDecoratorFactory(myWebSocketHandler);
    }
}
  1. 创建一个消息的实体类
/**
 * 一对一发送的消息
 */
@Data
public class SendMsg implements Serializable {
    //发送消息的用户id
    private String uid;

    //接收消息的用户id
    @NotNull(message = "未选择用户")
    private String toUid;

    //发送的文本消息
    @NotNull(message = "消息不能为空")
    private String content;
}
  1. 创建controller
@RestController
public class ImController {
    //发送消息的模板
    @Autowired
    private SimpMessagingTemplate simpMessagingTemplate;

    /**
     * 发送消息,一对一
     * Principal为连接websocket校验时返回的,可以直接在参数中使用
     * 也可以使用@Validated校验参数的合法性
     * 
     * @param msg
     * @param principal
     * @return
     */
    @MessageMapping("/send2user")
    public String send2user(@Validated SendMsg msg, Principal principal) {
    	//获取用户的uid
        String uid = principal.getName();
        //设置发送信息的uid
        msg.setUid(uid);
        System.out.println(uid + ":" + msg);
        //发送给订阅/user/{toUid}/msg的用户
        //这里的toUid是接收消息用户的uid
        simpMessagingTemplate.convertAndSendToUser(msg.getToUid(), "msg", msg);
        return "success";
    }

    /**
     * 发送消息,发送给所有订阅/topic/sys的用户
     * 也可以使用@SendTo注解,返回值为发送的消息即可
     *
     * @param msg
     * @return
     */
    @GetMapping("/sendAll")
//    @SendTo("/topic/sys")
    public String sendAll(String msg){
        System.out.println("广播消息:" + msg);
        //如果是群聊,根据传递参数的群聊房间号,动态拼接/topic/{房间号},前端订阅/topic/{房间号}即可
        simpMessagingTemplate.convertAndSend("/topic/sys", msg);
        return "success";
//        return msg;
    }
    
}
  1. html页面
<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
    <script src="https://cdn.staticfile.org/jquery/1.10.2/jquery.min.js"></script>
    <!-- sockjs stomp -->
    <script src="https://cdn.bootcss.com/sockjs-client/1.1.4/sockjs.min.js"></script>
    <script src="https://cdn.bootcss.com/stomp.js/2.3.3/stomp.min.js"></script>
    <script>
        let stompClient = null;
        //连接websocket
        function conn() {
            let uid = $("#uidInput").val();
            //可以在后面直接拼接参数,在MyHandshakeHandler中校验用户uid
            let socket = new SockJS('http://localhost:8080/im/conn?uid=' + uid);
            stompClient = Stomp.over(socket);
            stompClient.connect({}, function () {
                //订阅/user/{uid}/msg这个地址,接收往这个地址发送的消息
                stompClient.subscribe('/user/' + uid + '/msg', function (msg) {
                    //后台返回的信息是实体类,将其转换成json,添加到ul中
                    let msgBody = JSON.parse(msg.body);
                    $("#userMsg").append("<li>" + msgBody.uid +":"+ msgBody.content + "</li>")
                });
                //订阅/topic/sys这个地址,接收往这个地址发送的消息
                stompClient.subscribe('/topic/sys', function (msg) {
                    $("#sysMsg").append("<li>" + msg.body + "</li>")
                });
                //stompClient.subscribe()....多个订阅地址,也可以在外面定义。前提是stompClient已经连接

                //隐藏连接div
                $("#connDiv").hide();
                //显示消息div
                $("#msgDiv").show();
            }, function (err) {
                console.log(err);
            });
        }

        //发送一对一消息
        function send() {
            let content = $("#content").val();
            if (!content) {
                alert("请输入消息");
            }
            let toUid = $("#toUid").val();
            if (!toUid) {
                alert("请输入发送给用户的uid");
            }
            let msg = {"content": content, "toUid": toUid};
            if (!stompClient) {
                alert("未连接");
            }
            //前端发送消息以/im开头,往send2user中发送消息,消息为JSON.stringify(msg)
            stompClient.send("/im/send2user", {}, JSON.stringify(msg));
        }
    </script>
</head>
<body>
<div id="connDiv">
    <input type="text" id="uidInput" placeholder="请输入uid">
    <button onclick="conn()">连接</button>
</div>
<div id="msgDiv" style="display: none">
    <input id="content" type="text" placeholder="消息内容"/>
    <input id="toUid" type="text" placeholder="发送给用户的uid">
    <button onclick="send()">发送</button>
    <br/>
    <label>用户消息:</label>
    <ul id="userMsg"></ul>
    <label>系统消息:</label>
    <ul id="sysMsg"></ul>
</div>
</body>
</html>
  1. 测试

浏览器访问http://localhost:8080/index.html
开启两个页面,分别输入两个不同的uid,这里使用111,222
控制台:

111登陆
222登陆

从111的页面发送消息给222,并从222的页面发送消息给111
控制台:

111:SendMsg(uid=111, toUid=222, content=在吗?)
222:SendMsg(uid=222, toUid=111, content=在的)

从浏览器访问sendAll接口,发送消息给所有订阅/topic/sys的用户
http://localhost:8080/sendAll?msg=大家晚上好
控制台:

广播消息:大家晚上好

示例图:
在这里插入图片描述

项目地址


作者博客

作者公众号
在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值