dromara/RuoYi-Cloud-Plus
1
0
Fork 0
mirror of https://gitee.com/dromara/RuoYi-Cloud-Plus.git synced 2025-09-02 02:34:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

update 优化 登录消息 支持集群发送

Browse Source
This commit is contained in:
疯狂的狮子Li
2024-01-18 12:16:16 +08:00
parent 1c8dbf3ffa
commit af2e8935b4
4 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ruoyi-api/ruoyi-api-resource/src/main/java/org/dromara/resource/api/RemoteMessageService.java
View File

@@ -13,7 +13,7 @@ public interface RemoteMessageService {
* @param sessionKey session主键 一般为用户id
* @param message 消息文本
*/
void sendMessage(Long sessionKey, String message);
void publishMessage(Long sessionKey, String message);
void publishAll(String message);
}

4
ruoyi-api/ruoyi-api-resource/src/main/java/org/dromara/resource/api/RemoteMessageServiceStub.java
View File

@@ -20,9 +20,9 @@ public class RemoteMessageServiceStub implements RemoteMessageService {
* @param sessionKey session主键 一般为用户id
* @param message 消息文本
*/
public void sendMessage(Long sessionKey, String message) {
public void publishMessage(Long sessionKey, String message) {
try {
remoteMessageService.sendMessage(sessionKey, message);
remoteMessageService.publishMessage(sessionKey, message);
} catch (Exception e) {
log.warn("websocket 功能未开启或服务未找到");
}

2
ruoyi-auth/src/main/java/org/dromara/auth/controller/TokenController.java
View File

@@ -97,7 +97,7 @@ public class TokenController {
Long userId = LoginHelper.getUserId();
scheduledExecutorService.schedule(() -> {
remoteMessageService.sendMessage(userId, "欢迎登录RuoYi-Cloud-Plus微服务管理系统");
remoteMessageService.publishMessage(userId, "欢迎登录RuoYi-Cloud-Plus微服务管理系统");
}, 3, TimeUnit.SECONDS);
return R.ok(loginVo);
}

10
ruoyi-modules/ruoyi-resource/src/main/java/org/dromara/resource/dubbo/RemoteMessageServiceImpl.java
View File

@@ -3,10 +3,13 @@ package org.dromara.resource.dubbo;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.dubbo.config.annotation.DubboService;
import org.dromara.common.websocket.dto.WebSocketMessageDto;
import org.dromara.common.websocket.utils.WebSocketUtils;
import org.dromara.resource.api.RemoteMessageService;
import org.springframework.stereotype.Service;
import java.util.List;
/**
* 短信服务
*
@@ -25,8 +28,11 @@ public class RemoteMessageServiceImpl implements RemoteMessageService {
* @param message 消息文本
*/
@Override
public void sendMessage(Long sessionKey, String message) {
WebSocketUtils.sendMessage(sessionKey, message);
public void publishMessage(Long sessionKey, String message) {
WebSocketMessageDto dto = new WebSocketMessageDto();
dto.setMessage(message);
dto.setSessionKeys(List.of(sessionKey));
WebSocketUtils.publishMessage(dto);
}
/**