1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18 package org.apache.hadoop.hbase.zookeeper;
19
20 import org.apache.hadoop.classification.InterfaceAudience;
21 import org.apache.hadoop.hbase.Abortable;
22 import org.apache.hadoop.hbase.ServerName;
23 import org.apache.hadoop.hbase.exceptions.DeserializationException;
24 import org.apache.hadoop.hbase.protobuf.ProtobufUtil;
25 import org.apache.hadoop.hbase.protobuf.generated.HBaseProtos;
26 import org.apache.hadoop.hbase.protobuf.generated.ZooKeeperProtos;
27 import org.apache.zookeeper.KeeperException;
28 import org.apache.zookeeper.data.Stat;
29
30 import java.io.IOException;
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50 @InterfaceAudience.Private
51 public class MasterAddressTracker extends ZooKeeperNodeTracker {
52
53
54
55
56
57
58
59
60
61
62
63 public MasterAddressTracker(ZooKeeperWatcher watcher, Abortable abortable) {
64 super(watcher, watcher.getMasterAddressZNode(), abortable);
65 }
66
67
68
69
70
71
72 public ServerName getMasterAddress() {
73 return getMasterAddress(false);
74 }
75
76
77
78
79
80
81
82
83
84 public ServerName getMasterAddress(final boolean refresh) {
85 try {
86 return ServerName.parseFrom(super.getData(refresh));
87 } catch (DeserializationException e) {
88 LOG.warn("Failed parse", e);
89 return null;
90 }
91 }
92
93
94
95
96
97
98
99
100
101
102
103 public static ServerName getMasterAddress(final ZooKeeperWatcher zkw)
104 throws KeeperException, IOException {
105 byte [] data = ZKUtil.getData(zkw, zkw.getMasterAddressZNode());
106 if (data == null){
107 throw new IOException("Can't get master address from ZooKeeper; znode data == null");
108 }
109 try {
110 return ServerName.parseFrom(data);
111 } catch (DeserializationException e) {
112 KeeperException ke = new KeeperException.DataInconsistencyException();
113 ke.initCause(e);
114 throw ke;
115 }
116 }
117
118
119
120
121
122
123
124
125
126
127
128
129 public static boolean setMasterAddress(final ZooKeeperWatcher zkw,
130 final String znode, final ServerName master)
131 throws KeeperException {
132 return ZKUtil.createEphemeralNodeAndWatch(zkw, znode, toByteArray(master));
133 }
134
135
136
137
138
139 public boolean hasMaster() {
140 return super.getData(false) != null;
141 }
142
143
144
145
146
147
148 static byte [] toByteArray(final ServerName sn) {
149 ZooKeeperProtos.Master.Builder mbuilder = ZooKeeperProtos.Master.newBuilder();
150 HBaseProtos.ServerName.Builder snbuilder = HBaseProtos.ServerName.newBuilder();
151 snbuilder.setHostName(sn.getHostname());
152 snbuilder.setPort(sn.getPort());
153 snbuilder.setStartCode(sn.getStartcode());
154 mbuilder.setMaster(snbuilder.build());
155 return ProtobufUtil.prependPBMagic(mbuilder.build().toByteArray());
156 }
157
158
159
160
161 public static boolean deleteIfEquals(ZooKeeperWatcher zkw, final String content) {
162 if (content == null){
163 throw new IllegalArgumentException("Content must not be null");
164 }
165
166 try {
167 Stat stat = new Stat();
168 byte[] data = ZKUtil.getDataNoWatch(zkw, zkw.getMasterAddressZNode(), stat);
169 ServerName sn = ServerName.parseFrom(data);
170 if (sn != null && content.equals(sn.toString())) {
171 return (ZKUtil.deleteNode(zkw, zkw.getMasterAddressZNode(), stat.getVersion()));
172 }
173 } catch (KeeperException e) {
174 LOG.warn("Can't get or delete the master znode", e);
175 } catch (DeserializationException e) {
176 LOG.warn("Can't get or delete the master znode", e);
177 }
178
179 return false;
180 }
181 }