From 3940ebf3e39da287974c4f49474e668d33fed2df Mon Sep 17 00:00:00 2001 From: zhangjian <zhangjianrock@163.com> Date: 星期六, 19 八月 2023 17:28:27 +0800 Subject: [PATCH] 处理连接失败异常 --- ard-work/src/main/java/com/ruoyi/device/uav/service/UavService.java | 77 ++++++++++++++++++++++++-------------- 1 files changed, 48 insertions(+), 29 deletions(-) diff --git a/ard-work/src/main/java/com/ruoyi/device/uav/service/UavService.java b/ard-work/src/main/java/com/ruoyi/device/uav/service/UavService.java index 1950bfd..99dae26 100644 --- a/ard-work/src/main/java/com/ruoyi/device/uav/service/UavService.java +++ b/ard-work/src/main/java/com/ruoyi/device/uav/service/UavService.java @@ -1,10 +1,13 @@ package com.ruoyi.device.uav.service; import com.dtflys.forest.exceptions.ForestNetworkException; +import com.dtflys.forest.exceptions.ForestRuntimeException; 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; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.http.HttpMethod; @@ -15,28 +18,30 @@ import javax.annotation.PostConstruct; import javax.crypto.Cipher; import javax.crypto.spec.SecretKeySpec; +import java.io.IOException; import java.lang.reflect.InvocationTargetException; import java.lang.reflect.Method; import java.util.Map; @Service -public class UavService implements IUavService { +@Slf4j(topic = "uav") +public class UavService { public static final String USERNAME = "ardbailu";//鐢ㄦ埛鍚� public static final String PASSWORD = "ardkj12345";//瀵嗙爜 public static final String SALT = "0123456789012345";//鐩� - //token 鐧婚檰鍚庢瘡娆¤姹�,鍦╤eader涓惡甯� - private String token;//鐧诲綍鎴愬姛杩斿洖鐨則oken + private Map uavUser;//鐧诲綍鐨勭敤鎴蜂俊鎭� private ObjectMapper om = new ObjectMapper(); @Autowired private UavClient uavClient; + @Autowired + private RedisCache redisCache; @PostConstruct public void created() { this.login(); } - @Override public String doUavRequest(HttpMethod method, String url, String data) { String res = null; Method requestMethod = null; @@ -47,56 +52,71 @@ 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) { - throw new RuntimeException(ex); - } catch (InvocationTargetException ex) { - throw new RuntimeException(ex); - } + res = (String) requestMethod.invoke(this.uavClient, url, this.getToken(), 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; } - @Override + public Map getUavUser() { + 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, "{}"); + } 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 token; + } + public void login() { + log.debug("鐧诲綍鏃犱汉鏈哄閮ㄦ帴鍙�"); 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); - Map data = (Map) resMap.get("data"); - if (data != null) { - this.token = (String) data.get("access_token"); + if (resMap != null) { + redisCache.setCacheMap("uav:uavUser", (Map) resMap.get("data")); } - } catch (ForestNetworkException e) { - e.printStackTrace(); + } catch (ForestRuntimeException e) { + log.error("鏃犱汉鏈鸿繛鎺ヨ秴鏃�" + e.getMessage()); } catch (JsonMappingException e) { - throw new RuntimeException(e); + e.printStackTrace(); } catch (JsonProcessingException e) { - throw new RuntimeException(e); + e.printStackTrace(); + } catch (IOException e) { + e.printStackTrace(); } } public String Encrypt(String sSrc, String sKey) { if (sKey == null) { - System.out.print("Key涓虹┖null"); return null; } // 鍒ゆ柇Key鏄惁涓�16浣� if (sKey.length() != 16) { - System.out.print("Key闀垮害涓嶆槸16浣�"); return null; } byte[] encrypted = null; @@ -108,7 +128,6 @@ encrypted = cipher.doFinal(sSrc.getBytes("utf-8")); } catch (Exception e) { e.printStackTrace(); - ; } return new Base64().encodeToString(encrypted);//姝ゅ浣跨敤BASE64鍋氳浆鐮佸姛鑳斤紝鍚屾椂鑳借捣鍒�2娆″姞瀵嗙殑浣滅敤銆� } -- Gitblit v1.9.3