Set classloader for JMX endpoints to application classloader, #12088.

Author: Dav1dde

spring-boot-project/spring-boot-actuator/src/main/java/org/springframework/boot/actuate/endpoint/jmx/EndpointMBean.java

@@ -90,6 +90,7 @@ public Object invoke(String actionName, Object[] params, String[] signature)
 					+ "' has no operation named " + actionName;
 			throw new ReflectionException(new IllegalArgumentException(message), message);
 		}
+		Thread.currentThread().setContextClassLoader(getClass().getClassLoader());
 		return invoke(operation, params);
 	}
 

spring-boot-project/spring-boot-actuator/src/test/java/org/springframework/boot/actuate/endpoint/jmx/EndpointMBeanTests.java

@@ -16,6 +16,9 @@
 
 package org.springframework.boot.actuate.endpoint.jmx;
 
+import java.net.URL;
+import java.net.URLClassLoader;
+
 import javax.management.Attribute;
 import javax.management.AttributeList;
 import javax.management.AttributeNotFoundException;
@@ -29,6 +32,8 @@
 import org.junit.rules.ExpectedException;
 import reactor.core.publisher.Mono;
 
+import org.springframework.util.ClassUtils;
+
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.hamcrest.CoreMatchers.instanceOf;
 import static org.mockito.Mockito.mock;
@@ -84,6 +89,23 @@ public void invokeShouldInvokeJmxOperation()
 		assertThat(result).isEqualTo("result");
 	}
 
+	@Test
+	public void invokeShouldInvokeJmxOperationWithApplicationClassLoader()
+		throws ReflectionException, MBeanException {
+		TestExposableJmxEndpoint endpoint = new TestExposableJmxEndpoint(
+			new TestJmxOperation((arguments) -> ClassUtils.getDefaultClassLoader()));
+		EndpointMBean bean = new EndpointMBean(this.responseMapper, endpoint);
+		Object result;
+		Thread.currentThread().setContextClassLoader(new URLClassLoader(new URL[]{}, getClass().getClassLoader()));
+		try {
+			result = bean.invoke("testOperation", NO_PARAMS, NO_SIGNATURE);
+		}
+		finally {
+			Thread.currentThread().setContextClassLoader(getClass().getClassLoader());
+		}
+		assertThat(result).isEqualTo(getClass().getClassLoader());
+	}
+
 	@Test
 	public void invokeWhenActionNameIsNotAnOperationShouldThrowException()
 			throws MBeanException, ReflectionException {