Use loader for chained plugin

apache · Jan 26, 2025 · d156f23 · d156f23
1 parent 2de4d0b
commit d156f23
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/...in/src/main/java/org/apache/gravitino/authorization/chain/ChainedAuthorizationPlugin.java b/...in/src/main/java/org/apache/gravitino/authorization/chain/ChainedAuthorizationPlugin.java
@@ -72,8 +72,13 @@ private void initPlugins(String catalogProvider, Map<String, String> properties)
               try {
                 BaseAuthorization<?> authorization =
                     BaseAuthorization.createAuthorization(classLoader, authzProvider);
+
+                // Load the authorization plugin with the class loader of the catalog.
+                // Because the JDBC authorization plugin may load JDBC driver using the class
+                // loader.
                 AuthorizationPlugin authorizationPlugin =
-                    authorization.newPlugin(metalake, catalogProvider, pluginConfig);
+                    classLoader.withClassLoader(
+                        cl -> authorization.newPlugin(metalake, catalogProvider, pluginConfig));
                 plugins.add(authorizationPlugin);
               } catch (Exception e) {
                 throw new RuntimeException(e);