新增读取bifang静态阈值配置接口

修改galaxy工具类库版本
2021-08-18 19:15:49 +08:00
parent c957f3ec1c
commit f744677021
10 changed files with 303 additions and 140 deletions
--- a/src/main/java/com/zdjizhi/etl/ParseStaticThreshold.java
+++ b/src/main/java/com/zdjizhi/etl/ParseStaticThreshold.java
@@ -0,0 +1,126 @@
+package com.zdjizhi.etl;
+
+import com.fasterxml.jackson.databind.JavaType;
+import com.zdjizhi.common.CommonConfig;
+import com.zdjizhi.common.DosDetectionThreshold;
+import com.zdjizhi.utils.HttpClientUtils;
+import com.zdjizhi.utils.JsonMapper;
+import org.apache.http.client.utils.URIBuilder;
+import org.apache.http.message.BasicHeader;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.net.URISyntaxException;
+import java.util.ArrayList;
+import java.util.HashMap;
+
+/**
+ * @author wlh
+ */
+public class ParseStaticThreshold {
+    private static Logger logger = LoggerFactory.getLogger(ParseStaticThreshold.class);
+    private static String encryptpwd;
+
+    private static JsonMapper jsonMapperInstance = JsonMapper.getInstance();
+    private static JavaType hashmapJsonType = jsonMapperInstance.createCollectionType(HashMap.class, String.class, Object.class);
+    private static JavaType thresholdType = jsonMapperInstance.createCollectionType(ArrayList.class, DosDetectionThreshold.class);
+
+    static {
+        //加载加密登录密码
+        encryptpwd = getEncryptpwd();
+    }
+
+    /**
+     * 获取加密密码
+     */
+    private static String getEncryptpwd(){
+        String psw = HttpClientUtils.ERROR_MESSAGE;
+        try {
+            URIBuilder uriBuilder = new URIBuilder(CommonConfig.BIFANG_SERVER_URI);
+            HashMap<String, String> parms = new HashMap<>();
+            parms.put("password",CommonConfig.BIFANG_SERVER_PASSWORD);
+            HttpClientUtils.setUrlWithParams(uriBuilder,CommonConfig.BIFANG_SERVER_ENCRYPTPWD_PATH,parms);
+            String resposeJsonStr = HttpClientUtils.httpGet(uriBuilder.build());
+            if (!HttpClientUtils.ERROR_MESSAGE.equals(resposeJsonStr)){
+                HashMap<String, Object> resposeMap = jsonMapperInstance.fromJson(resposeJsonStr, hashmapJsonType);
+                boolean success = (boolean)resposeMap.get("success");
+                if (success){
+                    HashMap<String, Object> data = jsonMapperInstance.fromJson(jsonMapperInstance.toJson(resposeMap.get("data")), hashmapJsonType);
+                    psw = data.get("encryptpwd").toString();
+                }
+            }
+        }catch (URISyntaxException e){
+            logger.error("构造URI异常",e);
+        }catch (Exception e){
+            logger.error("获取encryptpwd失败",e);
+        }
+        return psw;
+    }
+
+    /**
+     * 登录bifang服务，获取token
+     * @return token
+     */
+    private static String loginBifangServer(){
+        String token = HttpClientUtils.ERROR_MESSAGE;
+        try {
+            if (!HttpClientUtils.ERROR_MESSAGE.equals(encryptpwd)){
+                URIBuilder uriBuilder = new URIBuilder(CommonConfig.BIFANG_SERVER_URI);
+                HashMap<String, String> parms = new HashMap<>();
+                parms.put("username",CommonConfig.BIFANG_SERVER_USER);
+                parms.put("password",encryptpwd);
+                HttpClientUtils.setUrlWithParams(uriBuilder,CommonConfig.BIFANG_SERVER_LOGIN_PATH,parms);
+                String resposeJsonStr = HttpClientUtils.httpPost(uriBuilder.build(), null);
+                if (!HttpClientUtils.ERROR_MESSAGE.equals(resposeJsonStr)){
+                    HashMap<String, Object> resposeMap = jsonMapperInstance.fromJson(resposeJsonStr, hashmapJsonType);
+                    boolean success = (boolean)resposeMap.get("success");
+                    if (success){
+                        HashMap<String, Object> data = jsonMapperInstance.fromJson(jsonMapperInstance.toJson(resposeMap.get("data")), hashmapJsonType);
+                        token = data.get("token").toString();
+                    }
+                }
+            }
+        }catch (Exception e){
+            logger.error("登录失败,未获取到token",e);
+        }
+        return token;
+    }
+
+    /**
+     * 获取静态阈值配置列表
+     * @return thresholds
+     */
+    private static ArrayList<DosDetectionThreshold> getDosDetectionThreshold(){
+        ArrayList<DosDetectionThreshold> thresholds = null;
+        try {
+            URIBuilder uriBuilder = new URIBuilder(CommonConfig.BIFANG_SERVER_URI);
+            HttpClientUtils.setUrlWithParams(uriBuilder,CommonConfig.BIFANG_SERVER_POLICY_THRESHOLD_PATH,null);
+            String token = loginBifangServer();
+            if (!HttpClientUtils.ERROR_MESSAGE.equals(token)){
+                BasicHeader authorization = new BasicHeader("Authorization", token);
+                String resposeJsonStr = HttpClientUtils.httpGet(uriBuilder.build(), authorization);
+                if (!HttpClientUtils.ERROR_MESSAGE.equals(resposeJsonStr)){
+                    HashMap<String, Object> resposeMap = jsonMapperInstance.fromJson(resposeJsonStr, hashmapJsonType);
+                    boolean success = (boolean)resposeMap.get("success");
+                    if (success){
+                        HashMap<String, Object> data = jsonMapperInstance.fromJson(jsonMapperInstance.toJson(resposeMap.get("data")), hashmapJsonType);
+                        thresholds = jsonMapperInstance.fromJson(jsonMapperInstance.toJson(data.get("list")), thresholdType);
+                        logger.info("获取到静态阈值配置{}条",thresholds.size());
+                    }
+                }
+            }
+        }catch (Exception e){
+            logger.error("获取静态阈值配置失败",e);
+        }
+        return thresholds;
+    }
+
+    public static void main(String[] args) {
+        ArrayList<DosDetectionThreshold> dosDetectionThreshold = getDosDetectionThreshold();
+        if (dosDetectionThreshold != null && !dosDetectionThreshold.isEmpty()){
+            dosDetectionThreshold.forEach(s -> System.out.println(s.toString()));
+        }
+    }
+
+
+}