超时机制
一般应用于处理阻塞问题
场景:
- 复杂度较大的计算(解析)某个数值、加解密计算等
- 请求中遇到阻塞,避免长时间等待
- 网络波动,避免长时间请求,浪费时间
1. requests 请求超时机制
reqeusts 依赖中的Post请求中自带 timeout 参数,可以直接设置
response = requests.post(url,
data=request_body,
headers=headers,
timeout=timeout)
2. 其他函数时间超时机制
自定义一个超时函数 timeout()
注:这里用到了装饰器,后面文章会详细说明
import signal
from functools import wraps
import errno
import os
class TimeoutError(Exception):
pass
def timeout(seconds=10, error_message=os.strerror(errno.ETIME)):
def decorator(func):
def _handle_timeout(signum, frame):
raise TimeoutError(error_message)
def wrapper(*args, **kwargs):
signal.signal(signal.SIGALRM, _handle_timeout)
signal.alarm(seconds)
try:
result = fun