##// END OF EJS Templates
fix(deprecations): fixed sqlalchemy warning about non-cachable JsonObject
r5223:6c71ac19 default
Show More
process_management.py
168 lines | 5.5 KiB | text/x-python | PythonLexer
/ rhodecode / apps / admin / views / process_management.py
copyrights: updated for 2023
r5088 # Copyright (C) 2016-2023 RhodeCode GmbH
process-managemet: added simple page to monitor worker processes of RhodeCode.
r1885 #
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License, version 3
# (only), as published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
# This program is dual-licensed. If you wish to learn more about the
# RhodeCode Enterprise Edition, including its added features, Support services,
# and proprietary license terms, please see https://rhodecode.com/licenses/
import logging
import psutil
processes: add a way to control processes via gunicorn control....
r2503 import signal
application: not use config.scan(), and replace all @add_view decorator into a explicit add_view call for faster app start.
r4610
process-managemet: added simple page to monitor worker processes of RhodeCode.
r1885
from rhodecode.apps._base import BaseAppView
application: re-organize imports for pyramid to prepare code for speedup optimization.
r3238 from rhodecode.apps._base.navigation import navigation_list
processes: show how many workers we have set from configuration.
r3155 from rhodecode.lib import system_info
process-managemet: added simple page to monitor worker processes of RhodeCode.
r1885 from rhodecode.lib.auth import (
LoginRequired, HasPermissionAllDecorator, CSRFRequired)
process-management: use safer way to extract data for processes....
r2540 from rhodecode.lib.utils2 import safe_int, StrictAttributeDict
process-managemet: added simple page to monitor worker processes of RhodeCode.
r1885
log = logging.getLogger(__name__)
class AdminProcessManagementView(BaseAppView):
def load_default_context(self):
c = self._get_local_tmpl_context()
process-management: use safer way to extract data for processes....
r2540 return c
pylons: remove pylons as dependency...
r2351
process-management: use safer way to extract data for processes....
r2540 def _format_proc(self, proc, with_children=False):
try:
mem = proc.memory_info()
proc_formatted = StrictAttributeDict({
'pid': proc.pid,
'name': proc.name(),
'mem_rss': mem.rss,
'mem_vms': mem.vms,
processes: reduce measure interval to 0.1 to make the processes page load a bit faster (else blocking a lot)
r3915 'cpu_percent': proc.cpu_percent(interval=0.1),
process-management: use safer way to extract data for processes....
r2540 'create_time': proc.create_time(),
'cmd': ' '.join(proc.cmdline()),
})
if with_children:
proc_formatted.update({
'children': [self._format_proc(x)
for x in proc.children(recursive=True)]
})
except Exception:
log.exception('Failed to load proc')
proc_formatted = None
return proc_formatted
def get_processes(self):
proc_list = []
for p in psutil.process_iter():
if 'gunicorn' in p.name():
proc = self._format_proc(p, with_children=True)
if proc:
proc_list.append(proc)
return proc_list
process-managemet: added simple page to monitor worker processes of RhodeCode.
r1885
processes: show how many workers we have set from configuration.
r3155 def get_workers(self):
workers = None
try:
rc_config = system_info.rhodecode_config().value['config']
workers = rc_config['server:main'].get('workers')
except Exception:
pass
return workers or '?'
process-managemet: added simple page to monitor worker processes of RhodeCode.
r1885 @LoginRequired()
@HasPermissionAllDecorator('hg.admin')
def process_management(self):
_ = self.request.translate
c = self.load_default_context()
c.active = 'process_management'
c.navlist = navigation_list(self.request)
process-management: use safer way to extract data for processes....
r2540 c.gunicorn_processes = self.get_processes()
processes: show how many workers we have set from configuration.
r3155 c.gunicorn_workers = self.get_workers()
process-managemet: added simple page to monitor worker processes of RhodeCode.
r1885 return self._get_template_context(c)
@LoginRequired()
@HasPermissionAllDecorator('hg.admin')
process-management: enable auto-refresh to track usage live.
r2500 def process_management_data(self):
_ = self.request.translate
c = self.load_default_context()
process-management: use safer way to extract data for processes....
r2540 c.gunicorn_processes = self.get_processes()
process-management: enable auto-refresh to track usage live.
r2500 return self._get_template_context(c)
@LoginRequired()
@HasPermissionAllDecorator('hg.admin')
process-managemet: added simple page to monitor worker processes of RhodeCode.
r1885 @CSRFRequired()
def process_management_signal(self):
pids = self.request.json.get('pids', [])
result = []
processes: better handling of PID that are not part of RhodeCode processes....
r2661
process-managemet: added simple page to monitor worker processes of RhodeCode.
r1885 def on_terminate(proc):
processes: shorten the message and remove exit code which isn't shown properly anyway.
r3492 msg = "terminated"
process-managemet: added simple page to monitor worker processes of RhodeCode.
r1885 result.append(msg)
procs = []
for pid in pids:
pid = safe_int(pid)
if pid:
try:
proc = psutil.Process(pid)
except psutil.NoSuchProcess:
continue
children = proc.children(recursive=True)
if children:
processes: better handling of PID that are not part of RhodeCode processes....
r2661 log.warning('Wont kill Master Process')
process-managemet: added simple page to monitor worker processes of RhodeCode.
r1885 else:
procs.append(proc)
for p in procs:
processes: better handling of PID that are not part of RhodeCode processes....
r2661 try:
p.terminate()
except psutil.AccessDenied as e:
apps: modernize for python3
r5093 log.warning(f'Access denied: {e}')
processes: better handling of PID that are not part of RhodeCode processes....
r2661
process-managemet: added simple page to monitor worker processes of RhodeCode.
r1885 gone, alive = psutil.wait_procs(procs, timeout=10, callback=on_terminate)
for p in alive:
processes: better handling of PID that are not part of RhodeCode processes....
r2661 try:
p.kill()
except psutil.AccessDenied as e:
apps: modernize for python3
r5093 log.warning(f'Access denied: {e}')
process-managemet: added simple page to monitor worker processes of RhodeCode.
r1885
return {'result': result}
processes: add a way to control processes via gunicorn control....
r2503
@LoginRequired()
@HasPermissionAllDecorator('hg.admin')
@CSRFRequired()
def process_management_master_signal(self):
pid_data = self.request.json.get('pid_data', {})
pid = safe_int(pid_data['pid'])
action = pid_data['action']
if pid:
try:
proc = psutil.Process(pid)
except psutil.NoSuchProcess:
return {'result': 'failure_no_such_process'}
children = proc.children(recursive=True)
if children:
# master process
if action == '+' and len(children) <= 20:
proc.send_signal(signal.SIGTTIN)
elif action == '-' and len(children) >= 2:
proc.send_signal(signal.SIGTTOU)
else:
return {'result': 'failure_wrong_action'}
return {'result': 'success'}
return {'result': 'failure_not_master'}