mirror of
https://github.com/frappe/gunicorn.git
synced 2026-07-01 18:21:30 +08:00
95b7ffeeaa
- Bump version to 25.0.2 - Update copyright year to 2026 in LICENSE and NOTICE - Add license headers to all Python source files - Add changelog entry for 25.0.2
42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
#
|
|
# This file is part of gunicorn released under the MIT license.
|
|
# See the NOTICE for more information.
|
|
|
|
import signal
|
|
import commands
|
|
import threading
|
|
import time
|
|
|
|
max_mem = 100000
|
|
|
|
class MemoryWatch(threading.Thread):
|
|
|
|
def __init__(self, server, max_mem):
|
|
super().__init__()
|
|
self.daemon = True
|
|
self.server = server
|
|
self.max_mem = max_mem
|
|
self.timeout = server.timeout / 2
|
|
|
|
def memory_usage(self, pid):
|
|
try:
|
|
out = commands.getoutput("ps -o rss -p %s" % pid)
|
|
except OSError:
|
|
return -1
|
|
used_mem = sum(int(x) for x in out.split('\n')[1:])
|
|
return used_mem
|
|
|
|
def run(self):
|
|
while True:
|
|
for (pid, worker) in list(self.server.WORKERS.items()):
|
|
if self.memory_usage(pid) > self.max_mem:
|
|
self.server.log.info("Pid %s killed (memory usage > %s)",
|
|
pid, self.max_mem)
|
|
self.server.kill_worker(pid, signal.SIGTERM)
|
|
time.sleep(self.timeout)
|
|
|
|
|
|
def when_ready(server):
|
|
mw = MemoryWatch(server, max_mem)
|
|
mw.start()
|