catalog-be/src/main/resources/scripts/sdcBePy/common/healthCheck.py

   1 #!/usr/bin/env python3
   2 import sys
   3 import time
   4 from argparse import ArgumentParser
   5 from datetime import datetime
   6
   7 from sdcBePy import properties
   8 from sdcBePy.common.bColors import BColors
   9 from sdcBePy.common.properties import init_properties
  10 from sdcBePy.common.sdcBeProxy import SdcBeProxy
  11
  12 colors = BColors()
  13
  14
  15 def check_backend(sdc_be_proxy=None, reply_append_count=1, be_host=None, be_port=None, header=None, scheme=None, debug=False):
  16     if sdc_be_proxy is None:
  17         sdc_be_proxy = SdcBeProxy(be_host, be_port, header, scheme, debug=debug)
  18
  19     for i in range(1, reply_append_count + 1):
  20         if sdc_be_proxy.check_backend() == 200:
  21             print('[INFO]: Backend is up and running')
  22             return True
  23         else:
  24             print('[WARRING]: ' + datetime.now().strftime('%Y/%m/%d %H:%M:%S') + colors.FAIL
  25                   + ' Backend not responding, try #' + str(i) + colors.END_C)
  26             time.sleep(properties.retry_time)
  27
  28     return False
  29
  30
  31 def run(be_host, be_port, header, protocol):
  32     if not check_backend(reply_append_count=properties.retry_attempts, be_host=be_host,
  33                          be_port=be_port, header=header, scheme=protocol):
  34         print('[ERROR]: ' + time.strftime('%Y/%m/%d %H:%M:%S') + colors.FAIL + ' Backend is DOWN :-(' + colors.END_C)
  35         sys.exit()
  36
  37
  38 def get_args():
  39     parser = ArgumentParser()
  40
  41     parser.add_argument('-i', '--ip', required=True)
  42     parser.add_argument('-p', '--port', required=True)
  43     parser.add_argument('--header')
  44     parser.add_argument('--https', action='store_true')
  45
  46     args = parser.parse_args()
  47
  48     init_properties(10, 10)
  49     return [args.ip, args.port, args.header, 'https' if args.https else 'http']
  50
  51
  52 def main():
  53     run(*get_args())
  54
  55
  56 if __name__ == '__main__':
  57     main()