HBASE-17673 Monitored RPC Handler not shown in the WebUI (Allan Yang)
This commit is contained in:
parent
f4e0ea24e6
commit
4a5eba5e59
|
@ -53,6 +53,7 @@ public class TaskMonitor {
|
|||
|
||||
private static TaskMonitor instance;
|
||||
private CircularFifoBuffer tasks = new CircularFifoBuffer(MAX_TASKS);
|
||||
private List<TaskAndWeakRefPair> rpcTasks = Lists.newArrayList();
|
||||
|
||||
/**
|
||||
* Get singleton instance.
|
||||
|
@ -88,7 +89,7 @@ public class TaskMonitor {
|
|||
new Class<?>[] { MonitoredRPCHandler.class },
|
||||
new PassthroughInvocationHandler<MonitoredRPCHandler>(stat));
|
||||
TaskAndWeakRefPair pair = new TaskAndWeakRefPair(stat, proxy);
|
||||
tasks.add(pair);
|
||||
rpcTasks.add(pair);
|
||||
return proxy;
|
||||
}
|
||||
|
||||
|
@ -120,13 +121,19 @@ public class TaskMonitor {
|
|||
*/
|
||||
public synchronized List<MonitoredTask> getTasks() {
|
||||
purgeExpiredTasks();
|
||||
ArrayList<MonitoredTask> ret = Lists.newArrayListWithCapacity(tasks.size());
|
||||
ArrayList<MonitoredTask> ret = Lists.newArrayListWithCapacity(tasks.size() + rpcTasks.size());
|
||||
for (Iterator<TaskAndWeakRefPair> it = tasks.iterator();
|
||||
it.hasNext();) {
|
||||
TaskAndWeakRefPair pair = it.next();
|
||||
MonitoredTask t = pair.get();
|
||||
ret.add(t.clone());
|
||||
}
|
||||
for (Iterator<TaskAndWeakRefPair> it = rpcTasks.iterator();
|
||||
it.hasNext();) {
|
||||
TaskAndWeakRefPair pair = it.next();
|
||||
MonitoredTask t = pair.get();
|
||||
ret.add(t.clone());
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
|
|
|
@ -101,6 +101,27 @@ public class TestTaskMonitor {
|
|||
assertEquals("task 10", tm.getTasks().get(0).getDescription());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testDoNotPurgeRPCTask() throws Exception {
|
||||
int RPCTaskNums = 10;
|
||||
for(int i = 0; i < RPCTaskNums; i++) {
|
||||
TaskMonitor.get().createRPCStatus("PRCTask" + i);
|
||||
}
|
||||
for(int i = 0; i < TaskMonitor.MAX_TASKS; i++) {
|
||||
TaskMonitor.get().createStatus("otherTask" + i);
|
||||
}
|
||||
int remainRPCTask = 0;
|
||||
for(MonitoredTask task :TaskMonitor.get().getTasks()) {
|
||||
if(task instanceof MonitoredRPCHandler) {
|
||||
remainRPCTask++;
|
||||
}
|
||||
}
|
||||
assertEquals("RPC Tasks have been purged!", RPCTaskNums, remainRPCTask);
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in New Issue