xieyinlu/ruoyi-ai
1
0
Fork 0
mirror of https://gitcode.com/ageerle/ruoyi-ai.git synced 2026-04-04 15:27:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

敏感数据处理

Browse Source
This commit is contained in:
wzx
2025-03-18 17:11:35 +08:00
parent 5e98b26903
commit 9423051f98
2 changed files with 29 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/controller/system/SysModelController.java
View File

@@ -25,12 +25,14 @@ import org.ruoyi.system.domain.vo.SysUserVo;
import org.ruoyi.system.service.ISysModelService;
import org.ruoyi.system.service.ISysPackagePlanService;
import org.ruoyi.system.service.ISysUserService;
import org.ruoyi.system.util.DesensitizationUtil;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;
/**
* 系统模型
@@ -50,7 +52,6 @@ public class SysModelController extends BaseController {
private final ISysUserService userService;
/**
* 查询系统模型列表 - 全部
*/
@@ -82,6 +83,14 @@ public class SysModelController extends BaseController {
List<String> array = new ArrayList<>(Arrays.asList(sysPackagePlanVo.getPlanDetail().split(",")));
sysModelVos.removeIf(model -> !array.contains(model.getModelName()));
}
sysModelVos.stream().map(vo -> {
String maskedApiHost = DesensitizationUtil.maskData(vo.getApiHost());
String maskedApiKey = DesensitizationUtil.maskData(vo.getApiKey());
vo.setApiHost(maskedApiHost);
vo.setApiKey(maskedApiKey);
return vo;
})
.collect(Collectors.toList());
return R.ok(sysModelVos);
}

19
ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/util/DesensitizationUtil.java Normal file
View File

@@ -0,0 +1,19 @@
package org.ruoyi.system.util;
public class DesensitizationUtil {
public static String maskData(String data) {
if (data == null || data.length() <= 4) {
return data;
}
int start = 2;
int end = data.length() - 2;
StringBuilder masked = new StringBuilder();
masked.append(data, 0, start);
for (int i = start; i < end; i++) {
masked.append('*');
}
masked.append(data.substring(end));
return masked.toString();
}
}