mirror of
https://github.com/nicolargo/glances.git
synced 2026-01-08 05:48:38 -05:00
da700c8cfc
For use with blocking functions and utilities Co-authored-by: Raz Crimson <52282402+razcrimson@users.noreply.github.com>
49 lines
1.2 KiB
Python
49 lines
1.2 KiB
Python
import time
|
|
from multiprocessing import Process, Queue
|
|
|
|
import psutil
|
|
|
|
|
|
def exit_after(seconds, default=None):
|
|
"""Exit the function if it takes more than 'second' seconds to complete.
|
|
In this case, return the value of 'default' (default: None)."""
|
|
|
|
def handler(q, func, args, kwargs):
|
|
q.put(func(*args, **kwargs))
|
|
|
|
def decorator(func):
|
|
def wraps(*args, **kwargs):
|
|
q = Queue()
|
|
p = Process(target=handler, args=(q, func, args, kwargs))
|
|
p.start()
|
|
p.join(timeout=seconds)
|
|
if not p.is_alive():
|
|
return q.get()
|
|
|
|
p.terminate()
|
|
p.join(timeout=0.1)
|
|
if p.is_alive():
|
|
# Kill in case processes doesn't terminate
|
|
# Happens with cases like broken NFS connections
|
|
p.kill()
|
|
return default
|
|
|
|
return wraps
|
|
|
|
return decorator
|
|
|
|
|
|
class Issue3290:
|
|
@exit_after(1, default=None)
|
|
def blocking_io_call(self, fs):
|
|
try:
|
|
return psutil.disk_usage(fs)
|
|
except OSError:
|
|
return None
|
|
|
|
|
|
issue = Issue3290()
|
|
while True:
|
|
print(f"{time.time()} {issue.blocking_io_call('/home/nicolargo/tmp/hang')}")
|
|
time.sleep(1)
|