一:消息群发
- 创建Spring Boot项目,并添加如下倚赖
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
<dependency>
<groupId>org.webjars</groupId>
<artifactId>webjars-locator-core</artifactId>
</dependency>
<dependency>
<groupId>org.webjars</groupId>
<artifactId>sockjs-client</artifactId>
<version>1.1.2</version>
</dependency>
<dependency>
<groupId>org.webjars</groupId>
<artifactId>stomp-websocket</artifactId>
<version>2.3.3</version>
</dependency>
<dependency>
<groupId>org.webjars</groupId>
<artifactId>jquery</artifactId>
<version>3.3.1</version>
</dependency>
- 配置WebSocket
- spring
框架提供了基于websocket的STOMP支持,STOMP是一个简单的可互操作的协议,通常被用于服务器与客户端的异步消息传递
package org.sang.wschat.config;
import org.springframework.context.annotation.Configuration;
import org.springframework.messaging.simp.config.MessageBrokerRegistry;
import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;
@Configuration
@EnableWebSocketMessageBroker//开启WebSocket消息代理
//继承WebSocketMessageBrokerConfigurer进行WebSocket配置
public class WebSocketConfig
implements WebSocketMessageBrokerConfigurer {
@Override
public void configureMessageBroker(MessageBrokerRegistry config) {
//消息代理的前缀,即如果消息的前缀是/topic就会将消息转发给消息代理broker,
//然后再由消息代理将消息转发给当前连接的客户端
config.enableSimpleBroker("/topic");
//即定义了前缀为/app的路径即可通过@MessageMapping注解的方法进行处理
config.setApplicationDestinationPrefixes("/app");
}
@Override
public void registerStompEndpoints(StompEndpointRegistry registry) {
//定义一个前缀为/chat的endPoint,并开启SockJS支持(可解决浏览器的兼容性)
registry.addEndpoint("/chat").withSockJS();
}
}
- 实体类
package org.sang.wschat.model;
public class Message {
private String name;
private String content;
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public String getContent() {
return content;
}
public void setContent(String content) {
this.content = content;
}
}
- 定义一个Controller
package org.sang.wschat.controller;
import org.sang.wschat.model.Chat;
import org.sang.wschat.model.Message;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.messaging.handler.annotation.MessageMapping;
import org.springframework.messaging.handler.annotation.SendTo;
import org.springframework.messaging.simp.SimpMessagingTemplate;
import org.springframework.stereotype.Controller;
import java.security.Principal;
@Controller
public class GreetingController {
//根据配置下面被注解的方法将接受来自"/app/hello"路径发送来的消息,经方法处理后,
//将消息转发到SentTo定义的路径上,再转发给消息代理,然后再广播给客户端
@MessageMapping("/hello")
@SendTo("/topic/greetings")
public Message greeting(Message message) throws Exception {
return message;
}
}
- 界面构建
- static目录下创建chat.html
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>群聊</title>
<script src="/webjars/jquery/jquery.min.js"></script>
<script src="/webjars/sockjs-client/sockjs.min.js"></script>
<script src="/webjars/stomp-websocket/stomp.min.js"></script>
<script src="/app.js"></script>
</head>
<body>
<div>
<label for="name">请输入用户名:</label>
<input type="text" id="name" placeholder="用户名">
</div>
<div>
<button id="connect" type="button">连接</button>
<button id="disconnect" type="button" disabled="disabled">断开连接</button>
</div>
<div id="chat" style="display: none;">
<div>
<label for="name">请输入聊天内容:</label>
<input type="text" id="content" placeholder="聊天内容">
</div>
<button id="send" type="button">发送</button>
<div id="greetings">
<div id="conversation" style="display: none">群聊进行中...</div>
</div>
</div>
</body>
</html>
- 其中除了app.js另外三个js是通过pom文件通过倚赖引进的,app.js如下
var stompClient = null;
function setConnected(connected) {
$("#connect").prop("disabled", connected);
$("#disconnect").prop("disabled", !connected);
if (connected) {
$("#conversation").show();
$("#chat").show();
}
else {
$("#conversation").hide();
$("#chat").hide();
}
$("#greetings").html("");
}
/*表示创建一个webSocket连接,而建立连接时还必须先输入用户名*/
function connect() {
if (!$("#name").val()) {
return;
}
/*使用SockJS创建连接,然后创建一个Stomp实例发起连接*/
var socket = new SockJS('/chat');
stompClient = Stomp.over(socket);
stompClient.connect({}, function (frame) {
setConnected(true);/*页面设置*/
/*订阅服务端发回来的消息,并展示*/
stompClient.subscribe('/topic/greetings', function (greeting) {
showGreeting(JSON.parse(greeting.body));
});
});
}
/*断开一个webSocket连接*/
function disconnect() {
if (stompClient !== null) {
stompClient.disconnect();
}
setConnected(false);
}
function sendName() {
stompClient.send("/app/hello",
{},
JSON.stringify({'name': $("#name").val(),'content':$("#content").val()}));
}
function showGreeting(message) {
$("#greetings")
.append("<div>" + message.name+":"+message.content + "</div>");
}
$(function () {
$( "#connect" ).click(function() { connect(); });
$( "#disconnect" ).click(function() { disconnect(); });
$( "#send" ).click(function() { sendName(); });
});
6.测试
- 启动项目访问http://localhost:8080/chat.html ,输入用户名连接,再启动其它浏览器,重复步骤即可实现群聊
二:消息点对点发送
- 上面发送消息使用到了@SendTo注解,由由该注解处理过的方法转发给broker,再由broker进行消息广播,而spring还提供了SimpMessagingTemplate,可将该类直接注入Controlelr并代替@SendTo注解,而有了SimpMessagingTemplate的支持,可以在任意地方发送消息到broker,即点对点发送
- 继续添加倚赖
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-security</artifactId>
</dependency>
- 配置Spring Security,添加两个用户,并且所有地址都需认证后才能访问
package org.sang.wschat.config;
import org.springframework.context.annotation.Bean;
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.configuration.WebSecurityConfigurerAdapter;
import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;
import org.springframework.security.crypto.password.PasswordEncoder;
@Configuration
public class WebSecurityConfig extends WebSecurityConfigurerAdapter {
@Bean
PasswordEncoder passwordEncoder() {
return new BCryptPasswordEncoder();
}
@Override
protected void configure(AuthenticationManagerBuilder auth) throws Exception {
auth.inMemoryAuthentication()
.withUser("admin")
.password("$2a$10$RMuFXGQ5AtH4wOvkUqyvuecpqUSeoxZYqilXzbz50dceRsga.WYiq")
.roles("admin")
.and()
.withUser("sang")
.password("$2a$10$RMuFXGQ5AtH4wOvkUqyvuecpqUSeoxZYqilXzbz50dceRsga.WYiq")
.roles("user");
}
@Override
protected void configure(HttpSecurity http) throws Exception {
http.authorizeRequests()
.anyRequest().authenticated()
.and()
.formLogin().permitAll();
}
}
- 对WebSocketConfig中下面的这行进行改造,可以看到又增加了一个broker前缀"/queue",为了方便对不同类消息发送进行管理
config.enableSimpleBroker("/topic","/queue");
- chat实体类
package org.sang.wschat.model;
public class Chat {
private String to;
private String from;
private String content;
public String getTo() {
return to;
}
public void setTo(String to) {
this.to = to;
}
public String getFrom() {
return from;
}
public void setFrom(String from) {
this.from = from;
}
public String getContent() {
return content;
}
public void setContent(String content) {
this.content = content;
}
}
- 向之前的Controller中添加一个注入和方法,即该点对点方法使用SimpMessagingTemplate 进行广播
@Autowired
SimpMessagingTemplate messagingTemplate;
/**
*
* @param principal 登陆用户的信息
* @param chat 客户端发来的消息
*/
@MessageMapping("/chat")
public void chat(Principal principal, Chat chat) {
//获取当前用户的用户名,并设置给Chat的from属性
String from = principal.getName();
chat.setFrom(from);
//发送消息
messagingTemplate.convertAndSendToUser(chat.getTo(),
"/queue/chat", chat);
}
- 创建聊天界面onlinechat.html
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>单聊</title>
<script src="/webjars/jquery/jquery.min.js"></script>
<script src="/webjars/sockjs-client/sockjs.min.js"></script>
<script src="/webjars/stomp-websocket/stomp.min.js"></script>
<script src="/chat.js"></script>
</head>
<body>
<div id="chat">
<div id="chatsContent">
</div>
<div>
请输入聊天内容:
<input type="text" id="content" placeholder="聊天内容">
目标用户:
<input type="text" id="to" placeholder="目标用户">
<button id="send" type="button">发送</button>
</div>
</div>
</body>
</html>
- chat.js
var stompClient = null;
function connect() {
var socket = new SockJS('/chat');
stompClient = Stomp.over(socket);
stompClient.connect({}, function (frame) {
stompClient.subscribe('/user/queue/chat', function (chat) {
showGreeting(JSON.parse(chat.body));
});
});
}
function sendMsg() {
stompClient.send("/app/chat", {},
JSON.stringify({'content':$("#content").val(),
'to':$("#to").val()}));
}
function showGreeting(message) {
$("#chatsContent")
.append("<div>" + message.from+":"+message.content + "</div>");
}
$(function () {
connect();
$( "#send" ).click(function() { sendMsg(); });
});
- chat.js与前文的app.js主要有三处不同,一是订阅地址多了个"/user"前缀,这是因为SimpMessagingTemplate中自动添加了路径前缀; 二是聊天发送路径为"/app/chat"; 三是发送的消息内容中有一个to字段用来描述消息的目标用户
- 测试
- 访问http://localhost:8080/onlinechat.html ,使用admin/123 ,sang/123进行登陆