Python控制线程和函数超时处理

2024-04-02 19:55

短信预约 -IT技能 免费直播动态提醒

前言

在我们使用线程池控制线程运行时，或者是写爬虫时不停发送request获取地址，都需要我们对运行的线程加以控制。有这样一种场景，当你写入一条网站请求时，该线程一直请求并没有回应导致线程造成堵塞，浪费珍贵的线程资源。当你写入一条错误的sql查询语句时，该查询语句一直运行需要花费大量时间，导致其他查询语句阻塞。遇到如上述情况必须对线程加以控制，才能使得程序脚本稳定性更高。

控制线程运行时间方法有多种：

这里介绍两种方法，一种是用eventlet库的方法进行，该方法只要进行了子程序调用就失效了。另一种方法是使用@func_set_timeout修饰器，该方法可作用于函数、线程等一系列子函数，是最方便最简单高效的方法。这里来进行代码操作演示这两种方法。

一、Eventlet

该库可以直接下载：

pip install eventlet
import time
import eventlet#导入eventlet
eventlet.monkey_patch()#引入patch
with eventlet.Timeout(2,False):#设置超时时间为2秒
    time.sleep(3)
    print('1')
print('2')

我们设定睡眠时间为3秒，超过了2秒，这样会导致print('1')这条语句执行跳过，直接输出2：

import time
import eventlet#导入eventlet
eventlet.monkey_patch()#引入patch
with eventlet.Timeout(2,False):#设置超时时间为2秒
    time.sleep(1)
    print('1')
print('2')

当我们修改time.sleep为1小于2时，并没有超时，此时应该输出1、2：

当我们想要调用子程序该函数是不起作用的。像这样，该函数没有效果，这时候我们需要使用第二种方法。

二、func-timeout

1.安装

直接安装就好了

pip install func-timeout

2.使用

一般是和try except一起用，超时则扔出错误。

from func_timeout import func_set_timeout
import func_timeout
@func_set_timeout(1)#设定函数超时执行时间
def task(i):
    time.sleep(2)
    print(i)
try:
    task(1)
except:
    print(2)

这时候sleep时间超过1应该输出2：

from func_timeout import func_set_timeout
import func_timeout
@func_set_timeout(3)#设定函数超时执行时间
def task(i):
    time.sleep(2)
    print(i)
try:
    task(1)
except:
    print(2)