调整代码逻辑+增加unitTest

Author: MentosL

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

@@ -0,0 +1,41 @@
+package io.arex.inst.httpservlet.convert;
+
+import io.arex.agent.bootstrap.util.CollectionUtil;
+import io.arex.inst.httpservlet.adapter.ServletAdapter;
+import io.arex.inst.httpservlet.convert.impl.DefaultHttpMessageConverter;
+
+import java.util.*;
+
+public class HttpMessageConvertFactory {
+
+
+    private static final List<HttpMessageConverter> cacheList = new ArrayList<>();
+
+
+
+    static {
+        ServiceLoader<HttpMessageConverter> load = ServiceLoader.load(HttpMessageConverter.class);
+        if (load != null) {
+            Iterator<HttpMessageConverter> iterator = load.iterator();
+            while (iterator.hasNext()) {
+                cacheList.add(iterator.next());
+            }
+        }
+    }
+
+
+    public static <P, R> HttpMessageConverter<P, R> getSupportedConverter(P p, R r, ServletAdapter adapter) {
+        synchronized (HttpMessageConvertFactory.class) {
+            if (CollectionUtil.isNotEmpty(cacheList)) {
+                for (HttpMessageConverter converter : cacheList) {
+                    if (converter.match(p, r, adapter)) {
+                        return converter;
+                    }
+                }
+            }
+            return DefaultHttpMessageConverter.getInstance();
+        }
+    }
+
+
+}

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 HttpMessageConverter<HttpServletRequest, HttpServletResponse> {
+
+    boolean match(HttpServletRequest request,HttpServletResponse response, ServletAdapter adapter);
+    byte[] getRequest(HttpServletRequest request, ServletAdapter adapter);
+    byte[] getResponse(HttpServletResponse request, ServletAdapter adapter);
+}

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

@@ -0,0 +1,36 @@
+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.HttpMessageConverter;
+
+public class ApplicationJsonBodyConverter<HttpServletRequest, HttpServletResponse> implements HttpMessageConverter<HttpServletRequest, HttpServletResponse> {
+
+    private static final String CONTENT_TYPE = "application/json";
+
+    @Override
+    public boolean match(HttpServletRequest request, HttpServletResponse response, ServletAdapter adapter) {
+        if (request == null || response ==null || adapter == null) {
+            return false;
+        }
+        String contentType = adapter.getContentType(request);
+        return StringUtil.isNotEmpty(contentType) && contentType.contains(CONTENT_TYPE);
+    }
+
+    @Override
+    public byte[] getRequest(HttpServletRequest request, ServletAdapter adapter) {
+        if (request == null || adapter == null) {
+            return new byte[0];
+        }
+        return adapter.getRequestBytes(request);
+    }
+
+    @Override
+    public byte[] getResponse(HttpServletResponse response, ServletAdapter adapter) {
+        return new byte[0];
+    }
+}
+
+
+
+

arex-instrumentation/servlet/arex-httpservlet/src/main/java/io/arex/inst/httpservlet/convert/HttpMessageConverter.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.HttpMessageConverter;
+
+
+public class ApplicationXmlBodyConverter<HttpServletRequest, HttpServletResponse> implements HttpMessageConverter<HttpServletRequest, HttpServletResponse> {
+
+    private static final String CONTENT_TYPE = "application/xml";
+
+    @Override
+    public boolean match(HttpServletRequest request, HttpServletResponse response, ServletAdapter adapter) {
+        if (request == null || response ==null || adapter == null) {
+            return false;
+        }
+        String contentType = adapter.getContentType(request);
+        return StringUtil.isNotEmpty(contentType) && contentType.contains(CONTENT_TYPE);
+    }
+
+    @Override
+    public byte[] getRequest(HttpServletRequest request, ServletAdapter adapter) {
+        return new byte[0];
+    }
+
+    @Override
+    public byte[] getResponse(HttpServletResponse response, ServletAdapter adapter) {
+        return new byte[0];
+    }
+
+}

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

@@ -0,0 +1,41 @@
+package io.arex.inst.httpservlet.convert.impl;
+
+import io.arex.inst.httpservlet.adapter.ServletAdapter;
+import io.arex.inst.httpservlet.convert.HttpMessageConverter;
+
+public class DefaultHttpMessageConverter<HttpServletRequest, HttpServletResponse> implements HttpMessageConverter<HttpServletRequest, HttpServletResponse> {
+
+    private DefaultHttpMessageConverter() {
+
+    }
+
+    private volatile static DefaultHttpMessageConverter instance;
+
+
+    public static final DefaultHttpMessageConverter getInstance() {
+        if (instance == null) {
+            synchronized (DefaultHttpMessageConverter.class) {
+                if (instance == null) {
+                    instance = new DefaultHttpMessageConverter();
+                }
+            }
+        }
+        return instance;
+    }
+
+
+    @Override
+    public boolean match(HttpServletRequest request, HttpServletResponse response, ServletAdapter adapter) {
+        return false;
+    }
+
+    @Override
+    public byte[] getRequest(HttpServletRequest request, ServletAdapter adapter) {
+        return new byte[0];
+    }
+
+    @Override
+    public byte[] getResponse(HttpServletResponse response, ServletAdapter adapter) {
+        return new byte[0];
+    }
+}