HBASE-17673 Monitored RPC Handler not shown in the WebUI (Allan Yang)

This commit is contained in:
tedyu 2017-02-28 07:43:32 -08:00
parent f4e0ea24e6
commit 4a5eba5e59
2 changed files with 30 additions and 2 deletions

View File

@ -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;
}

View File

@ -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);
}
}