from pickle import STOP
import subprocess
from re import sub
import sys
import getopt
sys.path.append(sys.path[0] + "/../")
from gspylib.common.GaussLog import GaussLog
from gspylib.common.ErrorCode import ErrorCode
from gspylib.common.LocalBaseOM import LocalBaseOM
from gspylib.common.ParameterParsecheck import Parameter
from domain_utils.cluster_file.cluster_log import ClusterLog
from domain_utils.domain_common.cluster_constants import ClusterConstants
from base_utils.os.env_util import EnvUtil
from gspylib.component.DSS.dss_checker import DssConfig
from base_utils.os.crontab_util import CrontabUtil
from domain_utils.cluster_file.cluster_dir import ClusterDir
from gspylib.common.Common import DefaultValue
class CMOptConst:
START = "start"
STOP = "stop"
class LocalCmOpt(LocalBaseOM):
"""
The class is used to do perform start
"""
def __init__(self):
"""
function: initialize the parameters
input: NA
output: NA
"""
super(LocalCmOpt, self).__init__()
self.user = ""
self.dataDir = ""
self.time_out = 300
self.logFile = ""
self.logger = None
self.installPath = ""
self.security_mode = ""
self.cluster_number = None
self.action = ""
def usage(self):
"""
General options:
-U USER the database program and cluster owner")
-D DATADIR data directory of instance
-t SECS seconds to wait
-l LOGFILE log file
-?, --help show this help, then exit
--action start or stop
"""
print(self.usage.__doc__)
def parseCommandLine(self):
"""
function: Check input parameters
input : NA
output: NA
"""
try:
opts, args = getopt.getopt(sys.argv[1:], "U:D:R:l:h?",
["help", "action="])
except getopt.GetoptError as e:
GaussLog.exitWithError(ErrorCode.GAUSS_500["GAUSS_50000"] % str(e))
if (len(args) > 0):
GaussLog.exitWithError(
ErrorCode.GAUSS_500["GAUSS_50000"] % str(args[0]))
for key, value in opts:
if key == "-U":
self.user = value
elif key == "-D":
self.dataDir = value
elif key == "-t":
self.time_out = int(value)
elif key == "-l":
self.logFile = value
elif key == "-R":
self.installPath = value
elif key == "--action":
self.action = value
elif key == "--help" or key == "-h" or key == "-?":
self.usage()
sys.exit(0)
else:
GaussLog.exitWithError(ErrorCode.GAUSS_500["GAUSS_50000"]
% key)
Parameter.checkParaVaild(key, value)
if self.user == "":
GaussLog.exitWithError(ErrorCode.GAUSS_500["GAUSS_50001"]
% 'U' + ".")
if self.logFile == "":
self.logFile = ClusterLog.getOMLogPath(
ClusterConstants.LOCAL_LOG_FILE, self.user, self.installPath)
def __initLogger(self):
"""
function: Init logger
input : NA
output: NA
"""
self.logger = GaussLog(self.logFile, "LocalCMOperation")
def init(self):
"""
function: constructor
"""
self.__initLogger()
self.readConfigInfo()
def do_stop_components(self):
cmd = ""
if CrontabUtil.check_user_crontab_permission():
crondesc = subprocess.getoutput("crontab -l | grep om_monitor")
if not crondesc.startswith("#"):
cmd += "crontab -l | sed '/om_monitor/s/^/#/' | crontab -;"
cmd += f"pkill -9 om_monitor -U {self.user};"
cmd += f"pkill -9 cm_agent -U {self.user};"
cmd += f"pkill -9 cm_server -U {self.user};"
self.logger.log(f"stop cm components: {cmd}")
status, output = subprocess.getstatusoutput(cmd)
self.logger.log(status, output)
def do_start_components(self):
mpprc_file = EnvUtil.getEnv(DefaultValue.MPPRC_FILE_ENV)
app_path = ClusterDir.getInstallDir(self.user)
log_path = ClusterLog.getOMLogPath(DefaultValue.OM_MONITOR_DIR_FILE,
self.user,
app_path)
cmd = ""
if CrontabUtil.check_user_crontab_permission():
crondesc = subprocess.getoutput("crontab -l | grep om_monitor")
if crondesc.startswith("#"):
cmd = "crontab -l | sed '/om_monitor/s/^#//' | crontab -;"
if mpprc_file != "" and mpprc_file is not None:
cmd += "source ~/.bashrc;source %s; nohup %s/bin/om_monitor -L %s " \
">>/dev/null 2>&1 &" % (mpprc_file, app_path, log_path)
else:
cmd += "source ~/.bashrc; nohup %s/bin/om_monitor -L %s >>" \
"/dev/null 2>&1 &" % (app_path, log_path)
self.logger.log(f"start cm components: {cmd}")
status, output = subprocess.getstatusoutput(cmd)
self.logger.log(status, output)
def do_operate(self):
"""
function: do start database
input : NA
output : NA
"""
print(self.action)
if self.action == CMOptConst.START:
self.do_start_components()
elif self.action == CMOptConst.STOP:
self.do_stop_components()
else:
self.logger.warn(f"action [{self.action}] is unknown, Do nothing.")
def main():
"""
main function
"""
try:
opt = LocalCmOpt()
opt.parseCommandLine()
opt.init()
except Exception as e:
GaussLog.exitWithError(ErrorCode.GAUSS_536["GAUSS_53608"] % str(e))
try:
opt.do_operate()
except Exception as e:
GaussLog.exitWithError(str(e))
if __name__ == "__main__":
main()