1 package storm.scheduler;
2
3 import java.lang.management.ManagementFactory;
4 import java.lang.management.ThreadMXBean;
5 import java.util.HashMap;
6 import java.util.Map;
7 import java.util.Set;
8
9 import cpuinfo.CPUInfo;
10
11 /**
12 * 负载监视器
13 * @author wxweven
14 * @version 1.0
15 * @email wxweven@qq.com
16 * @blog http://wxweven.com
17 * @Copyright: Copyright (c) wxweven 2009 - 2016
18 */
19 public class LoadMonitor {
20
21 private static final int SECS_TO_NANOSECS = 1000000000
;
22 private static LoadMonitor instance =
null;
23 private final long cpuSpeed;
// Hz
24 Map<Long, Long>
loadHistory;
25
26 public static LoadMonitor getInstance() {
27 if (instance ==
null) {
28 instance =
new LoadMonitor();
29 }
30 return instance;
31 }
32
33 private LoadMonitor() {
34 cpuSpeed = CPUInfo.getInstance().getCoreInfo(0
).getSpeed();
35 }
36
37 public Map<Long, Long> getLoadInfo(Set<Long>
threadIds) {
38 // get current load
39 Map<Long, Long> currentLoadInfo =
new HashMap<Long, Long>
();
40 ThreadMXBean threadBean =
ManagementFactory.getThreadMXBean();
41 for (
long id : threadIds) {
42 currentLoadInfo.put(id, threadBean.getThreadCpuTime(id));
43 }
44
45 // compute difference wrt history
46 Map<Long, Long> loadInfo =
new HashMap<Long, Long>
();
47 for (
long id : threadIds) {
48 // Long oldObj = (loadHistory != null)?loadHistory.get(id):0;
49 // long old = (oldObj != null)?oldObj.longValue():0;
50 long old = 0
;
51 if (loadHistory !=
null && loadHistory.get(id) !=
null) {
52 old =
loadHistory.get(id);
53 }
54 double deltaTime = (
double)(currentLoadInfo.get(id) - old) / SECS_TO_NANOSECS;
// sec
55 loadInfo.put(id, (
long)(deltaTime *
cpuSpeed));
56 }
57
58 // replace history with current
59 loadHistory =
currentLoadInfo;
60
61 return loadInfo;
62 }
63 }
转载于:https://www.cnblogs.com/wxweven/p/5517194.html
