##// END OF EJS Templates
pull-request: verify resolve TODO comment needs to be bound to the same PR as we're calling
pull-request: verify resolve TODO comment needs to be bound to the same PR as we're calling

File last commit:

r2351:59272121 default
r2441:9a41d4d3 default
Show More
process_management.py
91 lines | 3.0 KiB | text/x-python | PythonLexer
# -*- coding: utf-8 -*-
# Copyright (C) 2016-2017 RhodeCode GmbH
#
# 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
from pyramid.view import view_config
from rhodecode.apps._base import BaseAppView
from rhodecode.apps.admin.navigation import navigation_list
from rhodecode.lib.auth import (
LoginRequired, HasPermissionAllDecorator, CSRFRequired)
from rhodecode.lib.utils2 import safe_int
log = logging.getLogger(__name__)
class AdminProcessManagementView(BaseAppView):
def load_default_context(self):
c = self._get_local_tmpl_context()
return c
@LoginRequired()
@HasPermissionAllDecorator('hg.admin')
@view_config(
route_name='admin_settings_process_management', request_method='GET',
renderer='rhodecode:templates/admin/settings/settings.mako')
def process_management(self):
_ = self.request.translate
c = self.load_default_context()
c.active = 'process_management'
c.navlist = navigation_list(self.request)
c.gunicorn_processes = (
p for p in psutil.process_iter() if 'gunicorn' in p.name())
return self._get_template_context(c)
@LoginRequired()
@HasPermissionAllDecorator('hg.admin')
@CSRFRequired()
@view_config(
route_name='admin_settings_process_management_signal',
request_method='POST', renderer='json_ext')
def process_management_signal(self):
pids = self.request.json.get('pids', [])
result = []
def on_terminate(proc):
msg = "process `PID:{}` terminated with exit code {}".format(
proc.pid, proc.returncode)
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:
print('Wont kill Master Process')
else:
procs.append(proc)
for p in procs:
p.terminate()
gone, alive = psutil.wait_procs(procs, timeout=10, callback=on_terminate)
for p in alive:
p.kill()
return {'result': result}