mirror of https://github.com/apache/nifi.git
NIFI-9331 Corrected ThreadUtils using Objects.equals() for LockedStackFrame
- Removed unused arguments from ThreadUtils.createStackTrace() Signed-off-by: Pierre Villard <pierre.villard.fr@gmail.com> This closes #5480.
This commit is contained in:
parent
8d513c5ed3
commit
7b2f364cf5
|
@ -1538,7 +1538,7 @@ public class StandardProcessorNode extends ProcessorNode implements Connectable
|
|||
final long activeMillis = now - timestamp;
|
||||
final ThreadInfo threadInfo = threadInfoMap.get(thread.getId());
|
||||
|
||||
final String stackTrace = ThreadUtils.createStackTrace(thread, threadInfo, threadDetails.getDeadlockedThreadIds(), threadDetails.getMonitorDeadlockThreadIds(), activeMillis);
|
||||
final String stackTrace = ThreadUtils.createStackTrace(threadInfo, threadDetails.getDeadlockedThreadIds(), threadDetails.getMonitorDeadlockThreadIds());
|
||||
|
||||
final ActiveThreadInfo activeThreadInfo = new ActiveThreadInfo(thread.getName(), stackTrace, activeMillis, activeTask.isTerminated());
|
||||
threadList.add(activeThreadInfo);
|
||||
|
|
|
@ -20,10 +20,11 @@ package org.apache.nifi.util;
|
|||
import java.lang.management.LockInfo;
|
||||
import java.lang.management.MonitorInfo;
|
||||
import java.lang.management.ThreadInfo;
|
||||
import java.util.Objects;
|
||||
|
||||
public class ThreadUtils {
|
||||
|
||||
public static String createStackTrace(final Thread thread, final ThreadInfo threadInfo, final long[] deadlockedThreadIds, final long[] monitorDeadlockThreadIds, final long activeMillis) {
|
||||
public static String createStackTrace(final ThreadInfo threadInfo, final long[] deadlockedThreadIds, final long[] monitorDeadlockThreadIds) {
|
||||
final StringBuilder sb = new StringBuilder();
|
||||
sb.append("\"").append(threadInfo.getThreadName()).append("\" Id=");
|
||||
sb.append(threadInfo.getThreadId()).append(" ");
|
||||
|
@ -69,7 +70,7 @@ public class ThreadUtils {
|
|||
|
||||
final MonitorInfo[] monitors = threadInfo.getLockedMonitors();
|
||||
for (final MonitorInfo monitor : monitors) {
|
||||
if (monitor.getLockedStackFrame().equals(element)) {
|
||||
if (Objects.equals(monitor.getLockedStackFrame(), element)) {
|
||||
sb.append("\n\t- waiting on ").append(monitor);
|
||||
}
|
||||
}
|
||||
|
|
|
@ -0,0 +1,99 @@
|
|||
/*
|
||||
* Licensed to the Apache Software Foundation (ASF) under one or more
|
||||
* contributor license agreements. See the NOTICE file distributed with
|
||||
* this work for additional information regarding copyright ownership.
|
||||
* The ASF licenses this file to You 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
|
||||
*
|
||||
* http://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.apache.nifi.util;
|
||||
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.junit.jupiter.api.extension.ExtendWith;
|
||||
import org.mockito.Mock;
|
||||
import org.mockito.junit.jupiter.MockitoExtension;
|
||||
|
||||
import java.lang.management.LockInfo;
|
||||
import java.lang.management.MonitorInfo;
|
||||
import java.lang.management.ThreadInfo;
|
||||
|
||||
import static org.junit.jupiter.api.Assertions.assertNotNull;
|
||||
import static org.junit.jupiter.api.Assertions.assertTrue;
|
||||
import static org.mockito.Mockito.when;
|
||||
|
||||
@ExtendWith(MockitoExtension.class)
|
||||
public class ThreadUtilsTest {
|
||||
private static final String THREAD_NAME = "TestThread";
|
||||
|
||||
private static final String DECLARING_CLASS = ThreadUtilsTest.class.getName();
|
||||
|
||||
private static final String METHOD = "createStackTrace";
|
||||
|
||||
private static final String FILE_NAME = ThreadUtilsTest.class.getSimpleName();
|
||||
|
||||
private static final String MONITOR_CLASS = String.class.getName();
|
||||
|
||||
private static final int LINE_NUMBER = 100;
|
||||
|
||||
private static final int IDENTITY_HASH_CODE = 1024;
|
||||
|
||||
private static final int STACK_DEPTH = 1;
|
||||
|
||||
private static final int NULL_STACK_DEPTH = -1;
|
||||
|
||||
@Mock
|
||||
private ThreadInfo threadInfo;
|
||||
|
||||
@Test
|
||||
public void testCreateStackTrace() {
|
||||
final Thread.State threadState = Thread.State.BLOCKED;
|
||||
setThreadInfo(threadState, getStackTraceElement(MONITOR_CLASS));
|
||||
|
||||
final String stackTrace = ThreadUtils.createStackTrace(threadInfo, null, null);
|
||||
assertThreadInfoFound(stackTrace, threadState);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCreateStackTraceNullLockedStackFrame() {
|
||||
final Thread.State threadState = Thread.State.RUNNABLE;
|
||||
setThreadInfo(threadState, null);
|
||||
|
||||
final String stackTrace = ThreadUtils.createStackTrace(threadInfo, null, null);
|
||||
assertThreadInfoFound(stackTrace, threadState);
|
||||
}
|
||||
|
||||
private void setThreadInfo(final Thread.State threadState, final StackTraceElement lockedStackFrame) {
|
||||
when(threadInfo.getThreadName()).thenReturn(THREAD_NAME);
|
||||
|
||||
when(threadInfo.getThreadState()).thenReturn(threadState);
|
||||
|
||||
final StackTraceElement stackTraceElement = getStackTraceElement(DECLARING_CLASS);
|
||||
when(threadInfo.getStackTrace()).thenReturn(new StackTraceElement[]{stackTraceElement});
|
||||
|
||||
final int stackDepth = lockedStackFrame == null ? NULL_STACK_DEPTH : STACK_DEPTH;
|
||||
final MonitorInfo monitorInfo = new MonitorInfo(MONITOR_CLASS, IDENTITY_HASH_CODE, stackDepth, lockedStackFrame);
|
||||
when(threadInfo.getLockedMonitors()).thenReturn(new MonitorInfo[]{monitorInfo});
|
||||
|
||||
when(threadInfo.getLockedSynchronizers()).thenReturn(new LockInfo[]{});
|
||||
}
|
||||
|
||||
private void assertThreadInfoFound(final String stackTrace, final Thread.State threadState) {
|
||||
assertNotNull(stackTrace);
|
||||
assertTrue(stackTrace.contains(THREAD_NAME), "Thread Name not found");
|
||||
assertTrue(stackTrace.contains(threadState.toString()), "Thread State not found");
|
||||
assertTrue(stackTrace.contains(DECLARING_CLASS), "Stack Trace declaring class not found");
|
||||
assertTrue(stackTrace.contains(METHOD), "Stack Trace method not found");
|
||||
}
|
||||
|
||||
private StackTraceElement getStackTraceElement(final String declaringClass) {
|
||||
return new StackTraceElement(declaringClass, METHOD, FILE_NAME, LINE_NUMBER);
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue