WebSocket的简单认识&SpringBoot整合websocket

注意：

　　ws://localhost:86/WebSocket/chat 这个地址 是  协议://IP:Port/项目名称/websocket地址

第一次建立Socket连接的时候请求头和响应头如下：

 Connection 必须设置 Upgrade，表示客户端希望连接升级。

 Upgrade 字段必须设置 Websocket，表示希望升级到 Websocket 协议。

 Sec-WebSocket-Key 是随机的字符串，服务器端会用这些数据来构造出一个 SHA-1 的信息摘要。把 “Sec-WebSocket-Key” 加上一个特殊字符串 “258EAFA5-E914-47DA-95CA-C5AB0DC85B11”，然后计算 SHA-1 摘要，之后进行 BASE-64 编码，将结果做为 “Sec-WebSocket-Accept” 头的值，返回给客户端。如此操作，可以尽量避免普通 HTTP 请求被误认为 Websocket 协议。

 Sec-WebSocket-Version 表示支持的 Websocket 版本。RFC6455 要求使用的版本是 13，之前草案的版本均应当弃用。

 Origin 字段是可选的，通常用来表示在浏览器中发起此 Websocket 连接所在的页面，类似于 Referer。但是，与 Referer 不同的是，Origin 只包含了协议和主机名称。

 其他一些定义在 HTTP 协议中的字段，如 Cookie 等，也可以在 Websocket 中使用。

补充：上面是每个连接都会创建1个ChatServer对象，多例模式的对象。Session与HttpSession不一样。

 3. SpringBoot中开启WebSocket

　　springboor整合websocket也非常简单。如下：

1.  pom.xml增加如下依赖

org.springframework.boot

spring-boot-starter-websocket

2.代码如下

配置类：

package cn.qlq.config;

import org.springframework.context.annotation.Bean;

import org.springframework.context.annotation.Configuration;

import org.springframework.web.socket.server.standard.ServerEndpointExporter;

@Configuration

public class WebSocketConfig {

@Bean

public ServerEndpointExporter serverEndpointExporter() {

return new ServerEndpointExporter();

}

}

ChatServer.java

package cn.qlq.websocket;

import java.io.IOException;

import java.util.Hashtable;

import javax.websocket.OnClose;

import javax.websocket.OnError;

import javax.websocket.OnMessage;

import javax.websocket.OnOpen;

import javax.websocket.Session;

import javax.websocket.server.ServerEndpoint;

import org.springframework.stereotype.Component;

@ServerEndpoint("/chat")

@Component

public class ChatServer {

private boolean first = true;

private String name;

private Session session;

private static final Hashtable clientSet = new Hashtable();

public ChatServer() {

System.out.println("========ChatServer创建============");

}

/*

* 客户端连接时触发该方法

*/

@OnOpen

public void onOpen(Session session) throws IOException {

this.session = session;

}

/*

* 客户端断开时触发该方法

*/

@OnClose

public void onClose() {

clientSet.remove(name);

String message = String.format("【%s %s】", name, "离开了聊天室！");

broadcast(message);

}

/*

* 客户端收到消息时触发该方法

*/

@OnMessage

public void onMessage(String msg) {

if (first) {

name = msg;

clientSet.put(name, this);

String message = String.format("【%s %s】", name, "加入了聊天室！");

// 发送消息

broadcast(message);

first = false;

} else {

broadcast("【" + name + "】" + ":" + msg);

}

}

// 当客户端通信出现错误时，激发该方法

@OnError

public void onError(Throwable t) throws Throwable {

System.out.println("WebSocket服务端错误 " + t);

}

/**

* 将此方法设计为静态方法用于服务器主动调用该方法广播消息

*

* @param msg

*/

public static void broadcast(String msg) {

// 遍历服务器关联的所有客户端

ChatServer client = null;

for (String nickname : clientSet.keySet()) {

try {

client = (ChatServer) clientSet.get(nickname);

synchronized (client) {

// 发送消息

client.session.getBasicRemote().sendText(msg);

}

} catch (IOException e) {

System.out.println("聊天错误，向客户端 " + client + " 发送消息出现错误。");

clientSet.remove(nickname);

try {

client.session.close();

} catch (IOException e1) {

}

String message = String.format("【%s %s】", client.name, "已经被断开了连接。");

broadcast(message);

}

}

}

}

Controller层代码：

package cn.qlq.controller.websocket;

import org.springframework.stereotype.Controller;

import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.ResponseBody;

import cn.qlq.websocket.ChatServer;

@RequestMapping("/chat")

@Controller

public class ChatController {

/**

* 跳转到聊天页面

*

* @return

*/

@RequestMapping("/chatRoom")

public String chatRoom() {

return "websocket/chatRoom";

}

/**

* http请求发送消息

*

* @param msg

* @return

*/

@RequestMapping("/sendMsg")

public @ResponseBody String sendMsg(String msg) {

ChatServer.broadcast(msg);

return "success";

}

}

前端页面：

2.测试

http发送消息：

　　经测试，上面ChatServer也是多例模式，每次请求都会创建ChatServer实例对象。

Git地址：https://github.com/qiao-zhi/springboot-ssm

文章链接