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 java.util.ArrayList;
21 import java.util.List;
22 import java.util.concurrent.BlockingQueue;
23 import java.util.concurrent.LinkedBlockingQueue;
24
25 import org.apache.hadoop.conf.Configuration;
26 import org.apache.hadoop.hbase.Abortable;
27 import org.apache.hadoop.hbase.HBaseInterfaceAudience;
28 import org.apache.hadoop.hbase.classification.InterfaceAudience;
29 import org.apache.hadoop.hbase.classification.InterfaceStability;
30 import org.apache.hadoop.hbase.util.ReflectionUtils;
31
32
33
34
35
36 @InterfaceAudience.LimitedPrivate({ HBaseInterfaceAudience.COPROC, HBaseInterfaceAudience.PHOENIX })
37 @InterfaceStability.Evolving
38 public class BalancedQueueRpcExecutor extends RpcExecutor {
39
40 protected final List<BlockingQueue<CallRunner>> queues;
41 private final QueueBalancer balancer;
42
43 public BalancedQueueRpcExecutor(final String name, final int handlerCount, final int numQueues,
44 final int maxQueueLength) {
45 this(name, handlerCount, numQueues, maxQueueLength, null, null);
46 }
47
48 public BalancedQueueRpcExecutor(final String name, final int handlerCount, final int numQueues,
49 final int maxQueueLength, final Configuration conf, final Abortable abortable) {
50 this(name, handlerCount, numQueues, conf, abortable, LinkedBlockingQueue.class, maxQueueLength);
51 }
52
53 public BalancedQueueRpcExecutor(final String name, final int handlerCount, final int numQueues,
54 final Configuration conf, final Abortable abortable,
55 final Class<? extends BlockingQueue> queueClass, Object... initargs) {
56 super(name, Math.max(handlerCount, numQueues), conf, abortable);
57 queues = new ArrayList<BlockingQueue<CallRunner>>(numQueues);
58 this.balancer = getBalancer(numQueues);
59 initializeQueues(numQueues, queueClass, initargs);
60 }
61
62 protected void initializeQueues(final int numQueues,
63 final Class<? extends BlockingQueue> queueClass, Object... initargs) {
64 for (int i = 0; i < numQueues; ++i) {
65 queues.add((BlockingQueue<CallRunner>) ReflectionUtils.newInstance(queueClass, initargs));
66 }
67 }
68
69 @Override
70 public void dispatch(final CallRunner callTask) throws InterruptedException {
71 int queueIndex = balancer.getNextQueue();
72 queues.get(queueIndex).put(callTask);
73 }
74
75 @Override
76 public int getQueueLength() {
77 int length = 0;
78 for (final BlockingQueue<CallRunner> queue : queues) {
79 length += queue.size();
80 }
81 return length;
82 }
83
84 @Override
85 public List<BlockingQueue<CallRunner>> getQueues() {
86 return queues;
87 }
88 }