commit 5aac0bf5bb8e2494eeaa6e83fd81125b1c050b60
author Jingsong <jingsonglee0@gmail.com> Wed Jul 26 14:33:15 2023 +0800
committer Jingsong <jingsonglee0@gmail.com> Wed Jul 26 14:33:29 2023 +0800
tree 5def22cb62272fdb528677810bca0b1d3f45c0bb
parent 5668c234a942d74727374d5e73c84f2791a81da4

[bug] Fix thread classloader in PrestoPageSourceBase

paimon-presto-common/src/main/java/org/apache/paimon/presto/PrestoPageSourceBase.java

diff --git a/paimon-presto-common/src/main/java/org/apache/paimon/presto/PrestoPageSourceBase.java b/paimon-presto-common/src/main/java/org/apache/paimon/presto/PrestoPageSourceBase.java
index 1a02be4..6109c67 100644
--- a/paimon-presto-common/src/main/java/org/apache/paimon/presto/PrestoPageSourceBase.java
+++ b/paimon-presto-common/src/main/java/org/apache/paimon/presto/PrestoPageSourceBase.java
@@ -115,11 +115,15 @@
 
     @Override
     public Page getNextPage() {
-        try {
-            return nextPage();
-        } catch (IOException e) {
-            throw new UncheckedIOException(e);
-        }
+        return ClassLoaderUtils.runWithContextClassLoader(
+                () -> {
+                    try {
+                        return nextPage();
+                    } catch (IOException e) {
+                        throw new UncheckedIOException(e);
+                    }
+                },
+                PrestoPageSourceBase.class.getClassLoader());
     }
 
     @Override