yudao
/
spring-cloud
mirror of https://gitee.com/zhijiantianya/yudao-cloud.git
3
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

【同步】BOOT 和 CLOUD 的功能（IM）解决启动报错问题

master-jdk17
YunaiV 2026-06-01 00:22:51 +08:00
parent 0c261b4e02
commit 2ba231aa55
6 changed files with 79 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
yudao-module-im/yudao-module-im-server/pom.xml
View File

@ -35,6 +35,11 @@
<artifactId>yudao-module-system-api</artifactId>
<version>${revision}</version>
</dependency>
<dependency>
<groupId>cn.iocoder.cloud</groupId>
<artifactId>yudao-module-infra-api</artifactId>
<version>${revision}</version>
</dependency>
<!-- 业务组件 -->
<dependency>
@ -47,10 +52,6 @@
<groupId>cn.iocoder.cloud</groupId>
<artifactId>yudao-spring-boot-starter-security</artifactId>
</dependency>
<dependency>
<groupId>cn.iocoder.cloud</groupId>
<artifactId>yudao-spring-boot-starter-websocket</artifactId>
</dependency>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-validation</artifactId>

11
yudao-module-im/yudao-module-im-server/src/main/java/cn/iocoder/yudao/module/im/framework/rpc/config/RpcConfiguration.java Normal file
View File

@ -0,0 +1,11 @@
package cn.iocoder.yudao.module.im.framework.rpc.config;
import cn.iocoder.yudao.module.infra.api.websocket.WebSocketSenderApi;
import cn.iocoder.yudao.module.system.api.user.AdminUserApi;
import org.springframework.cloud.openfeign.EnableFeignClients;
import org.springframework.context.annotation.Configuration;
@Configuration(value = "imRpcConfiguration", proxyBeanMethods = false)
@EnableFeignClients(clients = {AdminUserApi.class, WebSocketSenderApi.class})
public class RpcConfiguration {
}

36
yudao-module-im/yudao-module-im-server/src/main/java/cn/iocoder/yudao/module/im/framework/security/config/SecurityConfiguration.java Normal file
View File

@ -0,0 +1,36 @@
package cn.iocoder.yudao.module.im.framework.security.config;
import cn.iocoder.yudao.framework.security.config.AuthorizeRequestsCustomizer;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.security.config.annotation.web.builders.HttpSecurity;
import org.springframework.security.config.annotation.web.configurers.AuthorizeHttpRequestsConfigurer;
/**
* IM 模块的 Security 配置
*/
@Configuration(proxyBeanMethods = false, value = "imSecurityConfiguration")
public class SecurityConfiguration {
@Bean("imAuthorizeRequestsCustomizer")
public AuthorizeRequestsCustomizer authorizeRequestsCustomizer() {
return new AuthorizeRequestsCustomizer() {
@Override
public void customize(AuthorizeHttpRequestsConfigurer<HttpSecurity>.AuthorizationManagerRequestMatcherRegistry registry) {
// Swagger 接口文档
registry.requestMatchers("/v3/api-docs/**").permitAll()
.requestMatchers("/webjars/**").permitAll()
.requestMatchers("/swagger-ui").permitAll()
.requestMatchers("/swagger-ui/**").permitAll();
// Spring Boot Actuator 的安全配置
registry.requestMatchers("/actuator").permitAll()
.requestMatchers("/actuator/**").permitAll();
// Druid 监控
registry.requestMatchers("/druid/**").permitAll();
}
};
}
}

14
yudao-module-im/yudao-module-im-server/src/main/java/cn/iocoder/yudao/module/im/service/websocket/ImWebSocketServiceImpl.java
View File

@ -2,10 +2,10 @@ package cn.iocoder.yudao.module.im.service.websocket;
import cn.hutool.extra.spring.SpringUtil;
import cn.iocoder.yudao.framework.common.enums.UserTypeEnum;
import cn.iocoder.yudao.framework.websocket.core.sender.WebSocketMessageSender;
import cn.iocoder.yudao.module.im.service.websocket.dto.ImChannelMessageDTO;
import cn.iocoder.yudao.module.im.service.websocket.dto.ImGroupMessageDTO;
import cn.iocoder.yudao.module.im.service.websocket.dto.ImPrivateMessageDTO;
import cn.iocoder.yudao.module.infra.api.websocket.WebSocketSenderApi;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.scheduling.annotation.Async;
@ -33,7 +33,7 @@ import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.
public class ImWebSocketServiceImpl implements ImWebSocketService {
@Resource
private WebSocketMessageSender webSocketMessageSender;
private WebSocketSenderApi webSocketSenderApi;
@Override
public void sendPrivateMessageAsync(Collection<Long> userIds, ImPrivateMessageDTO dto) {
@ -64,7 +64,7 @@ public class ImWebSocketServiceImpl implements ImWebSocketService {
public void doSendPrivateMessage(Collection<Long> userIds, ImPrivateMessageDTO dto) {
for (Long userId : getDistinctUserIds(userIds)) {
try {
webSocketMessageSender.sendObject(UserTypeEnum.ADMIN.getValue(), userId,
webSocketSenderApi.sendObject(UserTypeEnum.ADMIN.getValue(), userId,
ImPrivateMessageDTO.TYPE, dto);
} catch (Exception e) {
log.error("[doSendPrivateMessage][userId({}) dto({}) 发送失败]", userId, dto, e);
@ -79,7 +79,7 @@ public class ImWebSocketServiceImpl implements ImWebSocketService {
public void doSendGroupMessage(Collection<Long> userIds, ImGroupMessageDTO dto) {
for (Long userId : getDistinctUserIds(userIds)) {
try {
webSocketMessageSender.sendObject(UserTypeEnum.ADMIN.getValue(), userId,
webSocketSenderApi.sendObject(UserTypeEnum.ADMIN.getValue(), userId,
ImGroupMessageDTO.TYPE, dto);
} catch (Exception e) {
log.error("[doSendGroupMessage][userId({}) dto({}) 发送失败]", userId, dto, e);
@ -94,7 +94,7 @@ public class ImWebSocketServiceImpl implements ImWebSocketService {
public void doSendChannelMessage(Collection<Long> userIds, ImChannelMessageDTO dto) {
for (Long userId : getDistinctUserIds(userIds)) {
try {
webSocketMessageSender.sendObject(UserTypeEnum.ADMIN.getValue(), userId,
webSocketSenderApi.sendObject(UserTypeEnum.ADMIN.getValue(), userId,
ImChannelMessageDTO.TYPE, dto);
} catch (Exception e) {
log.error("[doSendChannelMessage][userId({}) dto({}) 发送失败]", userId, dto, e);
@ -104,12 +104,12 @@ public class ImWebSocketServiceImpl implements ImWebSocketService {
/**
* 异步广播频道 WebSocket 消息给当前所有在线管理端用户；
* 依赖 WebSocketMessageSender 按 UserType 广播能力，离线用户由客户端上线 pull 兜底
* 依赖 infra WebSocketSenderApi 按 UserType 广播能力，离线用户由客户端上线 pull 兜底
*/
@Async
public void doBroadcastChannelMessage(ImChannelMessageDTO dto) {
try {
webSocketMessageSender.sendObject(UserTypeEnum.ADMIN.getValue(),
webSocketSenderApi.sendObject(UserTypeEnum.ADMIN.getValue(),
ImChannelMessageDTO.TYPE, dto);
} catch (Exception e) {
log.error("[doBroadcastChannelMessage][dto({}) 广播失败]", dto, e);

36
yudao-module-im/yudao-module-im-server/src/test/java/cn/iocoder/yudao/module/im/service/websocket/ImWebSocketServiceImplTest.java
View File

@ -3,9 +3,9 @@ package cn.iocoder.yudao.module.im.service.websocket;
import cn.hutool.extra.spring.SpringUtil;
import cn.iocoder.yudao.framework.common.enums.UserTypeEnum;
import cn.iocoder.yudao.framework.test.core.ut.BaseMockitoUnitTest;
import cn.iocoder.yudao.framework.websocket.core.sender.WebSocketMessageSender;
import cn.iocoder.yudao.module.im.service.websocket.dto.ImGroupMessageDTO;
import cn.iocoder.yudao.module.im.service.websocket.dto.ImPrivateMessageDTO;
import cn.iocoder.yudao.module.infra.api.websocket.WebSocketSenderApi;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.Test;
import org.mockito.InjectMocks;
@ -33,7 +33,7 @@ public class ImWebSocketServiceImplTest extends BaseMockitoUnitTest {
private ImWebSocketServiceImpl imWebSocketService;
@Mock
private WebSocketMessageSender webSocketMessageSender;
private WebSocketSenderApi webSocketSenderApi;
@AfterEach
public void tearDown() {
@ -58,7 +58,7 @@ public class ImWebSocketServiceImplTest extends BaseMockitoUnitTest {
imWebSocketService.sendPrivateMessageAsync(2L, dto);
// 断言
verify(webSocketMessageSender).sendObject(
verify(webSocketSenderApi).sendObject(
eq(UserTypeEnum.ADMIN.getValue()), eq(2L), eq(ImPrivateMessageDTO.TYPE), eq(dto));
}
}
@ -78,7 +78,7 @@ public class ImWebSocketServiceImplTest extends BaseMockitoUnitTest {
imWebSocketService.sendPrivateMessageAsync(2L, dto);
// 断言：事务未提交，未推送
verify(webSocketMessageSender, never()).sendObject(anyInt(), anyLong(), anyString(), any());
verify(webSocketSenderApi, never()).sendObject(anyInt(), anyLong(), anyString(), any());
// 模拟事务提交
List<TransactionSynchronization> syncs =
@ -87,7 +87,7 @@ public class ImWebSocketServiceImplTest extends BaseMockitoUnitTest {
syncs.forEach(TransactionSynchronization::afterCommit);
// 断言：提交后推送
verify(webSocketMessageSender).sendObject(
verify(webSocketSenderApi).sendObject(
eq(UserTypeEnum.ADMIN.getValue()), eq(2L), eq(ImPrivateMessageDTO.TYPE), eq(dto));
} finally {
TransactionSynchronizationManager.clear();
@ -109,11 +109,11 @@ public class ImWebSocketServiceImplTest extends BaseMockitoUnitTest {
imWebSocketService.sendGroupMessageAsync(List.of(1L, 2L, 3L), dto);
verify(webSocketMessageSender).sendObject(
verify(webSocketSenderApi).sendObject(
eq(UserTypeEnum.ADMIN.getValue()), eq(1L), eq(ImGroupMessageDTO.TYPE), eq(dto));
verify(webSocketMessageSender).sendObject(
verify(webSocketSenderApi).sendObject(
eq(UserTypeEnum.ADMIN.getValue()), eq(2L), eq(ImGroupMessageDTO.TYPE), eq(dto));
verify(webSocketMessageSender).sendObject(
verify(webSocketSenderApi).sendObject(
eq(UserTypeEnum.ADMIN.getValue()), eq(3L), eq(ImGroupMessageDTO.TYPE), eq(dto));
}
}
@ -128,13 +128,13 @@ public class ImWebSocketServiceImplTest extends BaseMockitoUnitTest {
dto.setGroupId(10L);
// 给 1 号用户推送时抛异常，不能影响 2/3 号
doThrow(new RuntimeException("user offline"))
.when(webSocketMessageSender).sendObject(anyInt(), eq(1L), anyString(), any());
.when(webSocketSenderApi).sendObject(anyInt(), eq(1L), anyString(), any());
imWebSocketService.sendGroupMessageAsync(List.of(1L, 2L, 3L), dto);
// 2L 和 3L 也都被推送
verify(webSocketMessageSender).sendObject(anyInt(), eq(2L), anyString(), any());
verify(webSocketMessageSender).sendObject(anyInt(), eq(3L), anyString(), any());
verify(webSocketSenderApi).sendObject(anyInt(), eq(2L), anyString(), any());
verify(webSocketSenderApi).sendObject(anyInt(), eq(3L), anyString(), any());
}
}
@ -145,7 +145,7 @@ public class ImWebSocketServiceImplTest extends BaseMockitoUnitTest {
imWebSocketService.doSendGroupMessage(Collections.emptyList(), dto);
imWebSocketService.doSendGroupMessage(null, dto);
verifyNoInteractions(webSocketMessageSender);
verifyNoInteractions(webSocketSenderApi);
}
@Test
@ -154,11 +154,11 @@ public class ImWebSocketServiceImplTest extends BaseMockitoUnitTest {
imWebSocketService.doSendGroupMessage(Arrays.asList(1L, 2L, 1L, null), dto);
verify(webSocketMessageSender).sendObject(
verify(webSocketSenderApi).sendObject(
eq(UserTypeEnum.ADMIN.getValue()), eq(1L), eq(ImGroupMessageDTO.TYPE), eq(dto));
verify(webSocketMessageSender).sendObject(
verify(webSocketSenderApi).sendObject(
eq(UserTypeEnum.ADMIN.getValue()), eq(2L), eq(ImGroupMessageDTO.TYPE), eq(dto));
verifyNoMoreInteractions(webSocketMessageSender);
verifyNoMoreInteractions(webSocketSenderApi);
}
@Test
@ -170,12 +170,12 @@ public class ImWebSocketServiceImplTest extends BaseMockitoUnitTest {
// 准备：sender 抛异常
ImPrivateMessageDTO dto = new ImPrivateMessageDTO().setSenderId(1L).setReceiverId(2L);
doThrow(new RuntimeException("user offline"))
.when(webSocketMessageSender).sendObject(anyInt(), anyLong(), anyString(), any());
.when(webSocketSenderApi).sendObject(anyInt(), anyLong(), anyString(), any());
// 调用：异常应被吞掉，不向上抛
imWebSocketService.sendPrivateMessageAsync(2L, dto);
verify(webSocketMessageSender).sendObject(anyInt(), eq(2L), anyString(), any());
verify(webSocketSenderApi).sendObject(anyInt(), eq(2L), anyString(), any());
}
}
@ -190,7 +190,7 @@ public class ImWebSocketServiceImplTest extends BaseMockitoUnitTest {
imWebSocketService.sendGroupMessageAsync(42L, dto);
verify(webSocketMessageSender).sendObject(
verify(webSocketSenderApi).sendObject(
eq(UserTypeEnum.ADMIN.getValue()), eq(42L), eq(ImGroupMessageDTO.TYPE), eq(dto));
}
}

3
yudao-server/src/main/resources/application-local.yaml
View File

@ -164,6 +164,7 @@ logging:
cn.iocoder.yudao.module.iot.dal.mysql: debug
cn.iocoder.yudao.module.iot.dal.tdengine: DEBUG
cn.iocoder.yudao.module.ai.dal.mysql: debug
cn.iocoder.yudao.module.im.dal.mysql: debug
org.springframework.context.support.PostProcessorRegistrationDelegate: ERROR # TODO 芋艿：先禁用，Spring Boot 3.X 存在部分错误的 WARN 提示
debug: false
@ -253,4 +254,4 @@ justauth:
--- #################### iot相关配置 TODO 芋艿【IOT】：再瞅瞅 ####################
pf4j:
# pluginsDir: /tmp/
pluginsDir: ../plugins
pluginsDir: ../plugins