Spring/spring-boot
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Allow thread dump endpoint to call ThreadMXBean in a native image

Browse Source 
Closes gh-31680
This commit is contained in:
Andy Wilkinson
2022-10-04 16:58:08 +01:00
parent e6568596b9
commit 7c4e46e538
5 changed files with 6 additions and 183 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
spring-boot-project/spring-boot-actuator/src/main/java/org/springframework/boot/actuate/management/ThreadDumpEndpoint.java
View File

@@ -24,7 +24,6 @@ import java.util.function.Function;
import org.springframework.boot.actuate.endpoint.annotation.Endpoint;
import org.springframework.boot.actuate.endpoint.annotation.ReadOperation;
import org.springframework.core.NativeDetector;
/**
* {@link Endpoint @Endpoint} to expose thread info.
@@ -49,41 +48,7 @@ public class ThreadDumpEndpoint {
}
private <T> T getFormattedThreadDump(Function<ThreadInfo[], T> formatter) {
ThreadDumper threadDumper = createThreadDumper();
return formatter.apply(threadDumper.dumpAllThreads());
}
private ThreadDumper createThreadDumper() {
if (NativeDetector.inNativeImage()) {
throw new ThreadDumperUnavailableException("Running in native image");
}
return new ThreadMXBeanThreadDumper();
}
private interface ThreadDumper {
ThreadInfo[] dumpAllThreads();
}
private static class ThreadMXBeanThreadDumper implements ThreadDumper {
@Override
public ThreadInfo[] dumpAllThreads() {
return ManagementFactory.getThreadMXBean().dumpAllThreads(true, true);
}
}
/**
* Exception to be thrown if the {@link ThreadDumper} cannot be created.
*/
static class ThreadDumperUnavailableException extends RuntimeException {
ThreadDumperUnavailableException(String message) {
super(message);
}
return formatter.apply(ManagementFactory.getThreadMXBean().dumpAllThreads(true, true));
}
/**

60
spring-boot-project/spring-boot-actuator/src/main/java/org/springframework/boot/actuate/management/ThreadDumpEndpointWebExtension.java
View File

@@ -1,60 +0,0 @@
/*
* Copyright 2012-2022 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.springframework.boot.actuate.management;
import org.springframework.boot.actuate.endpoint.annotation.ReadOperation;
import org.springframework.boot.actuate.endpoint.web.WebEndpointResponse;
import org.springframework.boot.actuate.endpoint.web.annotation.EndpointWebExtension;
import org.springframework.boot.actuate.management.ThreadDumpEndpoint.ThreadDumpDescriptor;
import org.springframework.boot.actuate.management.ThreadDumpEndpoint.ThreadDumperUnavailableException;
/**
* {@link EndpointWebExtension @EndpointWebExtension} for the {@link ThreadDumpEndpoint}.
*
* @author Moritz Halbritter
* @since 3.0.0
*/
@EndpointWebExtension(endpoint = ThreadDumpEndpoint.class)
public class ThreadDumpEndpointWebExtension {
private final ThreadDumpEndpoint delegate;
public ThreadDumpEndpointWebExtension(ThreadDumpEndpoint delegate) {
this.delegate = delegate;
}
@ReadOperation
public WebEndpointResponse<ThreadDumpDescriptor> threadDump() {
try {
return new WebEndpointResponse<>(this.delegate.threadDump());
}
catch (ThreadDumperUnavailableException ex) {
return new WebEndpointResponse<>(WebEndpointResponse.STATUS_SERVICE_UNAVAILABLE);
}
}
@ReadOperation(produces = "text/plain;charset=UTF-8")
public WebEndpointResponse<String> textThreadDump() {
try {
return new WebEndpointResponse<>(this.delegate.textThreadDump());
}
catch (ThreadDumperUnavailableException ex) {
return new WebEndpointResponse<>(WebEndpointResponse.STATUS_SERVICE_UNAVAILABLE);
}
}
}

63
spring-boot-project/spring-boot-actuator/src/test/java/org/springframework/boot/actuate/management/ThreadDumpEndpointWebExtensionTests.java
View File

@@ -1,63 +0,0 @@
/*
* Copyright 2012-2022 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.springframework.boot.actuate.management;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.mockito.Mockito;
import org.springframework.boot.actuate.endpoint.web.WebEndpointResponse;
import org.springframework.boot.actuate.management.ThreadDumpEndpoint.ThreadDumpDescriptor;
import org.springframework.boot.actuate.management.ThreadDumpEndpoint.ThreadDumperUnavailableException;
import org.springframework.http.HttpStatus;
import static org.assertj.core.api.Assertions.assertThat;
/**
* Tests for {@link ThreadDumpEndpointWebExtension}.
*
* @author Moritz Halbritter
*/
class ThreadDumpEndpointWebExtensionTests {
private ThreadDumpEndpointWebExtension extension;
private ThreadDumpEndpoint delegateMock;
@BeforeEach
void setUp() {
this.delegateMock = Mockito.mock(ThreadDumpEndpoint.class);
this.extension = new ThreadDumpEndpointWebExtension(this.delegateMock);
}
@Test
void shouldHandleThreadDumperUnavailable() {
Mockito.when(this.delegateMock.threadDump())
.thenThrow(new ThreadDumperUnavailableException("No thread dumper available"));
WebEndpointResponse<ThreadDumpDescriptor> response = this.extension.threadDump();
assertThat(response.getStatus()).isEqualTo(HttpStatus.SERVICE_UNAVAILABLE.value());
}
@Test
void shouldHandleThreadDumperUnavailableText() {
Mockito.when(this.delegateMock.textThreadDump())
.thenThrow(new ThreadDumperUnavailableException("No thread dumper available"));
WebEndpointResponse<String> response = this.extension.textThreadDump();
assertThat(response.getStatus()).isEqualTo(HttpStatus.SERVICE_UNAVAILABLE.value());
}
}