Try to caputre exception

airlift · Aug 7, 2024 · efe3561 · efe3561
1 parent 131e712
commit efe3561
Showing 1 changed file with 16 additions and 8 deletions.
diff --git a/src/test/java/io/airlift/compress/v2/HadoopNative.java b/src/test/java/io/airlift/compress/v2/HadoopNative.java
@@ -68,14 +68,9 @@ private static void requireNativeZlib()
     {
         Configuration conf = new Configuration();
         if (!ZlibFactory.isNativeZlibLoaded(conf)) {
-            Throwable cause = null;
-            try {
-                Method initIDs = ZlibCompressor.class.getDeclaredMethod("initIDs");
-                initIDs.setAccessible(true);
-                initIDs.invoke(null);
-            }
-            catch (ReflectiveOperationException e) {
-                cause = e;
+            Throwable cause = initZlib(ZlibCompressor.class);
+            if (cause == null) {
+                cause = initZlib(ZlibDecompressor.class);
             }
             throw new RuntimeException("native zlib is not loaded", cause);
         }
@@ -91,6 +86,19 @@ private static void requireNativeZlib()
         }
     }
 
+    private static Throwable initZlib(Class<?> clazz)
+    {
+        try {
+            Method initIDs = clazz.getDeclaredMethod("initIDs");
+            initIDs.setAccessible(true);
+            initIDs.invoke(null);
+            return null;
+        }
+        catch (ReflectiveOperationException e) {
+            return e;
+        }
+    }
+
     private static void setStatic(Field field, Object value)
             throws IllegalAccessException
     {