Merge branch '2.0.x'

pull/12550/merge
Stephane Nicoll 7 years ago
commit 3cb2246e7a

@ -93,27 +93,22 @@ public class EndpointMBean implements DynamicMBean {
return invoke(operation, params); return invoke(operation, params);
} }
private Object invoke(JmxOperation operation, Object[] params) throws MBeanException { private Object invoke(JmxOperation operation, Object[] params)
throws MBeanException, ReflectionException {
try { try {
String[] parameterNames = operation.getParameters().stream() String[] parameterNames = operation.getParameters().stream()
.map(JmxOperationParameter::getName).toArray(String[]::new); .map(JmxOperationParameter::getName).toArray(String[]::new);
Map<String, Object> arguments = getArguments(parameterNames, params); Map<String, Object> arguments = getArguments(parameterNames, params);
Object result = invokeOperation(operation, arguments); Object result = operation
.invoke(new InvocationContext(SecurityContext.NONE, arguments));
if (REACTOR_PRESENT) { if (REACTOR_PRESENT) {
result = ReactiveHandler.handle(result); result = ReactiveHandler.handle(result);
} }
return this.responseMapper.mapResponse(result); return this.responseMapper.mapResponse(result);
} }
catch (InvalidEndpointRequestException ex) { catch (InvalidEndpointRequestException ex) {
throw new IllegalArgumentException(ex.getMessage(), ex); throw new ReflectionException(new IllegalArgumentException(
} ex.getMessage()), ex.getMessage());
}
private Object invokeOperation(JmxOperation operation,
Map<String, Object> arguments) throws MBeanException {
try {
return operation.invoke(new InvocationContext(SecurityContext.NONE,
arguments));
} }
catch (Exception ex) { catch (Exception ex) {
throw new MBeanException(translateIfNecessary(ex), ex.getMessage()); throw new MBeanException(translateIfNecessary(ex), ex.getMessage());

@ -30,6 +30,8 @@ import org.junit.rules.ExpectedException;
import reactor.core.publisher.Mono; import reactor.core.publisher.Mono;
import org.springframework.beans.FatalBeanException; import org.springframework.beans.FatalBeanException;
import org.springframework.boot.actuate.endpoint.InvalidEndpointRequestException;
import org.springframework.boot.actuate.endpoint.InvocationContext;
import static org.assertj.core.api.Assertions.assertThat; import static org.assertj.core.api.Assertions.assertThat;
import static org.hamcrest.CoreMatchers.instanceOf; import static org.hamcrest.CoreMatchers.instanceOf;
@ -124,6 +126,23 @@ public class EndpointMBeanTests {
bean.invoke("missingOperation", NO_PARAMS, NO_SIGNATURE); bean.invoke("missingOperation", NO_PARAMS, NO_SIGNATURE);
} }
@Test
public void invokeWhenOperationIsInvalidShouldThrowException()
throws MBeanException, ReflectionException {
TestJmxOperation operation = new TestJmxOperation() {
@Override
public Object invoke(InvocationContext context) {
throw new InvalidEndpointRequestException("test failure", "test");
}
};
TestExposableJmxEndpoint endpoint = new TestExposableJmxEndpoint(operation);
EndpointMBean bean = new EndpointMBean(this.responseMapper, endpoint);
this.thrown.expect(ReflectionException.class);
this.thrown.expectCause(instanceOf(IllegalArgumentException.class));
this.thrown.expectMessage("test failure");
bean.invoke("testOperation", NO_PARAMS, NO_SIGNATURE);
}
@Test @Test
public void invokeWhenMonoResultShouldBlockOnMono() public void invokeWhenMonoResultShouldBlockOnMono()
throws MBeanException, ReflectionException { throws MBeanException, ReflectionException {

Loading…
Cancel
Save