Add an option to JdkProxySource allowing to unwrap UndeclaredThrowableException

Author: reda-alaoui

src/main/java/org/apache/commons/pool3/proxy/BaseProxyHandler.java

@@ -16,6 +16,7 @@
  */
 package org.apache.commons.pool3.proxy;
 
+import java.lang.reflect.InvocationTargetException;
 import java.lang.reflect.Method;
 
 import org.apache.commons.pool3.UsageTracking;
@@ -32,18 +33,21 @@ class BaseProxyHandler<T> {
 
     private volatile T pooledObject;
     private final UsageTracking<T> usageTracking;
+    private final boolean unwrapInvocationTargetException;
 
-    /**
-     * Constructs a new wrapper for the given pooled object.
-     *
-     * @param pooledObject  The object to wrap
-     * @param usageTracking The instance, if any (usually the object pool) to
-     *                      be provided with usage tracking information for this
-     *                      wrapped object
-     */
-    BaseProxyHandler(final T pooledObject, final UsageTracking<T> usageTracking) {
+	/**
+	 * Constructs a new wrapper for the given pooled object.
+	 *
+	 * @param pooledObject                    The object to wrap
+	 * @param usageTracking                   The instance, if any (usually the object pool) to
+	 *                                        be provided with usage tracking information for this
+	 *                                        wrapped object
+	 * @param unwrapInvocationTargetException True to make the proxy throw {@link InvocationTargetException#getTargetException()} instead of {@link InvocationTargetException}
+	 */
+    BaseProxyHandler(final T pooledObject, final UsageTracking<T> usageTracking, boolean unwrapInvocationTargetException) {
         this.pooledObject = pooledObject;
         this.usageTracking = usageTracking;
+        this.unwrapInvocationTargetException = unwrapInvocationTargetException;
     }
 
     /**
@@ -73,7 +77,14 @@ Object doInvoke(final Method method, final Object[] args) throws Throwable {
         if (usageTracking != null) {
             usageTracking.use(object);
         }
-        return method.invoke(object, args);
+        try {
+            return method.invoke(object, args);
+        } catch (InvocationTargetException e) {
+            if (unwrapInvocationTargetException) {
+               throw e.getTargetException();
+            }
+            throw e;
+        }
     }
 
     /**

src/main/java/org/apache/commons/pool3/proxy/CglibProxyHandler.java

@@ -16,6 +16,7 @@
  */
 package org.apache.commons.pool3.proxy;
 
+import java.lang.reflect.InvocationTargetException;
 import java.lang.reflect.Method;
 
 import org.apache.commons.pool3.UsageTracking;
@@ -36,13 +37,14 @@ final class CglibProxyHandler<T> extends BaseProxyHandler<T>
     /**
      * Constructs a CGLib proxy instance.
      *
-     * @param pooledObject  The object to wrap
-     * @param usageTracking The instance, if any (usually the object pool) to
-     *                      be provided with usage tracking information for this
-     *                      wrapped object
+     * @param pooledObject                    The object to wrap
+     * @param usageTracking                   The instance, if any (usually the object pool) to
+     *                                        be provided with usage tracking information for this
+     *                                        wrapped object
+     * @param unwrapInvocationTargetException True to make the proxy throw {@link InvocationTargetException#getTargetException()} instead of {@link InvocationTargetException}
      */
-    CglibProxyHandler(final T pooledObject, final UsageTracking<T> usageTracking) {
-        super(pooledObject, usageTracking);
+    CglibProxyHandler(final T pooledObject, final UsageTracking<T> usageTracking, boolean unwrapInvocationTargetException) {
+        super(pooledObject, usageTracking, unwrapInvocationTargetException);
     }
 
     @Override

src/main/java/org/apache/commons/pool3/proxy/CglibProxySource.java

@@ -16,6 +16,7 @@
  */
 package org.apache.commons.pool3.proxy;
 
+import java.lang.reflect.InvocationTargetException;
 import org.apache.commons.pool3.UsageTracking;
 
 import net.sf.cglib.proxy.Enhancer;
@@ -31,14 +32,26 @@
 public class CglibProxySource<T> implements ProxySource<T> {
 
     private final Class<? extends T> superclass;
+    private final boolean unwrapInvocationTargetException;
+
+    /**
+     * Constructs a new proxy source for the given class.
+     *
+     * @param superclass                      The class to proxy
+     * @param unwrapInvocationTargetException True to make the proxy throw {@link InvocationTargetException#getTargetException()} instead of {@link InvocationTargetException}
+     */
+    public CglibProxySource(final Class<? extends T> superclass, boolean unwrapInvocationTargetException) {
+        this.superclass = superclass;
+        this.unwrapInvocationTargetException = unwrapInvocationTargetException;
+    }
 
     /**
      * Constructs a new proxy source for the given class.
      *
      * @param superclass The class to proxy
      */
     public CglibProxySource(final Class<? extends T> superclass) {
-        this.superclass = superclass;
+        this(superclass, false);
     }
 
     @SuppressWarnings("unchecked") // Case to T on return
@@ -48,7 +61,7 @@ public T createProxy(final T pooledObject, final UsageTracking<T> usageTracking)
         enhancer.setSuperclass(superclass);
 
         final CglibProxyHandler<T> proxyInterceptor =
-                new CglibProxyHandler<>(pooledObject, usageTracking);
+                new CglibProxyHandler<>(pooledObject, usageTracking, unwrapInvocationTargetException);
         enhancer.setCallback(proxyInterceptor);
 
         return (T) enhancer.create();

src/main/java/org/apache/commons/pool3/proxy/JdkProxyHandler.java

@@ -17,6 +17,7 @@
 package org.apache.commons.pool3.proxy;
 
 import java.lang.reflect.InvocationHandler;
+import java.lang.reflect.InvocationTargetException;
 import java.lang.reflect.Method;
 
 import org.apache.commons.pool3.UsageTracking;
@@ -34,13 +35,14 @@ final class JdkProxyHandler<T> extends BaseProxyHandler<T>
     /**
      * Constructs a Java reflection proxy instance.
      *
-     * @param pooledObject  The object to wrap
-     * @param usageTracking The instance, if any (usually the object pool) to
-     *                      be provided with usage tracking information for this
-     *                      wrapped object
+     * @param pooledObject                    The object to wrap
+     * @param usageTracking                   The instance, if any (usually the object pool) to
+     *                                        be provided with usage tracking information for this
+     *                                        wrapped object
+     * @param unwrapInvocationTargetException True to make the proxy throw {@link InvocationTargetException#getTargetException()} instead of {@link InvocationTargetException}
      */
-    JdkProxyHandler(final T pooledObject, final UsageTracking<T> usageTracking) {
-        super(pooledObject, usageTracking);
+    JdkProxyHandler(final T pooledObject, final UsageTracking<T> usageTracking, boolean unwrapInvocationTargetException) {
+        super(pooledObject, usageTracking, unwrapInvocationTargetException);
     }
 
     @Override

src/main/java/org/apache/commons/pool3/proxy/JdkProxySource.java

@@ -16,6 +16,7 @@
  */
 package org.apache.commons.pool3.proxy;
 
+import java.lang.reflect.InvocationTargetException;
 import java.lang.reflect.Proxy;
 import java.util.Arrays;
 
@@ -32,24 +33,37 @@ public class JdkProxySource<T> implements ProxySource<T> {
 
     private final ClassLoader classLoader;
     private final Class<?>[] interfaces;
+    private final boolean unwrapInvocationTargetException;
 
     /**
      * Constructs a new proxy source for the given interfaces.
      *
      * @param classLoader The class loader with which to create the proxy
      * @param interfaces  The interfaces to proxy
+     * @param unwrapInvocationTargetException True to make the proxy throw {@link InvocationTargetException#getTargetException()} instead of {@link InvocationTargetException}
      */
-    public JdkProxySource(final ClassLoader classLoader, final Class<?>[] interfaces) {
+    public JdkProxySource(final ClassLoader classLoader, final Class<?>[] interfaces, boolean unwrapInvocationTargetException) {
         this.classLoader = classLoader;
         // Defensive copy
         this.interfaces = Arrays.copyOf(interfaces, interfaces.length);
+        this.unwrapInvocationTargetException = unwrapInvocationTargetException;
+    }
+
+    /**
+     * Constructs a new proxy source for the given interfaces.
+     *
+     * @param classLoader The class loader with which to create the proxy
+     * @param interfaces  The interfaces to proxy
+     */
+    public JdkProxySource(final ClassLoader classLoader, final Class<?>[] interfaces) {
+        this(classLoader, interfaces, false);
     }
 
     @SuppressWarnings("unchecked") // Cast to T on return.
     @Override
     public T createProxy(final T pooledObject, final UsageTracking<T> usageTracking) {
         return (T) Proxy.newProxyInstance(classLoader, interfaces,
-                new JdkProxyHandler<>(pooledObject, usageTracking));
+                new JdkProxyHandler<>(pooledObject, usageTracking, unwrapInvocationTargetException));
     }
 
     @SuppressWarnings("unchecked")