View Javadoc

1   /**
2    *
3    * Licensed to the Apache Software Foundation (ASF) under one
4    * or more contributor license agreements.  See the NOTICE file
5    * distributed with this work for additional information
6    * regarding copyright ownership.  The ASF licenses this file
7    * to you under the Apache License, Version 2.0 (the
8    * "License"); you may not use this file except in compliance
9    * with the License.  You may obtain a copy of the License at
10   *
11   *     http://www.apache.org/licenses/LICENSE-2.0
12   *
13   * Unless required by applicable law or agreed to in writing, software
14   * distributed under the License is distributed on an "AS IS" BASIS,
15   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16   * See the License for the specific language governing permissions and
17   * limitations under the License.
18   */
19  
20  package org.apache.hadoop.hbase.zookeeper;
21  
22  import java.util.LinkedList;
23  import java.util.List;
24  import java.util.Properties;
25  import java.util.Map.Entry;
26  
27  import org.apache.hadoop.hbase.HConstants;
28  import org.apache.hadoop.hbase.ServerName;
29  import org.apache.hadoop.hbase.classification.InterfaceAudience;
30  import org.apache.hadoop.conf.Configuration;
31  import org.apache.hadoop.hbase.HBaseConfiguration;
32  import org.apache.hadoop.hbase.HBaseInterfaceAudience;
33  
34  /**
35   * Tool for reading ZooKeeper servers from HBase XML configuration and producing
36   * a line-by-line list for use by bash scripts.
37   */
38  @InterfaceAudience.LimitedPrivate(HBaseInterfaceAudience.TOOLS)
39  public class ZKServerTool {
40  
41    public static ServerName[] readZKNodes(Configuration conf) {
42      List<ServerName> hosts = new LinkedList<ServerName>();
43  
44      // Note that we do not simply grab the property
45      // HConstants.ZOOKEEPER_QUORUM from the HBaseConfiguration because the
46      // user may be using a zoo.cfg file.
47      Properties zkProps = ZKConfig.makeZKProps(conf);
48      for (Entry<Object, Object> entry : zkProps.entrySet()) {
49        String key = entry.getKey().toString().trim();
50        String value = entry.getValue().toString().trim();
51        if (key.startsWith("server.")) {
52          String[] parts = value.split(":");
53          String host = parts[0];
54  
55          int port = HConstants.DEFAULT_ZOOKEPER_CLIENT_PORT;
56          if (parts.length > 1) {
57            port = Integer.parseInt(parts[1]);
58          }
59          hosts.add(ServerName.valueOf(host, port, -1));
60        }
61      }
62      return hosts.toArray(new ServerName[hosts.size()]);
63    }
64  
65    /**
66     * Run the tool.
67     * @param args Command line arguments.
68     */
69    public static void main(String args[]) {
70      for(ServerName server: readZKNodes(HBaseConfiguration.create())) {
71        // bin/zookeeper.sh relies on the "ZK host" string for grepping which is case sensitive.
72        System.out.println("ZK host: " + server.getHostname());
73      }
74    }
75  }