生命无罪,健康万岁,我是laity。
我曾七次鄙视自己的灵魂:
第一次,当它本可进取时,却故作谦卑;
第二次,当它在空虚时,用爱欲来填充;
第三次,在困难和容易之间,它选择了容易;
第四次,它犯了错,却借由别人也会犯错来宽慰自己;
第五次,它自由软弱,却把它认为是生命的坚韧;
第六次,当它鄙夷一张丑恶的嘴脸时,却不知那正是自己面具中的一副;
第七次,它侧身于生活的污泥中,虽不甘心,却又畏首畏尾。
Spring定时任务 + webSocket实现定时给指定用户发送消息:类似于消息中心;
相信有需求的小伙伴读此文章可以有一定的帮助或者思路
在做这个业务的时候也遇到了很多的坑,但是现在我帮你踩完了。
因不同的人业务不同,所以这里我省略掉了我的业务逻辑,有需求的小伙伴可以私信我进一步沟通;
<dependency>
<groupId>org.springframework.bootgroupId>
<artifactId>spring-boot-starter-websocketartifactId>
<version>2.7.0version>
dependency>
/**
* @Project: JavaLaity
* @Description: 服务端
*/
@Slf4j
@Component
@ServerEndpoint(value = "/websocket/{userId}")
public class WebSocketServer {
/**
* concurrent包的线程安全Set,用来存放每个客户端对应的MyWebSocket对象,若要实现服务器端与单一客户端通信的话,可以使用Map来存放,其中key可以为用户标识
*/
private static final CopyOnWriteArraySet<WebSocketServer> webSocketSet = new CopyOnWriteArraySet<WebSocketServer>();
// public static ThreadLocal clientUser = new ThreadLocal<>();
/**
* 用于记录连接的客户端 (sid,session)
*/
public static Map<String, Session> clients = new ConcurrentHashMap<>();
/**
* 基于userId关联sid(用于解决同一用户id,在多个web端连接的问题)
*/
public static Map<Integer, Set<String>> conn = new ConcurrentHashMap<>();
/**
* 与某个客户端的连接会话,需要通过它来给客户端发送数据
*/
private Session session;
private String sid = null;
// 当WebSocket连接建立时,会调用标注有@OnOpen的方法
@OnOpen
public void onOpen(Session session, @PathParam("userId") Integer userId) {
System.out.println("WebSocket opened: " + session.getId());
this.session = session;
this.sid = UUID.randomUUID().toString();
clients.put(this.sid, session);
Set<String> clientSet = conn.get(userId);
if (clientSet == null) {
clientSet = new HashSet<>();
conn.put(userId, clientSet);
}
clientSet.add(this.sid);
// 加入set中
webSocketSet.add(this);
}
// 给所有当前连接的用户发送消息
public void sendMessageAll(String message) {
try {
if (webSocketSet.size() != 0) {
for (WebSocketServer item : webSocketSet) {
if (item != null) {
// 同步锁,解决多线程下发送消息异常关闭
// 避免高并发下多处频繁调用sendMessage()方法发送消息而导致的webSocket挂掉,我们在sendMessage这个方法里面加入同步锁,
// 锁住session,这样就能保障webSocket有条不絮的向前端推送消息
synchronized (item.session) {
item.session.getBasicRemote().sendText(message);
}
}
}
}
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* 根据用户ID发送给某一个用户
*/
public void sendMessageByUserId(Integer userId, String message) {
if (userId != null) {
Set<String> clientSet = conn.get(userId);
if (clientSet != null) {
Iterator<String> iterator = clientSet.iterator();
while (iterator.hasNext()) {
String sid = iterator.next();
Session session = clients.get(sid);
if (session != null) {
try {
session.getBasicRemote().sendText(message);
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
}
}
// TODO 当浏览器关闭触发该事件
@OnClose
public void OnClose(Session session) {
log.info(this.sid + "断开连接");
clients.remove(this.sid);
webSocketSet.remove(this);
}
// 当收到客户端(前端)发送的消息时,会调用@OnMessage方法
@OnMessage
public void OnMessage(String message, Session session) {
System.out.println("Received message:" + message);
}
// TODO 发生错误时调用
@OnError
public void OnError(Session session, Throwable error) {
System.out.println("发生错误!");
error.printStackTrace();
}
}
/**
* @Project: JavaLaity
* @Description: 实现业务逻辑的位置
*/
@Configuration
public class AideScheduleTask {
@Autowired
WebSocketServer webSocketServer;
@Autowired
RedisCache redisCache;
// 测试
@Scheduled(cron = "0/20 * * * * ?")
private void testTask() throws Exception {
// 使用ThreadLocal - 哈哈哈,弃用了,行不通!。
// TODO: 你需要的业务逻辑
SysUser sysUser = new SysUser().setId(1)
if (sysUser.getId() != null) {
webSocketServer.sendMessageByUserId(sysUser.getId(), "这个是老大可以发:role = ");
}
}
}
在写Corn表达式时需要注意的点:
个人认为有性能问题以及数据库压力的大问题,这里本人就只是简单的分享下设计思路,不同业务需求,实现策略不同。也有肯定公司不在乎这种性能问题。
愿每个人都能遵循自己的时钟,做不后悔的选择。我是Laity,正在前行的Laity。