소스 검색

AMBARI-16917: Use postgres database instead of template1 during service check in HAWQ (bhuvnesh2703)

Bhuvnesh Chaudhary 9 년 전
부모
커밋
8721b56f8d

+ 2 - 1
ambari-server/src/main/resources/common-services/HAWQ/2.0.0/package/alerts/alert_segment_registration_status.py

@@ -28,6 +28,7 @@ HAWQ_HOME='/usr/local/hawq'
 HAWQ_GREENPLUM_PATH_FILE = "{0}/greenplum_path.sh".format(HAWQ_HOME)
 HAWQ_SLAVES_FILE= "{0}/etc/slaves".format(HAWQ_HOME)
 HAWQMASTER_PORT = '{{hawq-site/hawq_master_address_port}}'
+POSTGRES = 'postgres'
 
 RESULT_STATE_OK = 'OK'
 RESULT_STATE_WARNING = 'WARNING'
@@ -94,7 +95,7 @@ def get_segment_list_db(port):
   """
   logger.debug("Fetching segment list from HAWQ Master Database.")
   query = " SELECT hostname FROM gp_segment_configuration where role = 'p' and status = 'u' "
-  cmd = "source {0} && psql -p {1} -t -d template1 -c \"{2};\"".format(HAWQ_GREENPLUM_PATH_FILE, port, query)
+  cmd = "source {0} && psql -p {1} -t -d {2} -c \"{3};\"".format(HAWQ_GREENPLUM_PATH_FILE, port, POSTGRES, query)
  
   returncode, command_output = call(cmd, user=HAWQ_USER, timeout=60)
   if returncode:

+ 2 - 1
ambari-server/src/main/resources/common-services/HAWQ/2.0.0/package/alerts/alert_sync_status.py

@@ -23,6 +23,7 @@ from resource_management.core.shell import call
 
 HAWQ_USER = 'gpadmin'
 HAWQ_GREENPLUM_PATH_FILE = '/usr/local/hawq/greenplum_path.sh'
+POSTGRES = 'postgres'
 
 HAWQMASTER_PORT = '{{hawq-site/hawq_master_address_port}}'
 HAWQSTANDBY_ADDRESS = '{{hawq-site/hawq_standby_address_host}}'
@@ -81,7 +82,7 @@ def get_sync_status(port):
   """
   
   query = "SELECT summary_state, error_message FROM gp_master_mirroring"
-  cmd = "source {0} && psql -p {1} -t --no-align -d template1 -c \"{2};\"".format(HAWQ_GREENPLUM_PATH_FILE, port, query)
+  cmd = "source {0} && psql -p {1} -t --no-align -d {2} -c \"{3};\"".format(HAWQ_GREENPLUM_PATH_FILE, port, POSTGRES, query)
 
   returncode, output = call(cmd, user=HAWQ_USER, timeout=60)
 

+ 1 - 0
ambari-server/src/main/resources/common-services/HAWQ/2.0.0/package/scripts/hawq_constants.py

@@ -28,6 +28,7 @@ CLUSTER = "cluster"
 IMMEDIATE = "immediate"
 FAST = "fast"
 ACTIVATE = "activate"
+POSTGRES = "postgres"
 
 # Users
 root_user = "root"

+ 1 - 1
ambari-server/src/main/resources/common-services/HAWQ/2.0.0/package/scripts/utils.py

@@ -87,7 +87,7 @@ def exec_ssh_cmd(hostname, cmd):
   return process.returncode, stdout, stderr
 
 
-def exec_psql_cmd(command, host, port, db="template1", tuples_only=True):
+def exec_psql_cmd(command, host, port, db=hawq_constants.POSTGRES, tuples_only=True):
   """
   Sets up execution environment and runs the HAWQ queries
   """