Apache
/
hbase
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

HBASE-21188 Print heap and gc informations in our junit ResourceChecker

This commit is contained in:
zhangduo 2018-09-12 22:16:27 +08:00
parent 76199a0a29
commit dc1dedb073
1 changed files with 42 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
hbase-common/src/test/java/org/apache/hadoop/hbase/ResourceCheckerJUnitListener.java
View File

@ -18,12 +18,15 @@
package org.apache.hadoop.hbase; package org.apache.hadoop.hbase;
import java.lang.management.ManagementFactory;
import java.lang.management.MemoryUsage;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.HashSet; import java.util.HashSet;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.Set; import java.util.Set;
import java.util.concurrent.ConcurrentHashMap; import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.TimeUnit;
import org.apache.hadoop.hbase.ResourceChecker.Phase; import org.apache.hadoop.hbase.ResourceChecker.Phase;
import org.apache.hadoop.hbase.util.JVM; import org.apache.hadoop.hbase.util.JVM;
import org.junit.runner.notification.RunListener; import org.junit.runner.notification.RunListener;
@ -139,6 +142,41 @@ public class ResourceCheckerJUnitListener extends RunListener {
} }
} }
static class MaxHeapMemoryMBResourceAnalyzer extends ResourceChecker.ResourceAnalyzer {
@Override
public int getVal(Phase phase) {
MemoryUsage usage = ManagementFactory.getMemoryMXBean().getHeapMemoryUsage();
return (int) (usage.getMax() / (1024 * 1024));
}
}
static class UsedHeapMemoryMBResourceAnalyzer extends ResourceChecker.ResourceAnalyzer {
@Override
public int getVal(Phase phase) {
MemoryUsage usage = ManagementFactory.getMemoryMXBean().getHeapMemoryUsage();
return (int) (usage.getUsed() / (1024 * 1024));
}
}
static class GCCountResourceAnalyzer extends ResourceChecker.ResourceAnalyzer {
@Override
public int getVal(Phase phase) {
return Math.toIntExact(ManagementFactory.getGarbageCollectorMXBeans().stream()
.mapToLong(b -> b.getCollectionCount()).sum());
}
}
static class GCTimeSecondResourceAnalyzer extends ResourceChecker.ResourceAnalyzer {
@Override
public int getVal(Phase phase) {
return Math.toIntExact(TimeUnit.MILLISECONDS.toSeconds(ManagementFactory
.getGarbageCollectorMXBeans().stream().mapToLong(b -> b.getCollectionTime()).sum()));
}
}
/** /**
* To be implemented by sub classes if they want to add specific ResourceAnalyzer. * To be implemented by sub classes if they want to add specific ResourceAnalyzer.
@ -155,6 +193,10 @@ public class ResourceCheckerJUnitListener extends RunListener {
rc.addResourceAnalyzer(new SystemLoadAverageResourceAnalyzer()); rc.addResourceAnalyzer(new SystemLoadAverageResourceAnalyzer());
rc.addResourceAnalyzer(new ProcessCountResourceAnalyzer()); rc.addResourceAnalyzer(new ProcessCountResourceAnalyzer());
rc.addResourceAnalyzer(new AvailableMemoryMBResourceAnalyzer()); rc.addResourceAnalyzer(new AvailableMemoryMBResourceAnalyzer());
rc.addResourceAnalyzer(new MaxHeapMemoryMBResourceAnalyzer());
rc.addResourceAnalyzer(new UsedHeapMemoryMBResourceAnalyzer());
rc.addResourceAnalyzer(new GCCountResourceAnalyzer());
rc.addResourceAnalyzer(new GCTimeSecondResourceAnalyzer());
addResourceAnalyzer(rc); addResourceAnalyzer(rc);