您现在的位置是：网站首页> 编程资料编程资料

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

2023-05-26 323人已围观

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

前言

在我们使用线程池控制线程运行时，或者是写爬虫时不停发送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)