websocket定时推送数据-编程知识

websocket定时推送数据

示例代码

1、添加pom.xml依赖

<dependency><groupId>org.springframework</groupId><artifactId>spring-websocket</artifactId>
</dependency>

2、创建websocket配置类

package com.success.socket;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.server.standard.ServerEndpointExporter;
/*** WebSocket配置类*/
@Configuration
public class WebSocketConfig {@Beanpublic ServerEndpointExporter serverEndpointExporter() {return new ServerEndpointExporter();}
}

3、创建WebSokcet工具类

package com.success.socket;import cn.hutool.json.JSONUtil;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.example.dataservice.models.common.ResultPO;
import com.success.service.HomePageService;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springblade.util.SpringUtils;
import org.springframework.stereotype.Component;import javax.websocket.*;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicInteger;@Component
@ServerEndpoint(value = "/ws/{userId}")
public class SocketServer {private final static Logger log = LoggerFactory.getLogger(SocketServer.class);//记录当前在线连接数private static final AtomicInteger onlineCount = new AtomicInteger(0);//存放所有在线的客户端	private static final ConcurrentHashMap<String, Session> clients = new ConcurrentHashMap<>();//注入homePageService ,实际的业务在这里实现private static HomePageService homePageService = SpringUtils.getApplicationContext().getBean(HomePageService.class);/*** 连接建立成功调用的方法* 连接成功返回 {"msg":"ok","event":"open"}*/@OnOpenpublic void onOpen(Session session, @PathParam("userId") String userId) {onlineCount.incrementAndGet(); // 在线数加1clients.put(session.getId(), session);JSONObject json = new JSONObject();json.put("event", "open");json.put("msg", "ok");this.sendMessage(json.toJSONString());log.info("有新连接加入：{}，当前在线人数为：{}", session.getId(), onlineCount.get());}/*** 连接关闭调用的方法*/@OnClosepublic void onClose(Session session) {onlineCount.decrementAndGet(); // 在线数减1clients.remove(session.getId());log.info("有一连接关闭：{}，当前在线人数为：{}", session.getId(), onlineCount.get());}/*** 收到客户端消息后调用的方法** @param message 客户端发送过来的消息 {"event":"getCountGeneral"}*                当业务改动数据时，可以主动发消息（不需要客户端主动请求）*/@OnMessagepublic void onMessage(String message, Session session) {log.info("服务端收到客户端[{}]的消息:{}", session.getId(), message);if (StringUtils.isBlank(message)) {JSONObject jsonRes = new JSONObject();jsonRes.put("msg", "event无法处理");this.sendMessage(jsonRes.toJSONString());}//解析发送的报文JSONObject jsonObject = JSON.parseObject(message);//24小时实时数据  前端发送{"event":"getCountGeneral"}if ("getCountGeneral".equals(jsonObject.getString("event"))) {this.sendMessage(getCountGeneral());}}@OnErrorpublic void onError(Session session, Throwable error) {log.error("链接发生错误:{},原因:{}", session.getId(), error.getMessage());error.printStackTrace();}/*** 获取24小时实时数据** @return*/public String getCountGeneral() {Map<String,Object> map = new HashMap<>();		//业务1map.put("interfaceCallTop",homePageService.getInterfaceCallTop(params));//业务2map.put("countGeneral",homePageService.getCountGeneral1());return JSONObject.toJSONString(map);}/*** 群发消息** @param message 消息内容*/public void sendMessage(String message) {if (clients.isEmpty()) {log.info("*********  当前无客户端链接  *********");}for (Map.Entry<String, Session> sessionEntry : clients.entrySet()) {Session toSession = sessionEntry.getValue();toSession.getAsyncRemote().sendText(message);log.info("服务端发送消息给客户端：{}", toSession.getId());}}/*** 启动群发任务，用于定时任务*/public void run() {//首页24小时实时数据 当日接口访问排行 当日单位调用排行 当日应用调用排行this.sendMessage(getCountGeneral());}
}

4、定时任务

有时候需要后端定时推送数据给前端，需要增加以下代码：

package com.success.utils;import com.success.socket.SocketServer;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;@Component
@RestController
public class DataPushScheduledTask {private final static Logger log = LoggerFactory.getLogger(DataPushScheduledTask.class);/*** 30秒一次* 定时场景数据推送*/@Scheduled(cron = "0/30 * * * * ? ")@GetMapping("/test")public void executeDataPush() {log.info("*********   定时任务执行   *********");new SocketServer().run();log.info("*********   定时任务完成   *********");}
}