Restapi改造

Author: yanghuaiGit

flinkx-restapi/flinkx-restapi-core/src/main/java/com/dtstack/flinkx/restapi/common/MapUtils.java

@@ -0,0 +1,37 @@
+package com.dtstack.flinkx.restapi.common;
+
+import com.dtstack.flinkx.util.GsonUtil;
+
+import java.util.Map;
+import java.util.Objects;
+
+public class MapUtils {
+
+
+    public static Object getData(Map<String, Object> data, String[] names) {
+        Map<String, Object> tempHashMap = data;
+        for (int i = 0; i < names.length; i++) {
+            if (tempHashMap.containsKey(names[i]) && i != names.length - 1) {
+                if (Objects.isNull(tempHashMap.get(names[i]))) {
+                    return null;
+                }
+                if (tempHashMap.get(names[i]) instanceof Map) {
+                    tempHashMap = (Map<String, Object>) tempHashMap.get(names[i]);
+                } else if (tempHashMap.get(names[i]) instanceof String) {
+                    try {
+                        tempHashMap = GsonUtil.GSON.fromJson((String) tempHashMap.get(names[i]), GsonUtil.gsonMapTypeToken);
+                    } catch (Exception e) {
+                        return null;
+                    }
+                } else {
+                    return null;
+                }
+            } else if (i == names.length - 1) {
+                return tempHashMap.get(names[i]);
+            } else {
+                return null;
+            }
+        }
+        return null;
+    }
+}

flinkx-restapi/flinkx-restapi-core/src/main/java/com/dtstack/flinkx/restapi/common/exception.java

@@ -1,4 +1,4 @@
-package com.dtstack.flinkx.restapi.common.handler;
+package com.dtstack.flinkx.restapi.common.exception;
 
 public class ReadRecordException extends RuntimeException {
     public ReadRecordException() {

…/common/handler/ReadRecordException.java …ommon/exception/ReadRecordException.javaflinkx-restapi/flinkx-restapi-core/src/main/java/com/dtstack/flinkx/restapi/common/handler/ReadRecordException.java renamed to flinkx-restapi/flinkx-restapi-core/src/main/java/com/dtstack/flinkx/restapi/common/exception/ReadRecordException.java flinkx-restapi/flinkx-restapi-core/src/main/java/com/dtstack/flinkx/restapi/common/handler/ReadRecordException.java renamed to flinkx-restapi/flinkx-restapi-core/src/main/java/com/dtstack/flinkx/restapi/common/exception/ReadRecordException.java

@@ -0,0 +1,22 @@
+package com.dtstack.flinkx.restapi.common.exception;
+
+public class ResponseBreakException extends RuntimeException{
+    public ResponseBreakException() {
+    }
+
+    public ResponseBreakException(String message) {
+        super(message);
+    }
+
+    public ResponseBreakException(String message, Throwable cause) {
+        super(message, cause);
+    }
+
+    public ResponseBreakException(Throwable cause) {
+        super(cause);
+    }
+
+    public ResponseBreakException(String message, Throwable cause, boolean enableSuppression, boolean writableStackTrace) {
+        super(message, cause, enableSuppression, writableStackTrace);
+    }
+}

flinkx-restapi/flinkx-restapi-core/src/main/java/com/dtstack/flinkx/restapi/common/exception/ResponseBreakException.java

@@ -0,0 +1,22 @@
+package com.dtstack.flinkx.restapi.common.exception;
+
+public class ResponseRetryException extends RuntimeException {
+    public ResponseRetryException() {
+    }
+
+    public ResponseRetryException(String message) {
+        super(message);
+    }
+
+    public ResponseRetryException(String message, Throwable cause) {
+        super(message, cause);
+    }
+
+    public ResponseRetryException(Throwable cause) {
+        super(cause);
+    }
+
+    public ResponseRetryException(String message, Throwable cause, boolean enableSuppression, boolean writableStackTrace) {
+        super(message, cause, enableSuppression, writableStackTrace);
+    }
+}

flinkx-restapi/flinkx-restapi-core/src/main/java/com/dtstack/flinkx/restapi/common/exception/ResponseRetryException.java

@@ -0,0 +1,26 @@
+package com.dtstack.flinkx.restapi.common.handler;
+
+import com.dtstack.flinkx.restapi.common.MapUtils;
+import com.dtstack.flinkx.restapi.common.exception.ResponseBreakException;
+import com.dtstack.flinkx.restapi.common.exception.ResponseRetryException;
+
+import java.util.Map;
+import java.util.Set;
+
+public class BreakDataHandler extends DataHandler {
+
+    public BreakDataHandler(String key, Set<String> value) {
+        super(key, value);
+    }
+
+    @Override
+    public void execute(Map<String, Object> responseData) {
+        String[] strings = new String[0];
+        strings[0] = key;
+        Object data = MapUtils.getData(responseData, strings);
+        if (value.contains(data.toString())) {
+            throw new ResponseBreakException("key:"+key+" contains"+data.toString()+" ,need break");
+        }
+
+    }
+}

flinkx-restapi/flinkx-restapi-core/src/main/java/com/dtstack/flinkx/restapi/common/handler/BreakDataHandler.java

@@ -3,13 +3,13 @@
 import java.util.Map;
 import java.util.Set;
 
-public abstract class Handler {
+public abstract class DataHandler {
 
-    private String key;
+    protected String key;
 
-    private Set<String> value;
+    protected Set<String> value;
 
-    public Handler(String key, Set<String> value) {
+    public DataHandler(String key, Set<String> value) {
         this.key = key;
         this.value = value;
     }
@@ -19,4 +19,5 @@ public boolean isPipei(Map<String, Object> responseData) {
     }
 
     public abstract void execute(Map<String, Object> responseData);
+
 }

…inkx/restapi/common/handler/Handler.java …/restapi/common/handler/DataHandler.javaflinkx-restapi/flinkx-restapi-core/src/main/java/com/dtstack/flinkx/restapi/common/handler/Handler.java renamed to flinkx-restapi/flinkx-restapi-core/src/main/java/com/dtstack/flinkx/restapi/common/handler/DataHandler.java flinkx-restapi/flinkx-restapi-core/src/main/java/com/dtstack/flinkx/restapi/common/handler/Handler.java renamed to flinkx-restapi/flinkx-restapi-core/src/main/java/com/dtstack/flinkx/restapi/common/handler/DataHandler.java

@@ -0,0 +1,29 @@
+package com.dtstack.flinkx.restapi.common.handler;
+
+import org.apache.commons.lang.StringUtils;
+
+import java.util.HashMap;
+import java.util.Map;
+
+public class DataHandlerFactory {
+
+
+    private static HashMap<String, DataHandler> handlerHashMap = new HashMap<>(16);
+
+    static {
+
+    }
+
+    public static DataHandler getDataHandler(Map dataHandlerParam) {
+        if (StringUtils.isNotBlank(dataHandlerParam.get("key").toString())) {
+            return handlerHashMap.get(dataHandlerParam.get("key").toString());
+        }else{
+            throw new IllegalArgumentException("dataHandler key not allow blank");
+        }
+    }
+
+    public static void destroy() {
+        //gc
+        handlerHashMap = null;
+    }
+}

flinkx-restapi/flinkx-restapi-core/src/main/java/com/dtstack/flinkx/restapi/common/handler/DataHandlerFactory.java

@@ -0,0 +1,24 @@
+package com.dtstack.flinkx.restapi.common.handler;
+
+import com.dtstack.flinkx.restapi.common.MapUtils;
+import com.dtstack.flinkx.restapi.common.exception.ResponseRetryException;
+
+import java.util.Map;
+import java.util.Set;
+
+public class RetryDataHandler extends DataHandler {
+
+    public RetryDataHandler(String key, Set<String> value) {
+        super(key, value);
+    }
+
+    @Override
+    public void execute(Map<String, Object> responseData) {
+        String[] strings = new String[0];
+        strings[0] = key;
+        Object data = MapUtils.getData(responseData, strings);
+        if (value.contains(data.toString())) {
+            throw new ResponseRetryException("key:"+key+" contains"+data.toString()+" ,need retry");
+        }
+    }
+}