zhouhaibin
5 months ago
153 changed files with 2279 additions and 1018 deletions
@ -0,0 +1,36 @@ |
|||||
|
<?xml version="1.0" encoding="UTF-8"?> |
||||
|
<project xmlns="http://maven.apache.org/POM/4.0.0" |
||||
|
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" |
||||
|
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd"> |
||||
|
<parent> |
||||
|
<groupId>org.dromara</groupId> |
||||
|
<artifactId>ruoyi-common</artifactId> |
||||
|
<version>${revision}</version> |
||||
|
</parent> |
||||
|
<modelVersion>4.0.0</modelVersion> |
||||
|
|
||||
|
<artifactId>ruoyi-common-sse</artifactId> |
||||
|
|
||||
|
<description> |
||||
|
ruoyi-common-sse 模块 |
||||
|
</description> |
||||
|
|
||||
|
<dependencies> |
||||
|
<dependency> |
||||
|
<groupId>org.dromara</groupId> |
||||
|
<artifactId>ruoyi-common-core</artifactId> |
||||
|
</dependency> |
||||
|
<dependency> |
||||
|
<groupId>org.dromara</groupId> |
||||
|
<artifactId>ruoyi-common-redis</artifactId> |
||||
|
</dependency> |
||||
|
<dependency> |
||||
|
<groupId>org.dromara</groupId> |
||||
|
<artifactId>ruoyi-common-satoken</artifactId> |
||||
|
</dependency> |
||||
|
<dependency> |
||||
|
<groupId>org.dromara</groupId> |
||||
|
<artifactId>ruoyi-common-json</artifactId> |
||||
|
</dependency> |
||||
|
</dependencies> |
||||
|
</project> |
||||
@ -0,0 +1,36 @@ |
|||||
|
package org.dromara.common.sse.config; |
||||
|
|
||||
|
import org.dromara.common.sse.controller.SseController; |
||||
|
import org.dromara.common.sse.core.SseEmitterManager; |
||||
|
import org.dromara.common.sse.listener.SseTopicListener; |
||||
|
import org.springframework.boot.autoconfigure.AutoConfiguration; |
||||
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; |
||||
|
import org.springframework.boot.context.properties.EnableConfigurationProperties; |
||||
|
import org.springframework.context.annotation.Bean; |
||||
|
|
||||
|
/** |
||||
|
* SSE 自动装配 |
||||
|
* |
||||
|
* @author Lion Li |
||||
|
*/ |
||||
|
@AutoConfiguration |
||||
|
@ConditionalOnProperty(value = "sse.enabled", havingValue = "true") |
||||
|
@EnableConfigurationProperties(SseProperties.class) |
||||
|
public class SseAutoConfiguration { |
||||
|
|
||||
|
@Bean |
||||
|
public SseEmitterManager sseEmitterManager() { |
||||
|
return new SseEmitterManager(); |
||||
|
} |
||||
|
|
||||
|
@Bean |
||||
|
public SseTopicListener sseTopicListener() { |
||||
|
return new SseTopicListener(); |
||||
|
} |
||||
|
|
||||
|
@Bean |
||||
|
public SseController sseController(SseEmitterManager sseEmitterManager) { |
||||
|
return new SseController(sseEmitterManager); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,21 @@ |
|||||
|
package org.dromara.common.sse.config; |
||||
|
|
||||
|
import lombok.Data; |
||||
|
import org.springframework.boot.context.properties.ConfigurationProperties; |
||||
|
|
||||
|
/** |
||||
|
* SSE 配置项 |
||||
|
* |
||||
|
* @author Lion Li |
||||
|
*/ |
||||
|
@Data |
||||
|
@ConfigurationProperties("sse") |
||||
|
public class SseProperties { |
||||
|
|
||||
|
private Boolean enabled; |
||||
|
|
||||
|
/** |
||||
|
* 路径 |
||||
|
*/ |
||||
|
private String path; |
||||
|
} |
||||
@ -0,0 +1,87 @@ |
|||||
|
package org.dromara.common.sse.controller; |
||||
|
|
||||
|
import cn.dev33.satoken.annotation.SaIgnore; |
||||
|
import cn.dev33.satoken.stp.StpUtil; |
||||
|
import lombok.RequiredArgsConstructor; |
||||
|
import org.dromara.common.core.domain.R; |
||||
|
import org.dromara.common.satoken.utils.LoginHelper; |
||||
|
import org.dromara.common.sse.core.SseEmitterManager; |
||||
|
import org.dromara.common.sse.dto.SseMessageDto; |
||||
|
import org.springframework.beans.factory.DisposableBean; |
||||
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; |
||||
|
import org.springframework.http.MediaType; |
||||
|
import org.springframework.web.bind.annotation.GetMapping; |
||||
|
import org.springframework.web.bind.annotation.RestController; |
||||
|
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter; |
||||
|
|
||||
|
import java.util.List; |
||||
|
|
||||
|
/** |
||||
|
* SSE 控制器 |
||||
|
* |
||||
|
* @author Lion Li |
||||
|
*/ |
||||
|
@RestController |
||||
|
@ConditionalOnProperty(value = "sse.enabled", havingValue = "true") |
||||
|
@RequiredArgsConstructor |
||||
|
public class SseController implements DisposableBean { |
||||
|
|
||||
|
private final SseEmitterManager sseEmitterManager; |
||||
|
|
||||
|
/** |
||||
|
* 建立 SSE 连接 |
||||
|
*/ |
||||
|
@GetMapping(value = "${sse.path}", produces = MediaType.TEXT_EVENT_STREAM_VALUE) |
||||
|
public SseEmitter connect() { |
||||
|
String tokenValue = StpUtil.getTokenValue(); |
||||
|
Long userId = LoginHelper.getUserId(); |
||||
|
return sseEmitterManager.connect(userId, tokenValue); |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 关闭 SSE 连接 |
||||
|
*/ |
||||
|
@SaIgnore |
||||
|
@GetMapping(value = "${sse.path}/close") |
||||
|
public R<Void> close() { |
||||
|
String tokenValue = StpUtil.getTokenValue(); |
||||
|
Long userId = LoginHelper.getUserId(); |
||||
|
sseEmitterManager.disconnect(userId, tokenValue); |
||||
|
return R.ok(); |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 向特定用户发送消息 |
||||
|
* |
||||
|
* @param userId 目标用户的 ID |
||||
|
* @param msg 要发送的消息内容 |
||||
|
*/ |
||||
|
@GetMapping(value = "${sse.path}/send") |
||||
|
public R<Void> send(Long userId, String msg) { |
||||
|
SseMessageDto dto = new SseMessageDto(); |
||||
|
dto.setUserIds(List.of(userId)); |
||||
|
dto.setMessage(msg); |
||||
|
sseEmitterManager.publishMessage(dto); |
||||
|
return R.ok(); |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 向所有用户发送消息 |
||||
|
* |
||||
|
* @param msg 要发送的消息内容 |
||||
|
*/ |
||||
|
@GetMapping(value = "${sse.path}/sendAll") |
||||
|
public R<Void> send(String msg) { |
||||
|
sseEmitterManager.publishAll(msg); |
||||
|
return R.ok(); |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 清理资源。此方法目前不执行任何操作，但避免因未实现而导致错误 |
||||
|
*/ |
||||
|
@Override |
||||
|
public void destroy() throws Exception { |
||||
|
// 销毁时不需要做什么 此方法避免无用操作报错
|
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,160 @@ |
|||||
|
package org.dromara.common.sse.core; |
||||
|
|
||||
|
import cn.hutool.core.collection.CollUtil; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.dromara.common.redis.utils.RedisUtils; |
||||
|
import org.dromara.common.sse.dto.SseMessageDto; |
||||
|
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter; |
||||
|
|
||||
|
import java.io.IOException; |
||||
|
import java.util.ArrayList; |
||||
|
import java.util.List; |
||||
|
import java.util.Map; |
||||
|
import java.util.concurrent.ConcurrentHashMap; |
||||
|
import java.util.function.Consumer; |
||||
|
|
||||
|
/** |
||||
|
* 管理 Server-Sent Events (SSE) 连接 |
||||
|
* |
||||
|
* @author Lion Li |
||||
|
*/ |
||||
|
@Slf4j |
||||
|
public class SseEmitterManager { |
||||
|
|
||||
|
/** |
||||
|
* 订阅的频道 |
||||
|
*/ |
||||
|
private final static String SSE_TOPIC = "global:sse"; |
||||
|
|
||||
|
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) { |
||||
|
// 从 USER_TOKEN_EMITTERS 中获取或创建当前用户的 SseEmitter 映射表（ConcurrentHashMap）
|
||||
|
// 每个用户可以有多个 SSE 连接，通过 token 进行区分
|
||||
|
Map<String, SseEmitter> emitters = USER_TOKEN_EMITTERS.computeIfAbsent(userId, k -> new ConcurrentHashMap<>()); |
||||
|
|
||||
|
// 创建一个新的 SseEmitter 实例，超时时间设置为 0 表示无限制
|
||||
|
SseEmitter emitter = new SseEmitter(0L); |
||||
|
|
||||
|
emitters.put(token, emitter); |
||||
|
|
||||
|
// 当 emitter 完成、超时或发生错误时，从映射表中移除对应的 token
|
||||
|
emitter.onCompletion(() -> emitters.remove(token)); |
||||
|
emitter.onTimeout(() -> emitters.remove(token)); |
||||
|
emitter.onError((e) -> emitters.remove(token)); |
||||
|
|
||||
|
try { |
||||
|
// 向客户端发送一条连接成功的事件
|
||||
|
emitter.send(SseEmitter.event().comment("connected")); |
||||
|
} catch (IOException e) { |
||||
|
// 如果发送消息失败，则从映射表中移除 emitter
|
||||
|
emitters.remove(token); |
||||
|
} |
||||
|
return emitter; |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 断开指定用户的 SSE 连接 |
||||
|
* |
||||
|
* @param userId 用户的唯一标识符，用于区分不同用户的连接 |
||||
|
* @param token 用户的唯一令牌，用于识别具体的连接 |
||||
|
*/ |
||||
|
public void disconnect(Long userId, String token) { |
||||
|
Map<String, SseEmitter> emitters = USER_TOKEN_EMITTERS.get(userId); |
||||
|
if (emitters != null) { |
||||
|
try { |
||||
|
emitters.get(token).send(SseEmitter.event().comment("disconnected")); |
||||
|
} catch (Exception ignore) { |
||||
|
} |
||||
|
emitters.remove(token); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 订阅SSE消息主题，并提供一个消费者函数来处理接收到的消息 |
||||
|
* |
||||
|
* @param consumer 处理SSE消息的消费者函数 |
||||
|
*/ |
||||
|
public void subscribeMessage(Consumer<SseMessageDto> consumer) { |
||||
|
RedisUtils.subscribe(SSE_TOPIC, SseMessageDto.class, consumer); |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 向指定的用户会话发送消息 |
||||
|
* |
||||
|
* @param userId 要发送消息的用户id |
||||
|
* @param message 要发送的消息内容 |
||||
|
*/ |
||||
|
public void sendMessage(Long userId, String message) { |
||||
|
Map<String, SseEmitter> emitters = USER_TOKEN_EMITTERS.get(userId); |
||||
|
if (emitters != null) { |
||||
|
for (Map.Entry<String, SseEmitter> entry : emitters.entrySet()) { |
||||
|
try { |
||||
|
entry.getValue().send(SseEmitter.event() |
||||
|
.name("message") |
||||
|
.data(message)); |
||||
|
} catch (Exception e) { |
||||
|
emitters.remove(entry.getKey()); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 本机全用户会话发送消息 |
||||
|
* |
||||
|
* @param message 要发送的消息内容 |
||||
|
*/ |
||||
|
public void sendMessage(String message) { |
||||
|
for (Long userId : USER_TOKEN_EMITTERS.keySet()) { |
||||
|
sendMessage(userId, message); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 发布SSE订阅消息 |
||||
|
* |
||||
|
* @param sseMessageDto 要发布的SSE消息对象 |
||||
|
*/ |
||||
|
public void publishMessage(SseMessageDto sseMessageDto) { |
||||
|
List<Long> unsentUserIds = new ArrayList<>(); |
||||
|
// 当前服务内用户,直接发送消息
|
||||
|
for (Long userId : sseMessageDto.getUserIds()) { |
||||
|
if (USER_TOKEN_EMITTERS.containsKey(userId)) { |
||||
|
sendMessage(userId, sseMessageDto.getMessage()); |
||||
|
continue; |
||||
|
} |
||||
|
unsentUserIds.add(userId); |
||||
|
} |
||||
|
// 不在当前服务内用户,发布订阅消息
|
||||
|
if (CollUtil.isNotEmpty(unsentUserIds)) { |
||||
|
SseMessageDto broadcastMessage = new SseMessageDto(); |
||||
|
broadcastMessage.setMessage(sseMessageDto.getMessage()); |
||||
|
broadcastMessage.setUserIds(unsentUserIds); |
||||
|
RedisUtils.publish(SSE_TOPIC, broadcastMessage, consumer -> { |
||||
|
log.info("SSE发送主题订阅消息topic:{} session keys:{} message:{}", |
||||
|
SSE_TOPIC, unsentUserIds, sseMessageDto.getMessage()); |
||||
|
}); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 向所有的用户发布订阅的消息(群发) |
||||
|
* |
||||
|
* @param message 要发布的消息内容 |
||||
|
*/ |
||||
|
public void publishAll(String message) { |
||||
|
SseMessageDto broadcastMessage = new SseMessageDto(); |
||||
|
broadcastMessage.setMessage(message); |
||||
|
RedisUtils.publish(SSE_TOPIC, broadcastMessage, consumer -> { |
||||
|
log.info("SSE发送主题订阅消息topic:{} message:{}", SSE_TOPIC, message); |
||||
|
}); |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,29 @@ |
|||||
|
package org.dromara.common.sse.dto; |
||||
|
|
||||
|
import lombok.Data; |
||||
|
|
||||
|
import java.io.Serial; |
||||
|
import java.io.Serializable; |
||||
|
import java.util.List; |
||||
|
|
||||
|
/** |
||||
|
* 消息的dto |
||||
|
* |
||||
|
* @author zendwang |
||||
|
*/ |
||||
|
@Data |
||||
|
public class SseMessageDto implements Serializable { |
||||
|
|
||||
|
@Serial |
||||
|
private static final long serialVersionUID = 1L; |
||||
|
|
||||
|
/** |
||||
|
* 需要推送到的session key 列表 |
||||
|
*/ |
||||
|
private List<Long> userIds; |
||||
|
|
||||
|
/** |
||||
|
* 需要发送的消息 |
||||
|
*/ |
||||
|
private String message; |
||||
|
} |
||||
@ -0,0 +1,48 @@ |
|||||
|
package org.dromara.common.sse.listener; |
||||
|
|
||||
|
import cn.hutool.core.collection.CollUtil; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.dromara.common.sse.core.SseEmitterManager; |
||||
|
import org.springframework.beans.factory.annotation.Autowired; |
||||
|
import org.springframework.boot.ApplicationArguments; |
||||
|
import org.springframework.boot.ApplicationRunner; |
||||
|
import org.springframework.core.Ordered; |
||||
|
|
||||
|
/** |
||||
|
* SSE 主题订阅监听器 |
||||
|
* |
||||
|
* @author Lion Li |
||||
|
*/ |
||||
|
@Slf4j |
||||
|
public class SseTopicListener implements ApplicationRunner, Ordered { |
||||
|
|
||||
|
@Autowired |
||||
|
private SseEmitterManager sseEmitterManager; |
||||
|
|
||||
|
/** |
||||
|
* 在Spring Boot应用程序启动时初始化SSE主题订阅监听器 |
||||
|
* |
||||
|
* @param args 应用程序参数 |
||||
|
* @throws Exception 初始化过程中可能抛出的异常 |
||||
|
*/ |
||||
|
@Override |
||||
|
public void run(ApplicationArguments args) throws Exception { |
||||
|
sseEmitterManager.subscribeMessage((message) -> { |
||||
|
log.info("SSE主题订阅收到消息session keys={} message={}", message.getUserIds(), message.getMessage()); |
||||
|
// 如果key不为空就按照key发消息 如果为空就群发
|
||||
|
if (CollUtil.isNotEmpty(message.getUserIds())) { |
||||
|
message.getUserIds().forEach(key -> { |
||||
|
sseEmitterManager.sendMessage(key, message.getMessage()); |
||||
|
}); |
||||
|
} else { |
||||
|
sseEmitterManager.sendMessage(message.getMessage()); |
||||
|
} |
||||
|
}); |
||||
|
log.info("初始化SSE主题订阅监听器成功"); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public int getOrder() { |
||||
|
return -1; |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,58 @@ |
|||||
|
package org.dromara.common.sse.utils; |
||||
|
|
||||
|
import lombok.AccessLevel; |
||||
|
import lombok.NoArgsConstructor; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.dromara.common.core.utils.SpringUtils; |
||||
|
import org.dromara.common.sse.core.SseEmitterManager; |
||||
|
import org.dromara.common.sse.dto.SseMessageDto; |
||||
|
|
||||
|
/** |
||||
|
* SSE工具类 |
||||
|
* |
||||
|
* @author Lion Li |
||||
|
*/ |
||||
|
@Slf4j |
||||
|
@NoArgsConstructor(access = AccessLevel.PRIVATE) |
||||
|
public class SseMessageUtils { |
||||
|
|
||||
|
private final static SseEmitterManager MANAGER = SpringUtils.getBean(SseEmitterManager.class); |
||||
|
|
||||
|
/** |
||||
|
* 向指定的WebSocket会话发送消息 |
||||
|
* |
||||
|
* @param userId 要发送消息的用户id |
||||
|
* @param message 要发送的消息内容 |
||||
|
*/ |
||||
|
public static void sendMessage(Long userId, String message) { |
||||
|
MANAGER.sendMessage(userId, message); |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 本机全用户会话发送消息 |
||||
|
* |
||||
|
* @param message 要发送的消息内容 |
||||
|
*/ |
||||
|
public static void sendMessage(String message) { |
||||
|
MANAGER.sendMessage(message); |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 发布SSE订阅消息 |
||||
|
* |
||||
|
* @param sseMessageDto 要发布的SSE消息对象 |
||||
|
*/ |
||||
|
public static void publishMessage(SseMessageDto sseMessageDto) { |
||||
|
MANAGER.publishMessage(sseMessageDto); |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 向所有的用户发布订阅的消息(群发) |
||||
|
* |
||||
|
* @param message 要发布的消息内容 |
||||
|
*/ |
||||
|
public static void publishAll(String message) { |
||||
|
MANAGER.publishAll(message); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1 @@ |
|||||
|
org.dromara.common.sse.config.SseAutoConfiguration |
||||
@ -0,0 +1,64 @@ |
|||||
|
package com.aizuda.snailjob.server.starter.filter; |
||||
|
|
||||
|
import jakarta.servlet.*; |
||||
|
import jakarta.servlet.http.HttpServletRequest; |
||||
|
import jakarta.servlet.http.HttpServletResponse; |
||||
|
|
||||
|
import java.io.IOException; |
||||
|
import java.nio.charset.StandardCharsets; |
||||
|
import java.util.Base64; |
||||
|
|
||||
|
public class ActuatorAuthFilter implements Filter { |
||||
|
|
||||
|
private final String username; |
||||
|
private final String password; |
||||
|
|
||||
|
public ActuatorAuthFilter(String username, String password) { |
||||
|
this.username = username; |
||||
|
this.password = password; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException { |
||||
|
HttpServletRequest request = (HttpServletRequest) servletRequest; |
||||
|
HttpServletResponse response = (HttpServletResponse) servletResponse; |
||||
|
|
||||
|
// 获取 Authorization 头
|
||||
|
String authHeader = request.getHeader("Authorization"); |
||||
|
|
||||
|
if (authHeader == null || !authHeader.startsWith("Basic ")) { |
||||
|
// 如果没有提供 Authorization 或者格式不对，则返回 401
|
||||
|
response.setHeader("WWW-Authenticate", "Basic realm=\"realm\""); |
||||
|
response.sendError(HttpServletResponse.SC_UNAUTHORIZED, "Unauthorized"); |
||||
|
return; |
||||
|
} |
||||
|
|
||||
|
// 解码 Base64 编码的用户名和密码
|
||||
|
String base64Credentials = authHeader.substring("Basic ".length()); |
||||
|
byte[] credDecoded = Base64.getDecoder().decode(base64Credentials); |
||||
|
String credentials = new String(credDecoded, StandardCharsets.UTF_8); |
||||
|
String[] split = credentials.split(":"); |
||||
|
if (split.length != 2) { |
||||
|
response.setHeader("WWW-Authenticate", "Basic realm=\"realm\""); |
||||
|
response.sendError(HttpServletResponse.SC_UNAUTHORIZED, "Unauthorized"); |
||||
|
return; |
||||
|
} |
||||
|
// 验证用户名和密码
|
||||
|
if (!username.equals(split[0]) && password.equals(split[1])) { |
||||
|
response.setHeader("WWW-Authenticate", "Basic realm=\"realm\""); |
||||
|
response.sendError(HttpServletResponse.SC_UNAUTHORIZED, "Unauthorized"); |
||||
|
return; |
||||
|
} |
||||
|
// 如果认证成功，继续处理请求
|
||||
|
filterChain.doFilter(request, response); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void init(FilterConfig filterConfig) { |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void destroy() { |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,29 @@ |
|||||
|
package com.aizuda.snailjob.server.starter.filter; |
||||
|
|
||||
|
import org.springframework.beans.factory.annotation.Value; |
||||
|
import org.springframework.boot.web.servlet.FilterRegistrationBean; |
||||
|
import org.springframework.context.annotation.Bean; |
||||
|
import org.springframework.context.annotation.Configuration; |
||||
|
|
||||
|
/** |
||||
|
* 权限安全配置 |
||||
|
* |
||||
|
* @author Lion Li |
||||
|
*/ |
||||
|
@Configuration |
||||
|
public class SecurityConfig { |
||||
|
|
||||
|
@Value("${spring.boot.admin.client.username}") |
||||
|
private String username; |
||||
|
@Value("${spring.boot.admin.client.password}") |
||||
|
private String password; |
||||
|
|
||||
|
@Bean |
||||
|
public FilterRegistrationBean<ActuatorAuthFilter> actuatorFilterRegistrationBean() { |
||||
|
FilterRegistrationBean<ActuatorAuthFilter> registrationBean = new FilterRegistrationBean<>(); |
||||
|
registrationBean.setFilter(new ActuatorAuthFilter(username, password)); |
||||
|
registrationBean.addUrlPatterns("/actuator", "/actuator/**"); |
||||
|
return registrationBean; |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,105 @@ |
|||||
|
package org.dromara.generator.config; |
||||
|
|
||||
|
import com.baomidou.dynamic.datasource.DynamicRoutingDataSource; |
||||
|
import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.anyline.data.datasource.DataSourceMonitor; |
||||
|
import org.anyline.data.runtime.DataRuntime; |
||||
|
import org.anyline.util.ConfigTable; |
||||
|
import org.springframework.jdbc.core.JdbcTemplate; |
||||
|
import org.springframework.jdbc.datasource.DataSourceUtils; |
||||
|
import org.springframework.stereotype.Component; |
||||
|
|
||||
|
import javax.sql.DataSource; |
||||
|
import java.sql.Connection; |
||||
|
import java.sql.DatabaseMetaData; |
||||
|
import java.util.HashMap; |
||||
|
import java.util.Map; |
||||
|
|
||||
|
/** |
||||
|
* anyline 适配 动态数据源改造 |
||||
|
* |
||||
|
* @author Lion Li |
||||
|
*/ |
||||
|
@Slf4j |
||||
|
@Component |
||||
|
public class MyBatisDataSourceMonitor implements DataSourceMonitor { |
||||
|
|
||||
|
public MyBatisDataSourceMonitor() { |
||||
|
// 调整执行模式为自定义
|
||||
|
ConfigTable.KEEP_ADAPTER = 2; |
||||
|
// 禁用缓存
|
||||
|
ConfigTable.METADATA_CACHE_SCOPE = 0; |
||||
|
} |
||||
|
|
||||
|
private final Map<String, String> features = new HashMap<>(); |
||||
|
|
||||
|
/** |
||||
|
* 数据源特征 用来定准 adapter 包含数据库或JDBC协议关键字<br/> |
||||
|
* 一般会通过 产品名_url 合成 如果返回null 上层方法会通过driver_产品名_url合成 |
||||
|
* |
||||
|
* @param datasource 数据源 |
||||
|
* @return String 返回null由上层自动提取 |
||||
|
*/ |
||||
|
@Override |
||||
|
public String feature(DataRuntime runtime, Object datasource) { |
||||
|
String feature = null; |
||||
|
if (datasource instanceof JdbcTemplate jdbc) { |
||||
|
DataSource ds = jdbc.getDataSource(); |
||||
|
if (ds instanceof DynamicRoutingDataSource) { |
||||
|
String key = DynamicDataSourceContextHolder.peek(); |
||||
|
feature = features.get(key); |
||||
|
if (null == feature) { |
||||
|
Connection con = null; |
||||
|
try { |
||||
|
con = DataSourceUtils.getConnection(ds); |
||||
|
DatabaseMetaData meta = con.getMetaData(); |
||||
|
String url = meta.getURL(); |
||||
|
feature = meta.getDatabaseProductName().toLowerCase().replace(" ", "") + "_" + url; |
||||
|
features.put(key, feature); |
||||
|
} catch (Exception e) { |
||||
|
log.error(e.getMessage(), e); |
||||
|
} finally { |
||||
|
if (null != con && !DataSourceUtils.isConnectionTransactional(con, ds)) { |
||||
|
DataSourceUtils.releaseConnection(con, ds); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
return feature; |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 数据源唯一标识 如果不实现则默认feature |
||||
|
* @param datasource 数据源 |
||||
|
* @return String 返回null由上层自动提取 |
||||
|
*/ |
||||
|
@Override |
||||
|
public String key(DataRuntime runtime, Object datasource) { |
||||
|
if(datasource instanceof JdbcTemplate jdbc){ |
||||
|
DataSource ds = jdbc.getDataSource(); |
||||
|
if(ds instanceof DynamicRoutingDataSource){ |
||||
|
return DynamicDataSourceContextHolder.peek(); |
||||
|
} |
||||
|
} |
||||
|
return runtime.getKey(); |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* ConfigTable.KEEP_ADAPTER=2 : 根据当前接口判断是否保持同一个数据源绑定同一个adapter<br/> |
||||
|
* DynamicRoutingDataSource类型的返回false,因为同一个DynamicRoutingDataSource可能对应多类数据库, 如果项目中只有一种数据库 应该直接返回true |
||||
|
* |
||||
|
* @param datasource 数据源 |
||||
|
* @return boolean |
||||
|
*/ |
||||
|
@Override |
||||
|
public boolean keepAdapter(DataRuntime runtime, Object datasource) { |
||||
|
if (datasource instanceof JdbcTemplate jdbc) { |
||||
|
DataSource ds = jdbc.getDataSource(); |
||||
|
return !(ds instanceof DynamicRoutingDataSource); |
||||
|
} |
||||
|
return true; |
||||
|
} |
||||
|
|
||||
|
} |
||||
Some files were not shown because too many files changed in this diff
Loading…
Reference in new issue