tcjlj/flvx
1
0
Fork 0
mirror of https://github.com/Sagit-chu/flvx.git synced 2026-10-07 10:16:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

增加日志输出

Browse Source
This commit is contained in:
qaq
2025-08-27 16:07:26 +08:00
parent ed098fd423
commit 59ddb9bbbb
1 changed files with 18 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
springboot-backend/src/main/java/com/admin/controller/FlowController.java
+18 -23
View File
@@ -140,6 +140,7 @@ public class FlowController extends BaseController {
public String uploadFlowData(@RequestBody String rawData, String secret) { public String uploadFlowData(@RequestBody String rawData, String secret) {
// 1. 验证节点权限 // 1. 验证节点权限
if (!isValidNode(secret)) { if (!isValidNode(secret)) {
log.info("节点密钥验证失败==>" + secret);
return SUCCESS_RESPONSE; return SUCCESS_RESPONSE;
} }
@@ -149,6 +150,7 @@ public class FlowController extends BaseController {
// 3. 解析为FlowDto列表 // 3. 解析为FlowDto列表
FlowDto flowDataList = JSONObject.parseObject(decryptedData, FlowDto.class); FlowDto flowDataList = JSONObject.parseObject(decryptedData, FlowDto.class);
if (Objects.equals(flowDataList.getN(), "web_api")) { if (Objects.equals(flowDataList.getN(), "web_api")) {
System.out.println("不处理web_api==>" + flowDataList.getN());
return SUCCESS_RESPONSE; return SUCCESS_RESPONSE;
} }
@@ -178,27 +180,17 @@ public class FlowController extends BaseController {
throw new IllegalArgumentException("数据不能为空"); throw new IllegalArgumentException("数据不能为空");
} }
try { EncryptedMessage encryptedMessage = JSON.parseObject(rawData, EncryptedMessage.class);
// 尝试解析为加密消息格式 if (encryptedMessage.isEncrypted() && encryptedMessage.getData() != null) {
EncryptedMessage encryptedMessage = JSON.parseObject(rawData, EncryptedMessage.class); // 获取或创建加密器
AESCrypto crypto = getOrCreateCrypto(secret);
if (encryptedMessage.isEncrypted() && encryptedMessage.getData() != null) { if (crypto == null) {
// 获取或创建加密器 log.info("⚠️ 收到加密消息但无法创建解密器，使用原始数据");
AESCrypto crypto = getOrCreateCrypto(secret); return rawData;
if (crypto == null) {
log.info("⚠️ 收到加密消息但无法创建解密器，使用原始数据");
return rawData;
}
// 解密数据
String decryptedData = crypto.decryptString(encryptedMessage.getData());
return decryptedData;
} }
} catch (Exception e) { // 解密数据
// 解析失败，可能是非加密格式，直接返回原始数据 return crypto.decryptString(encryptedMessage.getData());
log.info("数据未加密或解密失败，使用原始数据: {}", e.getMessage());
} }
return rawData; return rawData;
} }
@@ -237,7 +229,7 @@ public class FlowController extends BaseController {
checkUserRelatedLimits(userId, name); checkUserRelatedLimits(userId, name);
checkUserTunnelRelatedLimits(userTunnelId, name, userId); checkUserTunnelRelatedLimits(userTunnelId, name, userId);
} }
log.info("流量处理完成");
return SUCCESS_RESPONSE; return SUCCESS_RESPONSE;
} }
@@ -348,7 +340,8 @@ public class FlowController extends BaseController {
updateWrapper.setSql("in_flow = in_flow + " + flowStats.getD()); updateWrapper.setSql("in_flow = in_flow + " + flowStats.getD());
updateWrapper.setSql("out_flow = out_flow + " + flowStats.getU()); updateWrapper.setSql("out_flow = out_flow + " + flowStats.getU());
forwardService.update(null, updateWrapper); boolean update = forwardService.update(null, updateWrapper);
System.out.println(update);
} }
} }
@@ -361,7 +354,8 @@ public class FlowController extends BaseController {
updateWrapper.setSql("in_flow = in_flow + " + flowStats.getD()); updateWrapper.setSql("in_flow = in_flow + " + flowStats.getD());
updateWrapper.setSql("out_flow = out_flow + " + flowStats.getU()); updateWrapper.setSql("out_flow = out_flow + " + flowStats.getU());
userService.update(null, updateWrapper); boolean update = userService.update(null, updateWrapper);
System.out.println(update);
} }
} }
@@ -376,7 +370,8 @@ public class FlowController extends BaseController {
updateWrapper.eq("id", userTunnelId); updateWrapper.eq("id", userTunnelId);
updateWrapper.setSql("in_flow = in_flow + " + flowStats.getD()); updateWrapper.setSql("in_flow = in_flow + " + flowStats.getD());
updateWrapper.setSql("out_flow = out_flow + " + flowStats.getU()); updateWrapper.setSql("out_flow = out_flow + " + flowStats.getU());
userTunnelService.update(null, updateWrapper); boolean update = userTunnelService.update(null, updateWrapper);
System.out.println(update);
} }
} }