dromara/RuoYi-Cloud-Plus
1
0
Fork 0
mirror of https://gitee.com/dromara/RuoYi-Cloud-Plus.git synced 2025-11-28 01:00:05 +08: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 sessionKey session主键 一般为用户id
* @param message 消息文本 * @param message 消息文本
*/ */
void sendMessage(Long sessionKey, String message); void publishMessage(Long sessionKey, String message);
void publishAll(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 sessionKey session主键 一般为用户id
* @param message 消息文本 * @param message 消息文本
*/ */
public void sendMessage(Long sessionKey, String message) { public void publishMessage(Long sessionKey, String message) {
try { try {
remoteMessageService.sendMessage(sessionKey, message); remoteMessageService.publishMessage(sessionKey, message);
} catch (Exception e) { } catch (Exception e) {
log.warn("websocket 功能未开启或服务未找到"); 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(); Long userId = LoginHelper.getUserId();
scheduledExecutorService.schedule(() -> { scheduledExecutorService.schedule(() -> {
remoteMessageService.sendMessage(userId, "欢迎登录RuoYi-Cloud-Plus微服务管理系统"); remoteMessageService.publishMessage(userId, "欢迎登录RuoYi-Cloud-Plus微服务管理系统");
}, 3, TimeUnit.SECONDS); }, 3, TimeUnit.SECONDS);
return R.ok(loginVo); 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.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.apache.dubbo.config.annotation.DubboService; import org.apache.dubbo.config.annotation.DubboService;
import org.dromara.common.websocket.dto.WebSocketMessageDto;
import org.dromara.common.websocket.utils.WebSocketUtils; import org.dromara.common.websocket.utils.WebSocketUtils;
import org.dromara.resource.api.RemoteMessageService; import org.dromara.resource.api.RemoteMessageService;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import java.util.List;
/** /**
* 短信服务 * 短信服务
* *
@@ -25,8 +28,11 @@ public class RemoteMessageServiceImpl implements RemoteMessageService {
* @param message 消息文本 * @param message 消息文本
*/ */
@Override @Override
public void sendMessage(Long sessionKey, String message) { public void publishMessage(Long sessionKey, String message) {
WebSocketUtils.sendMessage(sessionKey, message); WebSocketMessageDto dto = new WebSocketMessageDto();
dto.setMessage(message);
dto.setSessionKeys(List.of(sessionKey));
WebSocketUtils.publishMessage(dto);
} }
/** /**