mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
128 lines
3.8 KiB
128 lines
3.8 KiB
#!/usr/bin/env python3
|
|
# coding: utf-8
|
|
|
|
import argparse
|
|
import shutil
|
|
import subprocess
|
|
import os
|
|
import signal
|
|
import tempfile
|
|
|
|
ANSIBLE_RUNNER_COMMAND = "ansible-runner"
|
|
DEFAULT_SHARE_DIR = "data/share"
|
|
DEFAULT_CONTROL_SOCK_PATH = os.path.join(DEFAULT_SHARE_DIR, "control.sock")
|
|
|
|
|
|
class ReceptorService:
|
|
def __init__(self):
|
|
self.pid_file = "tmp/receptor.pid"
|
|
self.receptor_command = [
|
|
'receptor',
|
|
'--local-only',
|
|
'--node', 'id=primary',
|
|
'--control-service',
|
|
'service=control',
|
|
'filename={}'.format(DEFAULT_CONTROL_SOCK_PATH),
|
|
'--work-command',
|
|
'worktype={}'.format(ANSIBLE_RUNNER_COMMAND),
|
|
'command={}'.format(ANSIBLE_RUNNER_COMMAND),
|
|
'params=worker',
|
|
'allowruntimeparams=true'
|
|
]
|
|
|
|
@staticmethod
|
|
def before_start():
|
|
os.makedirs(os.path.join(DEFAULT_SHARE_DIR), exist_ok=True)
|
|
status_dir = os.path.join(tempfile.gettempdir(), "receptor")
|
|
if os.path.exists(status_dir):
|
|
shutil.rmtree(status_dir)
|
|
|
|
def start(self):
|
|
self.before_start()
|
|
if os.path.exists(self.pid_file):
|
|
with open(self.pid_file, 'r') as f:
|
|
pid_str = f.read()
|
|
try:
|
|
pid = int(pid_str)
|
|
os.kill(pid, 0)
|
|
print("\n- Receptor service is already running.")
|
|
return
|
|
except ProcessLookupError:
|
|
print("\n- PID file exists but process does not, starting Receptor...")
|
|
except ValueError:
|
|
print("\n- PID file is corrupted, starting Receptor...")
|
|
os.remove(self.pid_file)
|
|
|
|
process = subprocess.Popen(self.receptor_command)
|
|
with open(self.pid_file, 'w') as f:
|
|
f.write(str(process.pid))
|
|
print("\n- Receptor service started successfully.")
|
|
|
|
def exit_handler(signum, frame):
|
|
process.terminate()
|
|
process.kill()
|
|
|
|
signal.signal(signal.SIGINT, exit_handler)
|
|
signal.signal(signal.SIGTERM, exit_handler)
|
|
process.wait()
|
|
|
|
def stop(self):
|
|
if not os.path.exists(self.pid_file):
|
|
print("\n- Receptor service is not running.")
|
|
return
|
|
with open(self.pid_file, 'r') as f:
|
|
pid = int(f.read())
|
|
try:
|
|
os.kill(pid, signal.SIGTERM)
|
|
os.remove(self.pid_file)
|
|
print("\n- Receptor service stopped successfully.")
|
|
except ProcessLookupError:
|
|
print("\n- Failed to stop Receptor service: Process does not exist.")
|
|
os.remove(self.pid_file)
|
|
|
|
def restart(self):
|
|
self.stop()
|
|
self.start()
|
|
|
|
def status(self):
|
|
if os.path.exists(self.pid_file):
|
|
with open(self.pid_file, 'r') as f:
|
|
pid_str = f.read()
|
|
try:
|
|
pid = int(pid_str)
|
|
os.kill(pid, 0)
|
|
print("\n- Receptor service is running.")
|
|
return
|
|
except ProcessLookupError:
|
|
print("\n- Receptor service is not running.")
|
|
else:
|
|
print("\n- Receptor service is not running.")
|
|
|
|
|
|
def handle_receptor_action(action):
|
|
srv = ReceptorService()
|
|
if action == "start":
|
|
srv.start()
|
|
elif action == 'stop':
|
|
srv.stop()
|
|
elif action == "restart":
|
|
srv.restart()
|
|
elif action == "status":
|
|
srv.status()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(
|
|
description="""
|
|
Jumpserver receptor service control tools;
|
|
"""
|
|
)
|
|
parser.add_argument(
|
|
'action', type=str,
|
|
choices=("start", "stop", "restart", "status"),
|
|
help="Action to run"
|
|
)
|
|
|
|
args = parser.parse_args()
|
|
handle_receptor_action(args.action)
|