Merge remote-tracking branch 'origin/dev' into warm-flow-future

This commit is contained in:
疯狂的狮子Li 2024-08-12 13:40:11 +08:00
commit 7be5d23451
5 changed files with 55 additions and 5 deletions

View File

@ -47,11 +47,11 @@
<aws.sdk.version>2.25.15</aws.sdk.version> <aws.sdk.version>2.25.15</aws.sdk.version>
<aws.crt.version>0.29.13</aws.crt.version> <aws.crt.version>0.29.13</aws.crt.version>
<!-- SMS 配置 --> <!-- SMS 配置 -->
<sms4j.version>3.3.1</sms4j.version> <sms4j.version>3.3.2</sms4j.version>
<!-- 限制框架中的fastjson版本 --> <!-- 限制框架中的fastjson版本 -->
<fastjson.version>1.2.83</fastjson.version> <fastjson.version>1.2.83</fastjson.version>
<!-- 面向运行时的D-ORM依赖 --> <!-- 面向运行时的D-ORM依赖 -->
<anyline.version>8.7.2-20240728</anyline.version> <anyline.version>8.7.2-20240808</anyline.version>
<!--工作流配置--> <!--工作流配置-->
<warm-flow>1.2.3</warm-flow> <warm-flow>1.2.3</warm-flow>

View File

@ -16,6 +16,11 @@ import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
import java.util.List; import java.util.List;
/**
* SSE 控制器
*
* @author Lion Li
*/
@RestController @RestController
@ConditionalOnProperty(value = "sse.enabled", havingValue = "true") @ConditionalOnProperty(value = "sse.enabled", havingValue = "true")
@RequiredArgsConstructor @RequiredArgsConstructor
@ -23,6 +28,9 @@ public class SseController implements DisposableBean {
private final SseEmitterManager sseEmitterManager; private final SseEmitterManager sseEmitterManager;
/**
* 建立 SSE 连接
*/
@GetMapping(value = "${sse.path}", produces = MediaType.TEXT_EVENT_STREAM_VALUE) @GetMapping(value = "${sse.path}", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
public SseEmitter connect() { public SseEmitter connect() {
String tokenValue = StpUtil.getTokenValue(); String tokenValue = StpUtil.getTokenValue();
@ -30,6 +38,9 @@ public class SseController implements DisposableBean {
return sseEmitterManager.connect(userId, tokenValue); return sseEmitterManager.connect(userId, tokenValue);
} }
/**
* 关闭 SSE 连接
*/
@SaIgnore @SaIgnore
@GetMapping(value = "${sse.path}/close") @GetMapping(value = "${sse.path}/close")
public R<Void> close() { public R<Void> close() {
@ -39,6 +50,12 @@ public class SseController implements DisposableBean {
return R.ok(); return R.ok();
} }
/**
* 向特定用户发送消息
*
* @param userId 目标用户的 ID
* @param msg 要发送的消息内容
*/
@GetMapping(value = "${sse.path}/send") @GetMapping(value = "${sse.path}/send")
public R<Void> send(Long userId, String msg) { public R<Void> send(Long userId, String msg) {
SseMessageDto dto = new SseMessageDto(); SseMessageDto dto = new SseMessageDto();
@ -48,12 +65,20 @@ public class SseController implements DisposableBean {
return R.ok(); return R.ok();
} }
/**
* 向所有用户发送消息
*
* @param msg 要发送的消息内容
*/
@GetMapping(value = "${sse.path}/sendAll") @GetMapping(value = "${sse.path}/sendAll")
public R<Void> send(String msg) { public R<Void> send(String msg) {
sseEmitterManager.publishAll(msg); sseEmitterManager.publishAll(msg);
return R.ok(); return R.ok();
} }
/**
* 清理资源此方法目前不执行任何操作但避免因未实现而导致错误
*/
@Override @Override
public void destroy() throws Exception { public void destroy() throws Exception {
// 销毁时不需要做什么 此方法避免无用操作报错 // 销毁时不需要做什么 此方法避免无用操作报错

View File

@ -13,8 +13,14 @@ import java.util.Map;
import java.util.concurrent.ConcurrentHashMap; import java.util.concurrent.ConcurrentHashMap;
import java.util.function.Consumer; import java.util.function.Consumer;
/**
* 管理 Server-Sent Events (SSE) 连接
*
* @author Lion Li
*/
@Slf4j @Slf4j
public class SseEmitterManager { public class SseEmitterManager {
/** /**
* 订阅的频道 * 订阅的频道
*/ */
@ -22,24 +28,44 @@ public class SseEmitterManager {
private final static Map<Long, Map<String, SseEmitter>> USER_TOKEN_EMITTERS = new ConcurrentHashMap<>(); private final static Map<Long, Map<String, SseEmitter>> USER_TOKEN_EMITTERS = new ConcurrentHashMap<>();
/**
* 建立与指定用户的 SSE 连接
*
* @param userId 用户的唯一标识符用于区分不同用户的连接
* @param token 用户的唯一令牌用于识别具体的连接
* @return 返回一个 SseEmitter 实例客户端可以通过该实例接收 SSE 事件
*/
public SseEmitter connect(Long userId, String token) { public SseEmitter connect(Long userId, String token) {
// USER_TOKEN_EMITTERS 中获取或创建当前用户的 SseEmitter 映射表ConcurrentHashMap
// 每个用户可以有多个 SSE 连接通过 token 进行区分
Map<String, SseEmitter> emitters = USER_TOKEN_EMITTERS.computeIfAbsent(userId, k -> new ConcurrentHashMap<>()); Map<String, SseEmitter> emitters = USER_TOKEN_EMITTERS.computeIfAbsent(userId, k -> new ConcurrentHashMap<>());
// 创建一个新的 SseEmitter 实例超时时间设置为 0 表示无限制
SseEmitter emitter = new SseEmitter(0L); SseEmitter emitter = new SseEmitter(0L);
emitters.put(token, emitter); emitters.put(token, emitter);
// emitter 完成超时或发生错误时从映射表中移除对应的 token
emitter.onCompletion(() -> emitters.remove(token)); emitter.onCompletion(() -> emitters.remove(token));
emitter.onTimeout(() -> emitters.remove(token)); emitter.onTimeout(() -> emitters.remove(token));
emitter.onError((e) -> emitters.remove(token)); emitter.onError((e) -> emitters.remove(token));
try { try {
// 向客户端发送一条连接成功的事件
emitter.send(SseEmitter.event().comment("connected")); emitter.send(SseEmitter.event().comment("connected"));
} catch (IOException e) { } catch (IOException e) {
// 如果发送消息失败则从映射表中移除 emitter
emitters.remove(token); emitters.remove(token);
} }
return emitter; return emitter;
} }
/**
* 断开指定用户的 SSE 连接
*
* @param userId 用户的唯一标识符用于区分不同用户的连接
* @param token 用户的唯一令牌用于识别具体的连接
*/
public void disconnect(Long userId, String token) { public void disconnect(Long userId, String token) {
Map<String, SseEmitter> emitters = USER_TOKEN_EMITTERS.get(userId); Map<String, SseEmitter> emitters = USER_TOKEN_EMITTERS.get(userId);
if (emitters != null) { if (emitters != null) {

View File

@ -8,7 +8,7 @@ import org.dromara.common.sse.core.SseEmitterManager;
import org.dromara.common.sse.dto.SseMessageDto; import org.dromara.common.sse.dto.SseMessageDto;
/** /**
* 工具类 * SSE工具类
* *
* @author Lion Li * @author Lion Li
*/ */

View File

@ -3,7 +3,6 @@ package org.dromara.generator.service;
import cn.hutool.core.collection.CollUtil; import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.io.IoUtil; import cn.hutool.core.io.IoUtil;
import cn.hutool.core.lang.Dict; import cn.hutool.core.lang.Dict;
import cn.hutool.core.util.ArrayUtil;
import cn.hutool.core.util.ObjectUtil; import cn.hutool.core.util.ObjectUtil;
import com.baomidou.dynamic.datasource.annotation.DS; import com.baomidou.dynamic.datasource.annotation.DS;
import com.baomidou.dynamic.datasource.annotation.DSTransactional; import com.baomidou.dynamic.datasource.annotation.DSTransactional;
@ -196,7 +195,7 @@ public class GenTableServiceImpl implements IGenTableService {
.filter(x -> !StringUtils.containsAnyIgnoreCase(x.getName(), TABLE_IGNORE)) .filter(x -> !StringUtils.containsAnyIgnoreCase(x.getName(), TABLE_IGNORE))
.filter(x -> tableNameSet.contains(x.getName())).toList(); .filter(x -> tableNameSet.contains(x.getName())).toList();
if (ArrayUtil.isEmpty(tableList)) { if (CollUtil.isEmpty(tableList)) {
return new ArrayList<>(); return new ArrayList<>();
} }
return tableList.stream().map(x -> { return tableList.stream().map(x -> {