springboot业务功能实战（四）告别轮询，websocket的集成使用

} else {

sessionPool.put(userName, session);

}

logger.info(“【websocket消息】有新的连接，总数为:” + webSockets.size());

}

@OnClose

public void onClose() {

webSockets.remove(this);

logger.info(“【websocket消息】连接断开，总数为:” + webSockets.size());

}

@OnMessage

public void onMessage(String message) {

logger.info(“【websocket消息】收到客户端消息:” + message);

}

/**

• 功能描述: 此为广播消息

• @param: [message] (消息)

• @return: void ()

*/

public void sendAllMessage(String message) {

for (WebSocket webSocket : webSockets) {

logger.info(“【websocket消息】广播消息:” + message);

try {

if (webSocket.session.isOpen()) {

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

}

} catch (Exception e) {

e.printStackTrace();

}

}

}

/**

• 功能描述:此为单点消息 (发送文本) 现在可以发送给多客户端

• @param: [userName, message] (接收人,发送消息)

• @return: void ()

*/

public void sendTextMessage(String userName, String message) {

// 遍历sessionPool

for (String key : sessionPool.keySet()) {

// 存在当前用户

if (key.toString().indexOf(userName) != -1) {

Session session = sessionPool.get(key);

if (session != null && session.isOpen()) {

try {

session.getAsyncRemote().sendText(message);

} catch (Exception e) {

e.printStackTrace();

}

}

}

}

}

/**

• 功能描述: 此为单点消息 (发送文本) 现在可以发送给多客户端

• @param: [userName, message] (接收人,发送消息)

• @return: void ()

*/

public void sendObjMessage(String userName, Object message) {

// 遍历sessionPool

for (String key : sessionPool.keySet()) {

// 存在当前用户

if (key.toString().indexOf(userName) != -1) {

Session session = sessionPool.get(key);

if (session != null && session.isOpen()) {

try {

session.getAsyncRemote().sendObject(message);

} catch (Exception e) {

e.printStackTrace();

}

}

}

}

}

}

发送信息

@RestController

@RequestMapping(“websocket”)

public class WebSocketController {

@GetMapping(“setMessage”)

@ApiOperation(value = “发送信息接口”, notes = “发送信息接口”)

public Result

webSocket(@ApiParam(name = “定时任务日志实体”, value = “定时任务日志实体”, required = false) @RequestBody MessageVO messageVO) {

Result result = new Result();

String userName = messageVO.getUserName();

String message = messageVO.getMessage();

WebSocket webSocket = new WebSocket();

webSocket.sendTextMessage(userName, message);

return result;

}

}

前段代码

import sysConfig from “…/config”;

import {Notification} from ‘element-ui’;

import {EVENT_TYPE} from “…/const”;

export function openSocket(userId) {

let ws = new WebSocket(${sysConfig.SOCKET_URL}/${userId});

// let ws = new WebSocket(ws://121.40.165.18:8800);

ws.onopen = function (evt) {

Notification({

title: ‘欢迎回来！’,

message: ${sysConfig.SOCKET_URL}/${userId}

});

};

ws.onmessage = function (e) {

console.log(typeof e.data);

try{

if(e.data!=undefined || e.data!=null){

let json= JSON.parse(e.data);

Notification({

title: json.messageTitle,

message: json.messageText

});

//通知页面更新

window.postMessage(EVENT_TYPE.updateMessage,‘/’);

}

}catch(err){

console.log(“webSocke异常,异常信息:”+err)