SpringBoot中的WebSocket点对点发送消息

添加依赖：

  

<dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-websocket</artifactId><version>1.3.5.RELEASE</version></dependency><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-security</artifactId></dependency>

配置SpringSecurity:

package com.example.demo.part3.chapter7.Chat;import org.springframework.context.annotation.Configuration;import org.springframework.security.config.annotation.authentication.builders.AuthenticationManagerBuilder;import org.springframework.security.config.annotation.web.builders.HttpSecurity;import org.springframework.security.config.annotation.web.builders.WebSecurity;import org.springframework.security.config.annotation.web.configuration.EnableWebSecurity;import org.springframework.security.config.annotation.web.configuration.WebSecurityConfigurerAdapter;@Configuration@EnableWebSecuritypublic class WebSecurityConfig extends WebSecurityConfigurerAdapter {    @Override    protected void configure(AuthenticationManagerBuilder auth) throws Exception {        auth                .inMemoryAuthentication()                //在内存中分别配置两个用户sasa sa和密码 ，角色是user                .withUser("sasa").password("123").roles("user")                .and()                .withUser("sa").password("123").roles("user");    }    @Override    public void configure(WebSecurity web) throws Exception {        ///resources/static/ 目录下的静态资源，spring security不拦截        web.ignoring().antMatchers("/resources/static/**");    }    @Override    protected void configure(HttpSecurity http) throws Exception {        http.authorizeRequests()                //设置spring security对 /  和  /login  路径不拦截                .antMatchers("/","/login").permitAll()                .anyRequest().authenticated()                .and()                .formLogin()                //设置spring security的登录页面访问路径为 /login                .loginPage("/login")                //登陆成功后转向 /chat 路径                .defaultSuccessUrl("/chat")                .permitAll()                .and()                .logout()                .permitAll();    }}

配置WebSocket:

package com.example.demo.part3.chapter7.Chat;import org.springframework.context.annotation.Configuration;import org.springframework.messaging.simp.config.MessageBrokerRegistry;import org.springframework.web.socket.config.annotation.AbstractWebSocketMessageBrokerConfigurer;import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;import org.springframework.web.socket.config.annotation.StompEndpointRegistry;@Configuration@EnableWebSocketMessageBrokerpublic class WebSocketConfig extends AbstractWebSocketMessageBrokerConfigurer{    @Override    public void configureMessageBroker(MessageBrokerRegistry registry) {        //点对点式应增加一个/queue消息代理        registry.enableSimpleBroker("/queue","/topic");    }    @Override    public void registerStompEndpoints(StompEndpointRegistry endpoint) {        //注册一个名为 endPointChat 的endpoint        endpoint.addEndpoint("/endPointChat").withSockJS();    }}

控制器：

package com.example.demo.part3.chapter7.Chat;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.messaging.handler.annotation.MessageMapping;import org.springframework.messaging.simp.SimpMessagingTemplate;import org.springframework.stereotype.Controller;import java.security.Principal;@Controllerpublic class ChatController {    @Autowired    //通过simpleMessagingTemplate向浏览器发送消息    private SimpMessagingTemplate messagingTemplate;    @MessageMapping("/chat")    //在springmvc中，可以直接在参数中获取principle,principle中包含当前用户的信息    public  void handleChat(Principal principal,String msg){        System.out.println(1);        if(principal.getName().equals("sasa")){            //通过 convertAndSendToUser方法向用户发送信息，第一个参数是消息接收人，            //第二个是浏览器订阅地址，第三个是消息本身            System.out.println(2);            messagingTemplate.convertAndSendToUser("sa",                    "/queue/notifications",principal.getName()                    +"-send:"+msg);        }else{            System.out.println(3);            messagingTemplate.convertAndSendToUser("sasa",                    "/queue/notifications",principal.getName()                    +"-send:"+msg);        }    }}

登录页面：

<!DOCTYPE html><html xmlns="http://www.w3.org/1999/xhtml" xmlns:th="http://www.thymeleaf.org"      xmlns:sec="http://www.thymeleaf.org/thymeleaf-extras-springsecurity3"><meta charset="UTF-8"/><head>    <title>登录页面</title></head><body>   <div th:if="${param.error}">无效的账号和密码</div>   <div th:if="${param.logout}">你已注销</div>   <form th:action="@{/login}" method="post">       <div>           <label>               账号:<input type="text" name="username"/>           </label>       </div>       <div>           <label>               密码:<input type="password" name="password"/>           </label>       </div>       <div>           <input type="submit" name="登录"/>       </div>   </form></body></html>

聊天页面：

<!DOCTYPE html><html xmlns="http://www.w3.org/1999/xhtml" xmlns:th="http://www.thymeleaf.org"      xmlns:sec="http://www.thymeleaf.org/thymeleaf-extras-springsecurity3"><meta charset="UTF-8"/><head>    <title>聊天框</title>    <script th:src="@{sockjs.min.js}"></script>    <script th:src="@{stomp.min.js}"></script>    <script th:src="@{jquery-1.8.3.min.js}"></script></head><body>   <form id="chatForm">       <textarea rows="4" cols="60" name="text" id="text" />       <input type="submit"/>   </form>   <script type="text/javascript">       $("#chatForm").submit(function (e) {           e.preventDefault();           var text=$("#text").val();           sendMsg(text);       });       //连接endpoint名称为 /endPointChat 的endpoint       var sock=new SockJS("/endPointChat");       var stomp=Stomp.over(sock);       stomp.connect('guest','guest',function (frame) {           //订阅 /user/queue/notifications 发送的消息，这里与           //在控制器的messagingTemplate.convertAndSendToUser中定义的订阅地址保持一致           //这里多了 /user ，并且这个 /user是必须的，使用了 /user 才会将消息发送到指定用户           stomp.subscribe("/user/queue/notifications",handleNotification);       });       function handleNotification(message) {           console.log("msg"+message);           $("#output").append('<b>'+message.body+'</b>');       }       function sendMsg(text) {           stomp.send("/chat",{},text);       }       $("#stop").click(function () {           sock.close();       })   </script> <div id="output"></div></body></html>

页面控制：

package com.example.demo.part3.chapter7.WebSocket;import org.springframework.context.annotation.Configuration;import org.springframework.web.servlet.config.annotation.ViewControllerRegistry;import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;@Configurationpublic class WebMvcConfig extends WebMvcConfigurerAdapter{    @Override    public void addViewControllers(ViewControllerRegistry registry) {        //super.addViewControllers(registry);        registry.addViewController("/login").setViewName("/login");        registry.addViewController("/chat").setViewName("/chat");    }}

效果展示：