/
find_active_impala.py
executable file
·83 lines (64 loc) · 2.38 KB
/
find_active_impala.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
#!/usr/bin/env python3
# vim:ts=4:sts=4:sw=4:et
#
# Author: Hari Sekhon
# Date: Wed Sep 13 13:58:21 CEST 2017
#
# https://github.com/HariSekhon/DevOps-Python-tools
#
# License: see accompanying Hari Sekhon LICENSE file
#
# If you're using my code you're welcome to connect with me on LinkedIn
# and optionally send me feedback to help steer this or other code I publish
#
# https://www.linkedin.com/in/HariSekhon
#
"""
Tool to return the first active Impala node from an argument list of hosts
Can mix and match between a comma separated list of hosts (--host server1,server2 or contents of the $HOST
environment variable if not specified) and general free-form space separated arguments, which is useful if piping
a host list through xargs.
Multi-threaded for speed and exits upon first available host response to minimize delay to ~ 1 second or less.
Useful for simplying scripting or generically extending tools that don't support HBase High Availability directly
By default checks the same --port on all servers. Hosts may have optional :<port> suffixes added to individually
override each one.
Exits with return code 1 and NO_AVAILABLE_SERVER if none of the namenodes are active, --quiet mode will not print
NO_AVAILABLE_SERVER.
Tested on Impala 2.9.0-cdh5.12.0
"""
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
#from __future__ import unicode_literals
import os
import sys
import traceback
#from random import shuffle
srcdir = os.path.abspath(os.path.dirname(__file__))
libdir = os.path.join(srcdir, 'pylib')
sys.path.append(libdir)
try:
# pylint: disable=wrong-import-position
from find_active_server import FindActiveServer
except ImportError as _:
print(traceback.format_exc(), end='')
sys.exit(4)
__author__ = 'Hari Sekhon'
__version__ = '0.7.1'
class FindActiveImpala(FindActiveServer):
def __init__(self):
# Python 2.x
super(FindActiveImpala, self).__init__()
# Python 3.x
# super().__init__()
self.default_port = 25000
self.protocol = 'http'
self.regex = r'\bimpalad\b'
def add_options(self):
self.add_hostoption(name='Impala', default_port=self.default_port)
self.add_ssl_opt()
self.add_common_opts()
def process_options(self):
self.validate_common_opts()
if __name__ == '__main__':
FindActiveImpala().main()