1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18 package org.apache.hadoop.hbase.ipc;
19
20 import org.apache.hadoop.conf.Configuration;
21 import org.apache.hadoop.hbase.DaemonThreadFactory;
22
23 import java.io.IOException;
24 import java.util.concurrent.ArrayBlockingQueue;
25 import java.util.concurrent.ThreadPoolExecutor;
26 import java.util.concurrent.TimeUnit;
27
28
29
30
31
32
33 public class FifoRpcScheduler extends RpcScheduler {
34
35 private final int handlerCount;
36 private final int maxQueueLength;
37 private ThreadPoolExecutor executor;
38
39 public FifoRpcScheduler(Configuration conf, int handlerCount) {
40 this.handlerCount = handlerCount;
41 this.maxQueueLength = conf.getInt("hbase.ipc.server.max.callqueue.length",
42 conf.getInt("ipc.server.max.callqueue.length",
43 handlerCount * RpcServer.DEFAULT_MAX_CALLQUEUE_LENGTH_PER_HANDLER));
44 }
45
46 @Override
47 public void init(Context context) {
48
49 }
50
51 @Override
52 public void start() {
53 this.executor = new ThreadPoolExecutor(
54 handlerCount,
55 handlerCount,
56 60,
57 TimeUnit.SECONDS,
58 new ArrayBlockingQueue<Runnable>(maxQueueLength),
59 new DaemonThreadFactory("FifoRpcScheduler.handler"),
60 new ThreadPoolExecutor.CallerRunsPolicy());
61 }
62
63 @Override
64 public void stop() {
65 this.executor.shutdown();
66 }
67
68 @Override
69 public void dispatch(final CallRunner task) throws IOException, InterruptedException {
70 executor.submit(new Runnable() {
71 @Override
72 public void run() {
73 task.setStatus(RpcServer.getStatus());
74 task.run();
75 }
76 });
77 }
78
79 @Override
80 public int getGeneralQueueLength() {
81 return executor.getQueue().size();
82 }
83
84 @Override
85 public int getPriorityQueueLength() {
86 return 0;
87 }
88
89 @Override
90 public int getReplicationQueueLength() {
91 return 0;
92 }
93
94 @Override
95 public int getActiveRpcHandlerCount() {
96 return executor.getActiveCount();
97 }
98 }