添加wvp过滤
This commit is contained in:
parent
095d9d257f
commit
3ad0509eb0
|
|
@ -0,0 +1,97 @@
|
||||||
|
package com.ruoyi.gateway.filter;
|
||||||
|
|
||||||
|
import com.ruoyi.gateway.utils.WvpTokenClient;
|
||||||
|
import org.slf4j.Logger;
|
||||||
|
import org.slf4j.LoggerFactory;
|
||||||
|
import org.springframework.cloud.gateway.filter.GatewayFilter;
|
||||||
|
import org.springframework.cloud.gateway.filter.factory.AbstractGatewayFilterFactory;
|
||||||
|
import org.springframework.http.server.reactive.ServerHttpRequest;
|
||||||
|
import org.springframework.stereotype.Component;
|
||||||
|
import org.springframework.web.server.ServerWebExchange;
|
||||||
|
import reactor.core.publisher.Mono;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* WVP Access Token 过滤器
|
||||||
|
* 用于为 WVP 请求添加或替换 access-token 请求头
|
||||||
|
*/
|
||||||
|
@Component
|
||||||
|
public class WvpAccessTokenFilter extends AbstractGatewayFilterFactory<WvpAccessTokenFilter.Config> {
|
||||||
|
|
||||||
|
private static final Logger log = LoggerFactory.getLogger(WvpAccessTokenFilter.class);
|
||||||
|
|
||||||
|
private static String cachedAccessToken = null;
|
||||||
|
private static final Object LOCK = new Object();
|
||||||
|
|
||||||
|
public WvpAccessTokenFilter() {
|
||||||
|
super(Config.class);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public GatewayFilter apply(Config config) {
|
||||||
|
return (exchange, chain) -> {
|
||||||
|
try {
|
||||||
|
// 获取 access token
|
||||||
|
String accessToken = getAccessToken();
|
||||||
|
|
||||||
|
log.info("WVP 请求 - 原始路径: {}", exchange.getRequest().getURI().getPath());
|
||||||
|
log.info("WVP 请求 - 添加 access-token");
|
||||||
|
|
||||||
|
// 修改请求,添加或替换 access-token 请求头
|
||||||
|
ServerHttpRequest modifiedRequest = exchange.getRequest().mutate()
|
||||||
|
.header("access-token", accessToken)
|
||||||
|
.build();
|
||||||
|
|
||||||
|
// 创建新的 exchange
|
||||||
|
ServerWebExchange modifiedExchange = exchange.mutate()
|
||||||
|
.request(modifiedRequest)
|
||||||
|
.build();
|
||||||
|
|
||||||
|
return chain.filter(modifiedExchange).then(Mono.fromRunnable(() -> {
|
||||||
|
log.info("WVP 响应 - Status: {}", modifiedExchange.getResponse().getStatusCode());
|
||||||
|
}));
|
||||||
|
} catch (Exception e) {
|
||||||
|
log.error("获取 WVP access token 失败", e);
|
||||||
|
// 即使获取 token 失败,也继续转发请求
|
||||||
|
return chain.filter(exchange);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取 access token,使用缓存机制
|
||||||
|
* 如果缓存为空,则调用 WvpTokenClient.login 获取新的 token
|
||||||
|
*/
|
||||||
|
private String getAccessToken() throws Exception {
|
||||||
|
if (cachedAccessToken == null) {
|
||||||
|
synchronized (LOCK) {
|
||||||
|
if (cachedAccessToken == null) {
|
||||||
|
log.info("缓存中没有 access token,正在登录 WVP...");
|
||||||
|
WvpTokenClient.LoginResponse loginResponse = WvpTokenClient.login("admin", "admin");
|
||||||
|
if (loginResponse.getCode() == 0 && loginResponse.getData() != null) {
|
||||||
|
cachedAccessToken = loginResponse.getData().getAccessToken();
|
||||||
|
log.info("WVP 登录成功,获取到 access token");
|
||||||
|
} else {
|
||||||
|
throw new Exception("WVP 登录失败: " + loginResponse.getMsg());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return cachedAccessToken;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 清除缓存的 access token(可用于 token 过期时重新获取)
|
||||||
|
*/
|
||||||
|
public static void clearCachedToken() {
|
||||||
|
synchronized (LOCK) {
|
||||||
|
cachedAccessToken = null;
|
||||||
|
log.info("已清除缓存的 WVP access token");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 配置类
|
||||||
|
*/
|
||||||
|
public static class Config {
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,149 @@
|
||||||
|
package com.ruoyi.gateway.utils;
|
||||||
|
|
||||||
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
||||||
|
import org.slf4j.Logger;
|
||||||
|
import org.slf4j.LoggerFactory;
|
||||||
|
|
||||||
|
import java.io.BufferedReader;
|
||||||
|
import java.io.InputStreamReader;
|
||||||
|
import java.net.HttpURLConnection;
|
||||||
|
import java.net.URL;
|
||||||
|
import java.net.URLEncoder;
|
||||||
|
import java.nio.charset.StandardCharsets;
|
||||||
|
import java.security.MessageDigest;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* WVP Token 客户端
|
||||||
|
* 用于获取 WVP-PRO 的 access token
|
||||||
|
*/
|
||||||
|
public class WvpTokenClient {
|
||||||
|
|
||||||
|
private static final Logger log = LoggerFactory.getLogger(WvpTokenClient.class);
|
||||||
|
|
||||||
|
// WVP 服务地址 - 使用容器名称进行容器间通信
|
||||||
|
private static final String WVP_BASE_URL = "http://wvp-pro:18978";
|
||||||
|
|
||||||
|
private static final ObjectMapper objectMapper = new ObjectMapper();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* WVP登录响应对象
|
||||||
|
*/
|
||||||
|
public static class LoginResponse {
|
||||||
|
private int code;
|
||||||
|
private String msg;
|
||||||
|
private LoginData data;
|
||||||
|
|
||||||
|
public int getCode() {
|
||||||
|
return code;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setCode(int code) {
|
||||||
|
this.code = code;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getMsg() {
|
||||||
|
return msg;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setMsg(String msg) {
|
||||||
|
this.msg = msg;
|
||||||
|
}
|
||||||
|
|
||||||
|
public LoginData getData() {
|
||||||
|
return data;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setData(LoginData data) {
|
||||||
|
this.data = data;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 登录数据对象
|
||||||
|
*/
|
||||||
|
public static class LoginData {
|
||||||
|
private String accessToken;
|
||||||
|
private String username;
|
||||||
|
|
||||||
|
public String getAccessToken() {
|
||||||
|
return accessToken;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setAccessToken(String accessToken) {
|
||||||
|
this.accessToken = accessToken;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getUsername() {
|
||||||
|
return username;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setUsername(String username) {
|
||||||
|
this.username = username;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 计算MD5哈希值
|
||||||
|
*
|
||||||
|
* @param input 输入字符串
|
||||||
|
* @return MD5哈希值(小写)
|
||||||
|
* @throws Exception 如果计算失败
|
||||||
|
*/
|
||||||
|
private static String md5(String input) throws Exception {
|
||||||
|
MessageDigest md = MessageDigest.getInstance("MD5");
|
||||||
|
byte[] messageDigest = md.digest(input.getBytes(StandardCharsets.UTF_8));
|
||||||
|
StringBuilder hexString = new StringBuilder();
|
||||||
|
for (byte b : messageDigest) {
|
||||||
|
String hex = Integer.toHexString(0xff & b);
|
||||||
|
if (hex.length() == 1) {
|
||||||
|
hexString.append('0');
|
||||||
|
}
|
||||||
|
hexString.append(hex);
|
||||||
|
}
|
||||||
|
return hexString.toString();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* WVP用户登录
|
||||||
|
*
|
||||||
|
* @param username 用户名
|
||||||
|
* @param password 密码(明文)
|
||||||
|
* @return 登录响应对象
|
||||||
|
* @throws Exception 如果请求失败
|
||||||
|
*/
|
||||||
|
public static LoginResponse login(String username, String password) throws Exception {
|
||||||
|
// 计算密码的MD5值
|
||||||
|
String passwordMd5 = md5(password);
|
||||||
|
|
||||||
|
// 构建 URL 参数
|
||||||
|
StringBuilder urlBuilder = new StringBuilder(WVP_BASE_URL);
|
||||||
|
urlBuilder.append("/api/user/login?");
|
||||||
|
urlBuilder.append("username=").append(URLEncoder.encode(username, StandardCharsets.UTF_8.name()));
|
||||||
|
urlBuilder.append("&password=").append(passwordMd5);
|
||||||
|
|
||||||
|
// 创建 HTTP 连接
|
||||||
|
URL url = new URL(urlBuilder.toString());
|
||||||
|
HttpURLConnection connection = (HttpURLConnection) url.openConnection();
|
||||||
|
connection.setRequestMethod("GET");
|
||||||
|
connection.setConnectTimeout(10000);
|
||||||
|
connection.setReadTimeout(10000);
|
||||||
|
|
||||||
|
// 获取响应
|
||||||
|
int responseCode = connection.getResponseCode();
|
||||||
|
if (responseCode == HttpURLConnection.HTTP_OK) {
|
||||||
|
BufferedReader reader = new BufferedReader(
|
||||||
|
new InputStreamReader(connection.getInputStream(), StandardCharsets.UTF_8));
|
||||||
|
StringBuilder response = new StringBuilder();
|
||||||
|
String line;
|
||||||
|
while ((line = reader.readLine()) != null) {
|
||||||
|
response.append(line);
|
||||||
|
}
|
||||||
|
reader.close();
|
||||||
|
|
||||||
|
// 解析JSON响应
|
||||||
|
return objectMapper.readValue(response.toString(), LoginResponse.class);
|
||||||
|
} else {
|
||||||
|
throw new Exception("HTTP 请求失败,状态码: " + responseCode);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue