63 lines
1.6 KiB
Python
63 lines
1.6 KiB
Python
"""
|
|
Copyright (c) 2022-2023 Wind River Systems, Inc.
|
|
|
|
SPDX-License-Identifier: Apache-2.0
|
|
|
|
"""
|
|
|
|
import datetime
|
|
import fnmatch
|
|
import os
|
|
import resource
|
|
|
|
from django.conf import settings
|
|
|
|
|
|
errorlog = "/var/log/horizon/gunicorn.log"
|
|
capture_output = True
|
|
|
|
# maxrss ceiling in kbytes
|
|
MAXRSS_CEILING = 512000
|
|
|
|
|
|
def worker_abort(worker):
|
|
path = ("/proc/%s/fd") % os.getpid()
|
|
contents = os.listdir(path)
|
|
upload_dir = getattr(settings, 'FILE_UPLOAD_TEMP_DIR', '/tmp')
|
|
pattern = os.path.join(upload_dir, '*.upload')
|
|
|
|
for i in contents:
|
|
f = os.path.join(path, i)
|
|
if os.path.exists(f):
|
|
try:
|
|
link = os.readlink(f)
|
|
if fnmatch.fnmatch(link, pattern):
|
|
worker.log.info(link)
|
|
os.remove(link)
|
|
except OSError:
|
|
pass
|
|
|
|
|
|
def post_worker_init(worker):
|
|
worker.nrq = 0
|
|
worker.restart = False
|
|
|
|
|
|
def pre_request(worker, req):
|
|
worker.nrq += 1
|
|
if worker.restart:
|
|
worker.nr = worker.max_requests - 1
|
|
maxrss = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss
|
|
msg = "%(date)s %(uri)s %(rss)u" % ({'date': datetime.datetime.now(),
|
|
'uri': getattr(req, "uri"),
|
|
'rss': maxrss})
|
|
worker.log.info(msg)
|
|
|
|
|
|
def post_request(worker, req, environ, resp):
|
|
worker.nrq -= 1
|
|
if not worker.restart:
|
|
maxrss = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss
|
|
if maxrss > MAXRSS_CEILING and worker.nrq == 0:
|
|
worker.restart = True
|