增加多类型匹配

增加多类型匹配

Author: MentosL

arex-instrumentation/servlet/arex-httpservlet/src/main/java/io/arex/inst/httpservlet/convert/BodyConvertFactory.java

@@ -0,0 +1,31 @@
+package io.arex.inst.httpservlet.convert;
+
+import io.arex.agent.bootstrap.util.CollectionUtil;
+
+import java.util.*;
+
+public class BodyConvertFactory {
+
+
+    private static final List<BodyConverters> cacheList = new ArrayList<>();
+
+    static {
+        ServiceLoader<BodyConverters> load = ServiceLoader.load(BodyConverters.class);
+        if (load != null) {
+            Iterator<BodyConverters> iterator = load.iterator();
+            while (iterator.hasNext()) {
+                cacheList.add(iterator.next());
+            }
+        }
+    }
+
+    public static List<BodyConverters> getAllList() {
+        synchronized (BodyConvertFactory.class) {
+            if (CollectionUtil.isNotEmpty(cacheList)) {
+                return Collections.unmodifiableList(cacheList);
+            }
+            return null;
+        }
+    }
+
+}

arex-instrumentation/servlet/arex-httpservlet/src/main/java/io/arex/inst/httpservlet/convert/BodyConverters.java

@@ -0,0 +1,10 @@
+package io.arex.inst.httpservlet.convert;
+
+import io.arex.inst.httpservlet.adapter.ServletAdapter;
+
+public interface BodyConverters<HttpServletRequest, HttpServletResponse> {
+
+    boolean match(HttpServletRequest request, ServletAdapter adapter);
+
+    String getBody(HttpServletRequest request, ServletAdapter adapter);
+}

arex-instrumentation/servlet/arex-httpservlet/src/main/java/io/arex/inst/httpservlet/convert/impl/ApplicationJsonBodyConvert.java

@@ -0,0 +1,31 @@
+package io.arex.inst.httpservlet.convert.impl;
+
+import io.arex.agent.bootstrap.util.StringUtil;
+import io.arex.inst.httpservlet.adapter.ServletAdapter;
+import io.arex.inst.httpservlet.convert.BodyConverters;
+
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.util.Base64;
+
+public class ApplicationJsonBodyConvert implements BodyConverters<HttpServletRequest, HttpServletResponse> {
+
+    private static final String CONTENT_TYPE = "application/json";
+
+    @Override
+    public boolean match(HttpServletRequest httpServletRequest, ServletAdapter adapter) {
+        if (httpServletRequest == null || adapter == null) {
+            return false;
+        }
+        String contentType = adapter.getContentType(httpServletRequest);
+        return StringUtil.isNotEmpty(contentType) && contentType.contains(CONTENT_TYPE);
+    }
+
+    @Override
+    public String getBody(HttpServletRequest httpServletRequest, ServletAdapter adapter) {
+        if (httpServletRequest == null || adapter == null) {
+            return null;
+        }
+        return Base64.getEncoder().encodeToString(adapter.getRequestBytes(httpServletRequest));
+    }
+}

arex-instrumentation/servlet/arex-httpservlet/src/main/java/io/arex/inst/httpservlet/convert/impl/ApplicationXmlBodyConvert.java

@@ -0,0 +1,22 @@
+package io.arex.inst.httpservlet.convert.impl;
+
+import io.arex.inst.httpservlet.adapter.ServletAdapter;
+import io.arex.inst.httpservlet.convert.BodyConverters;
+
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+
+public class ApplicationXmlBodyConvert implements BodyConverters<HttpServletRequest, HttpServletResponse> {
+
+    private static final String CONTENT_TYPE = "application/xml";
+
+    @Override
+    public boolean match(HttpServletRequest httpServletRequest, ServletAdapter adapter) {
+        return false;
+    }
+
+    @Override
+    public String getBody(HttpServletRequest httpServletRequest, ServletAdapter adapter) {
+        return null;
+    }
+}

arex-instrumentation/servlet/arex-httpservlet/src/main/resources/META-INF/services/io.arex.inst.httpservlet.convert.BodyConverters

@@ -0,0 +1,2 @@
+io.arex.inst.httpservlet.convert.impl.ApplicationJsonBodyConvert
+io.arex.inst.httpservlet.convert.impl.ApplicationXmlBodyConvert