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

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

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

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

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

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

Loading…
Cancel
Save