Set ClassLoader when calling EventListenerFactory

Before the change, certain libraries cannot be used in an event listener implementation.
prestodb · Feb 22, 2019 · e78c69e · e78c69e
1 parent 782fbf3
commit e78c69e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/presto-main/src/main/java/com/facebook/presto/eventlistener/EventListenerManager.java b/presto-main/src/main/java/com/facebook/presto/eventlistener/EventListenerManager.java
@@ -13,6 +13,7 @@
  */
 package com.facebook.presto.eventlistener;
 
+import com.facebook.presto.spi.classloader.ThreadContextClassLoader;
 import com.facebook.presto.spi.eventlistener.EventListener;
 import com.facebook.presto.spi.eventlistener.EventListenerFactory;
 import com.facebook.presto.spi.eventlistener.QueryCompletedEvent;
@@ -79,8 +80,10 @@ protected void setConfiguredEventListener(String name, Map<String, String> prope
         EventListenerFactory eventListenerFactory = eventListenerFactories.get(name);
         checkState(eventListenerFactory != null, "Event listener %s is not registered", name);
 
-        EventListener eventListener = eventListenerFactory.create(ImmutableMap.copyOf(properties));
-        this.configuredEventListener.set(Optional.of(eventListener));
+        try (ThreadContextClassLoader ignored = new ThreadContextClassLoader(eventListenerFactory.getClass().getClassLoader())) {
+            EventListener eventListener = eventListenerFactory.create(ImmutableMap.copyOf(properties));
+            this.configuredEventListener.set(Optional.of(eventListener));
+        }
 
         log.info("-- Loaded event listener %s --", name);
     }