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

2.X
疯狂的狮子Li 1 year ago
parent 1c8dbf3ffa
commit af2e8935b4

@ -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);
} }

@ -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 功能未开启或服务未找到");
} }

@ -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);
} }

@ -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);
} }
/** /**

Loading…
Cancel
Save