AXIS2-6030 Axis2 connections are not returned to connection pool on 1.8.0 with JAXWS

Author: robertlazarski

modules/jaxws/src/org/apache/axis2/jaxws/client/proxy/JAXWSProxyHandler.java

@@ -58,13 +58,16 @@
 import jakarta.xml.ws.WebServiceException;
 import jakarta.xml.ws.WebServiceFeature;
 import jakarta.xml.ws.soap.SOAPBinding;
+import java.io.InputStream;
 import java.lang.reflect.InvocationHandler;
 import java.lang.reflect.Method;
 import java.lang.reflect.Modifier;
 import java.security.PrivilegedActionException;
 import java.security.PrivilegedExceptionAction;
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Future;
+import java.util.Iterator;
+import java.util.Map;
 
 /**
  * ProxyHandler is the java.lang.reflect.InvocationHandler implementation. When a JAX-WS client
@@ -567,13 +570,36 @@ protected Object createResponse(Method method, Object[] args, MessageContext res
             // Free incoming stream
             try {
                 responseContext.freeInputStream();
+                closeInputStream(responseContext);
             }
             catch (Throwable t) {
                 throw ExceptionFactory.makeWebServiceException(t);
             }
         }
     }
 
+    private void closeInputStream(MessageContext responseContext) {
+        // accessing the input stream is not possible via get
+        // workaround using entry set
+        Iterator var2 = responseContext.getMEPContext().entrySet().iterator();
+    
+        while(var2.hasNext()) {
+            Object entry = var2.next();
+            if (entry instanceof Map.Entry && "TRANSPORT_IN".equals(((Map.Entry)entry).getKey())) {
+                Object prop = ((Map.Entry)entry).getValue();
+                if (prop instanceof InputStream) {
+                    try {
+                        InputStream inputStream = (InputStream)prop;
+                        inputStream.close();
+                    } catch (Exception var6) {
+                        log.error(var6.getMessage(), var6);
+                    }
+                }
+                break;
+            }
+        }
+    }
+
     protected static Throwable getFaultResponse(MessageContext msgCtx,
                                                 OperationDescription opDesc) {
         Message msg = msgCtx.getMessage();