1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18 package org.apache.hadoop.hbase.client;
19
20 import com.google.common.annotations.VisibleForTesting;
21 import org.apache.hadoop.conf.Configuration;
22 import org.apache.hadoop.hbase.HConstants;
23 import org.apache.hadoop.hbase.ServerName;
24 import org.apache.hadoop.hbase.classification.InterfaceAudience;
25 import org.apache.hadoop.hbase.client.backoff.ServerStatistics;
26 import org.apache.hadoop.hbase.protobuf.generated.ClientProtos;
27
28 import java.util.Map;
29 import java.util.concurrent.ConcurrentHashMap;
30
31
32
33
34 @InterfaceAudience.Private
35 public class ServerStatisticTracker {
36
37 private final Map<ServerName, ServerStatistics> stats =
38 new ConcurrentHashMap<ServerName, ServerStatistics>();
39
40 public void updateRegionStats(ServerName server, byte[] region, ClientProtos.RegionLoadStats
41 currentStats) {
42 ServerStatistics stat = stats.get(server);
43
44 if (stat == null) {
45
46 synchronized (this) {
47 stat = stats.get(server);
48
49 if (stat == null) {
50 stat = new ServerStatistics();
51 stats.put(server, stat);
52 }
53 }
54 }
55 stat.update(region, currentStats);
56 }
57
58 public ServerStatistics getStats(ServerName server) {
59 return this.stats.get(server);
60 }
61
62 public static ServerStatisticTracker create(Configuration conf) {
63 if (!conf.getBoolean(HConstants.ENABLE_CLIENT_BACKPRESSURE,
64 HConstants.DEFAULT_ENABLE_CLIENT_BACKPRESSURE)) {
65 return null;
66 }
67 return new ServerStatisticTracker();
68 }
69
70 @VisibleForTesting
71 ServerStatistics getServerStatsForTesting(ServerName server) {
72 return stats.get(server);
73 }
74 }