zhangjian
2023-09-23 2a18523931228c78ae359997986d004a39545d51
ard-work/src/main/java/com/ruoyi/device/uav/service/UavService.java
@@ -5,6 +5,7 @@
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.JsonMappingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.ruoyi.common.core.redis.RedisCache;
import com.ruoyi.utils.forest.UavClient;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.codec.binary.Base64;
@@ -29,12 +30,12 @@
    public static final String PASSWORD = "ardkj12345";//密码
    public static final String SALT = "0123456789012345";//盐
    //token 登陆后每次请求,在header中携带
    private String token;//登录成功返回的token
    private Map uavUser;//登录的用户信息
    private ObjectMapper om = new ObjectMapper();
    @Autowired
    private UavClient uavClient;
    @Autowired
    private RedisCache redisCache;
    @PostConstruct
    public void created() {
@@ -51,32 +52,42 @@
            e.printStackTrace();
        }
        //执行method
        this.getToken();//获取token验证,验证token有效性
        try {
            res = (String) requestMethod.invoke(this.uavClient, url, this.token, data);
        } catch (InvocationTargetException e) {
            ForestNetworkException fe = (ForestNetworkException) e.getCause();
            if (fe.getStatusCode() == 401) {//token失效,重新登录
                this.login();
            }
            try {//再次调用接口
                res = (String) requestMethod.invoke(this.uavClient, url, this.token, data);
            } catch (IllegalAccessException ex) {
                ex.printStackTrace();
            } catch (InvocationTargetException ex) {
                ex.printStackTrace();
            }
            String token = this.getToken();
            res = (String) requestMethod.invoke(this.uavClient, url, token, data);
        } catch (IllegalAccessException e) {
            throw new RuntimeException(e);
            System.out.println("doUavRequest 访问失败" + e.getMessage());
        } catch (InvocationTargetException e) {
            System.out.println("doUavRequest 执行失败" + e.getMessage());
            e.printStackTrace();
        }
        return res;
    }
    public Map getUavUser() {
        if (this.uavUser == null) {
            this.login();
        return redisCache.getCacheMap("uav:uavUser");
    }
    public String getToken() {
        //获取uav用户信息
        Map uavUser = redisCache.getCacheMap("uav:uavUser");
        String token = (String) uavUser.get("access_token");
        //验证token有效性
        try {
            String res = uavClient.GET("manage/api/v1/devices", token, "{}");
            System.out.println(res);
        } catch (ForestNetworkException fe) {
            if (fe.getStatusCode() == 401) {//token失效,重新登录
                this.login();
                //再次获取token
                uavUser = redisCache.getCacheMap("uav:uavUser");
                token = (String) uavUser.get("access_token");
                return token;
            }
        }
        return this.uavUser;
        return token;
    }
    public void login() {
@@ -84,20 +95,17 @@
        String codedPassword = this.Encrypt(PASSWORD, SALT);
        String body = "{\"username\":\"" + USERNAME + "\",\"password\":\"" + codedPassword + "\"}";
        try {
            //uavClient.GET("logout", null, "{}");
            String res = uavClient.POST("login", null, body);
            Map resMap = om.readValue(res, Map.class);
            this.uavUser = (Map) resMap.get("data");
            if (this.uavUser != null) {
                this.token = (String) this.uavUser.get("access_token");
            if (resMap != null) {
                redisCache.setCacheMap("uav:uavUser", (Map) resMap.get("data"));
            }
        } catch (ForestRuntimeException e) {
            log.error("无人机连接超时" + e.getMessage());
        } catch (JsonMappingException e) {
            e.printStackTrace();
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }