fix: thread context classloader propagation in forkjoinpool (#466)

Author: YongwuHe

arex-instrumentation-api/src/main/java/io/arex/inst/runtime/listener/EventProcessor.java

@@ -66,13 +66,13 @@ private static void addEnterLog() {
     /**
      * user loader to load serializer, ex: ParallelWebappClassLoader
      */
-    private static void initSerializer() {
+    private static void initSerializer(ClassLoader contextClassLoader) {
         if (!existJacksonDependency) {
             AdviceClassesCollector.INSTANCE.appendToClassLoaderSearch("jackson",
-                Thread.currentThread().getContextClassLoader());
+                    contextClassLoader);
         }
         Serializer.initSerializerConfigMap();
-        final List<StringSerializable> serializableList = ServiceLoader.load(StringSerializable.class, Thread.currentThread().getContextClassLoader());
+        final List<StringSerializable> serializableList = ServiceLoader.load(StringSerializable.class, contextClassLoader);
         Serializer.builder(serializableList).build();
     }
 
@@ -114,19 +114,21 @@ public static void onExit(){
      */
     public static void onRequest(){
         if (INIT_DEPENDENCY.compareAndSet(InitializeEnum.START, InitializeEnum.RUNNING)) {
+            ClassLoader contextClassLoader = Thread.currentThread().getContextClassLoader();
+            // https://bugs.openjdk.org/browse/JDK-8172726
             CompletableFuture.runAsync(() -> {
-                initSerializer();
-                initLog();
-                RequestHandlerManager.init();
+                initSerializer(contextClassLoader);
+                initLog(contextClassLoader);
+                RequestHandlerManager.init(contextClassLoader);
                 INIT_DEPENDENCY.set(InitializeEnum.COMPLETE);
             });
         }
         TimeCache.remove();
         ContextManager.remove();
     }
 
-    private static void initLog() {
-        List<Logger> extensionLoggerList = ServiceLoader.load(Logger.class, Thread.currentThread().getContextClassLoader());
+    private static void initLog(ClassLoader contextClassLoader) {
+        List<Logger> extensionLoggerList = ServiceLoader.load(Logger.class, contextClassLoader);
         LogManager.build(extensionLoggerList);
     }
 }

arex-instrumentation-api/src/main/java/io/arex/inst/runtime/request/RequestHandlerManager.java

@@ -12,8 +12,8 @@
 public class RequestHandlerManager {
     private static final Map<String, List<RequestHandler>> REQUEST_HANDLER_CACHE = new ConcurrentHashMap<>();
 
-    public static void init() {
-        final List<RequestHandler> requestHandlers = ServiceLoader.load(RequestHandler.class);
+    public static void init(ClassLoader contextClassLoader) {
+        final List<RequestHandler> requestHandlers = ServiceLoader.load(RequestHandler.class, contextClassLoader);
         final Map<String, List<RequestHandler>> requestHandlerMap = requestHandlers.stream().collect(Collectors.groupingBy(RequestHandler::name));
         REQUEST_HANDLER_CACHE.putAll(requestHandlerMap);
     }

arex-instrumentation-api/src/test/java/io/arex/inst/runtime/request/RequestHandlerManagerTest.java

@@ -15,6 +15,7 @@ class RequestHandlerManagerTest {
 
     @BeforeAll
     static void setUp() {
+        ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
         Mockito.mockStatic(ServiceLoader.class);
         requestHandler = Mockito.mock(RequestHandler.class);
         requestHandlerError = Mockito.mock(RequestHandler.class);
@@ -23,8 +24,8 @@ static void setUp() {
         Mockito.doThrow(new RuntimeException()).when(requestHandlerError).preHandle("request");
         Mockito.doThrow(new RuntimeException()).when(requestHandlerError).handleAfterCreateContext("request");
         Mockito.doThrow(new RuntimeException()).when(requestHandlerError).postHandle("request", "response");
-        Mockito.when(ServiceLoader.load(RequestHandler.class)).thenReturn(Arrays.asList(requestHandler, requestHandlerError));
-        RequestHandlerManager.init();
+        Mockito.when(ServiceLoader.load(RequestHandler.class, classLoader)).thenReturn(Arrays.asList(requestHandler, requestHandlerError));
+        RequestHandlerManager.init(classLoader);
     }
 
     @AfterAll
@@ -71,4 +72,4 @@ void postHandle() {
         // error
         assertDoesNotThrow(() -> RequestHandlerManager.postHandle("request", "response", "testError"));
     }
-}
+}