From 1fe0965d93e376fd5ae7392365f2f7e7ce7f9b4f Mon Sep 17 00:00:00 2001
From: Xiajijian <1176061145@qq.com>
Date: Wed, 11 Feb 2026 08:48:37 +0800
Subject: [PATCH] 1. 修改用户为空时,获取不到无人机遥测数据。

---
 ard-work/src/main/java/com/ard/work/component/forest/TokenInterceptor.java |   33 ++++++++++++++++++++-------------
 1 files changed, 20 insertions(+), 13 deletions(-)

diff --git a/ard-work/src/main/java/com/ard/work/component/forest/TokenInterceptor.java b/ard-work/src/main/java/com/ard/work/component/forest/TokenInterceptor.java
index 72318db..52f29ac 100644
--- a/ard-work/src/main/java/com/ard/work/component/forest/TokenInterceptor.java
+++ b/ard-work/src/main/java/com/ard/work/component/forest/TokenInterceptor.java
@@ -1,18 +1,21 @@
 package com.ard.work.component.forest;
 
-import com.ard.work.config.TyConfiguration;
 import com.ard.work.domain.receiver.AccessTokenReceiver;
+import com.ard.work.service.IArdServerInfoService;
+import com.dtflys.forest.exceptions.ForestRuntimeException;
 import com.dtflys.forest.http.ForestRequest;
-import com.dtflys.forest.interceptor.Interceptor;
+import com.dtflys.forest.http.ForestResponse;
+import com.dtflys.forest.interceptor.ForestInterceptor;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Component;
+
 import javax.annotation.Resource;
 
 @Component
 @Slf4j
-public class TokenInterceptor<T> implements Interceptor<T> {
+public class TokenInterceptor implements ForestInterceptor {
     @Resource
-    private TyConfiguration tyConfiguration;
+    private IArdServerInfoService ardServerService;
 
     /**
      * 该方法在请求发送之前被调用, 若返回false则不会继续发送请求
@@ -21,14 +24,18 @@
      */
     @Override
     public boolean beforeExecute(ForestRequest req) {
-        log.info("invoke Simple beforeExecute");
-        // 执行在发送请求之前处理的代码
-        // 从缓存里获取token
-        AccessTokenReceiver accessToken = tyConfiguration.getAccessToken();
-        req.addHeader("x-cid", accessToken.getCompanyId());
-        req.addHeader("x-token", accessToken.getAccessToken());
-        // req.addHeader("Authorization", ""));  // 添加Header
-        // req.addQuery("username", "foo");  // 添加URL的Query参数
-        return true;  // 继续执行请求返回true
+        try {
+            String serverId = (String) req.getVariableValue("serverId");
+            AccessTokenReceiver token = ardServerService.getAccessToken(serverId);
+            if (token == null) {
+                return false; // 阻止请求继续
+            }
+            req.addHeader("x-cid", token.getCompanyId());
+            req.addHeader("x-token", token.getAccessToken());
+        } catch (ForestRuntimeException e) {
+            log.error("拦截器处理异常:{}", e.getMessage());
+            return false;
+        }
+        return true;
     }
 }
\ No newline at end of file

--
Gitblit v1.9.3