FilesExpand file tree

 master

/

find_active_apache_drill.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

executable file

·

84 lines (65 loc) · 2.52 KB

 master

/

find_active_apache_drill.py

Copy path

Top

File metadata and controls

• Code
• Blame

executable file

·

84 lines (65 loc) · 2.52 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

84

#!/usr/bin/env python

# 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 responsive Apache Drill 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 Apache Drill 0.7, 0.8, 0.9, 1.0, 1.1, 1.2, 1.3, 1.4, 1.5, 1.6, 1.7, 1.8, 1.9, 1.10, 1.11, 1.12, 1.13, 1.14, 1.15

"""

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 FindActiveApacheDrill(FindActiveServer):

def __init__(self):

# Python 2.x

super(FindActiveApacheDrill, self).__init__()

# Python 3.x

# super().__init__()

self.default_port = 8047

self.protocol = 'http'

self.url_path = '/status'

self.regex = r'Running'

def add_options(self):

self.add_hostoption(name='Apache Drill', default_port=self.default_port)

self.add_ssl_opt()

self.add_common_opts()

def process_options(self):

self.validate_common_opts()

if __name__ == '__main__':

FindActiveApacheDrill().main()