Springboot 使用 webSocket

4. 第140-157行，是针对特定客户端发送消息。服务器和客户端在建立连接成功后就生成了一个WebSocket对象，并存在集合中，对象里特有的属性是我们设置的userID。所以通过唯一的userID就能标识服务器与该客户端建立的那个连接啦！这样要求发送消息时，传入userID与消息，服务器在自己的WebSocket连接集合中遍历找到对应客户端的连接，就可以直接发消息过去啦~~

package com.cuc.happyseat.websocket;

import java.io.IOException;

import java.util.concurrent.CopyOnWriteArraySet;

import javax.websocket.EncodeException;

import javax.websocket.OnClose;

import javax.websocket.OnError;

import javax.websocket.OnMessage;

import javax.websocket.OnOpen;

import javax.websocket.Session;

import javax.websocket.server.PathParam;

import javax.websocket.server.ServerEndpoint;

import org.springframework.stereotype.Component;

/*@ServerEndpoint注解是一个类层次的注解，它的功能主要是将目前的类定义成一个websocket服务器端,

• 注解的值将被用于监听用户连接的终端访问URL地址,客户端可以通过这个URL来连接到WebSocket服务器端

*/

@ServerEndpoint(“/websocket/{userID}”)

@Component

public class WebSocketServer {

//每个客户端都会有相应的session,服务端可以发送相关消息

private Session session;

//接收userID

private Integer userID;

//J.U.C包下线程安全的类，主要用来存放每个客户端对应的webSocket连接

private static CopyOnWriteArraySet copyOnWriteArraySet = new CopyOnWriteArraySet();

public Integer getUserID() {

return userID;

}

/**

• @Name：onOpen

• @Description：打开连接。进入页面后会自动发请求到此进行连接

• @Author：mYunYu

• @Create Date：14:46 2018/11/15

• @Parameters：@PathParam(“userID”) Integer userID

• @Return：

*/

@OnOpen

public void onOpen(Session session, @PathParam(“userID”) Integer userID) {

this.session = session;

this.userID = userID;

System.out.println(this.session.getId());

//System.out.println(“userID:” + userID);

copyOnWriteArraySet.add(this);

System.out.println(“websocket有新的连接, 总数:”+ copyOnWriteArraySet.size());

}

/**

• @Name：onClose

• @Description：用户关闭页面，即关闭连接

• @Author：mYunYu

• @Create Date：14:46 2018/11/15

• @Parameters：

• @Return：

*/

@OnClose

public void onClose() {

copyOnWriteArraySet.remove(this);

System.out.println(“websocket连接断开, 总数:”+ copyOnWriteArraySet.size());

}

/**

• @Name：onMessage

• @Description：测试客户端发送消息，测试是否联通

• @Author：mYunYu

• @Create Date：14:46 2018/11/15

• @Parameters：

• @Return：

*/

@OnMessage

public void onMessage(String message) {

System.out.println(“websocket收到客户端发来的消息:”+message);

}

/**

• @Name：onError

• @Description：出现错误

• @Author：mYunYu

• @Create Date：14:46 2018/11/15

• @Parameters：

• @Return：

*/

@OnError

public void onError(Session session, Throwable error) {

System.out.println(“发生错误：” + error.getMessage() + “; sessionId:” + session.getId());

error.printStackTrace();

}

public void sendMessage(Object object){

//遍历客户端

for (WebSocketServer webSocket : copyOnWriteArraySet) {

System.out.println(“websocket广播消息：” + object.toString());

try {

//服务器主动推送

webSocket.session.getBasicRemote().sendObject(object) ;

} catch (Exception e) {

e.printStackTrace();

}

}

}

/**

• @Name：sendMessage

• @Description：用于发送给客户端消息（群发）

• @Author：mYunYu

• @Create Date：14:46 2018/11/15

• @Parameters：

• @Return：

*/

public void sendMessage(String message) {

//遍历客户端

for (WebSocketServer webSocket : copyOnWriteArraySet) {

System.out.println(“websocket广播消息：” + message);

try {

//服务器主动推送

webSocket.session.getBasicRemote().sendText(message);

} catch (Exception e) {

e.printStackTrace();

}

}

}

/**

• @throws Exception

• @Name：sendMessage

• @Description：用于发送给指定客户端消息

• @Author：mYunYu

• @Create Date：14:47 2018/11/15

• @Parameters：

• @Return：

*/

public void sendMessage(Integer userID, String message) throws Exception {

Session session = null;

WebSocketServer tempWebSocket = null;

for (WebSocketServer webSocket : copyOnWriteArraySet) {

if (webSocket.getUserID() == userID) {

tempWebSocket = webSocket;

session = webSocket.session;

break;

}

}

if (session != null) {

//服务器主动推送

tempWebSocket.session.getBasicRemote().sendText(message);

} else {

System.out.println(“没有找到你指定ID的会话：{}”+ “; userId:” + userID);

}

}

}

• Controller类的编写。

1. 1. 我在看博客的时候，发现有的博主写了Controller类，有的没写，我就有点疑惑了。后来，咳咳，发现特地写了一个Controller类只是为了测试。。

2. 一般在实际项目中，在确保建立连接过程没有问题的情况下，我们就直接在一些写好的接口中写 WebSocketServer.sendMessage(param, message)语句就行了。

3. 也因此，你写的位置就决定了你什么时候给你的客户端发消息，这样也就实现了主动推送消息的功能咯~

前提是：在你的Controller类里，以@Resource的方式注入WebSocket，而不是@Autowired方式哦(⊙o⊙)。

@Resource

WebSocketServer webSocket;

@Autowired

UserService userService;

调用示例：

if(userID>0) {

boolean location = userService.getLocation(userID);

if(location==false) {//验证用户当前不在馆内

boolean i = userService.modifyLocation(userID, true);

if(i==true) {

modelMap.put(“successEnter”, true);

//发消息给客户端

webSocket.sendMessage(userID, “success”);

}

}else {

modelMap.put(“successEnter”, false);

//发消息给客户端

webSocket.sendMessage(userID, “fail”);

}

}else {