Python控制線程和函數(shù)超時(shí)處理

更新時(shí)間：2022年06月02日 11:56:58 作者：fanstuck

這篇文章主要介紹了Python控制線程和函數(shù)超時(shí)處理，文中利用兩種方法進(jìn)行eventlet庫(kù)和@func_set_timeout修飾器，文章具體詳細(xì)介紹，需要的小伙伴可以參考一下

前言

在我們使用線程池控制線程運(yùn)行時(shí)，或者是寫爬蟲(chóng)時(shí)不停發(fā)送request獲取地址，都需要我們對(duì)運(yùn)行的線程加以控制。有這樣一種場(chǎng)景，當(dāng)你寫入一條網(wǎng)站請(qǐng)求時(shí)，該線程一直請(qǐng)求并沒(méi)有回應(yīng)導(dǎo)致線程造成堵塞，浪費(fèi)珍貴的線程資源。當(dāng)你寫入一條錯(cuò)誤的sql查詢語(yǔ)句時(shí)，該查詢語(yǔ)句一直運(yùn)行需要花費(fèi)大量時(shí)間，導(dǎo)致其他查詢語(yǔ)句阻塞。遇到如上述情況必須對(duì)線程加以控制，才能使得程序腳本穩(wěn)定性更高。

控制線程運(yùn)行時(shí)間方法有多種：

這里介紹兩種方法，一種是用eventlet庫(kù)的方法進(jìn)行，該方法只要進(jìn)行了子程序調(diào)用就失效了。另一種方法是使用@func_set_timeout修飾器，該方法可作用于函數(shù)、線程等一系列子函數(shù)，是最方便最簡(jiǎn)單高效的方法。這里來(lái)進(jìn)行代碼操作演示這兩種方法。

一、Eventlet

該庫(kù)可以直接下載：

pip install eventlet
import time
import eventlet#導(dǎo)入eventlet
eventlet.monkey_patch()#引入patch
with eventlet.Timeout(2,False):#設(shè)置超時(shí)時(shí)間為2秒
? ? time.sleep(3)
? ? print('1')
print('2')

我們?cè)O(shè)定睡眠時(shí)間為3秒，超過(guò)了2秒，這樣會(huì)導(dǎo)致print('1')這條語(yǔ)句執(zhí)行跳過(guò)，直接輸出2：

import time
import eventlet#導(dǎo)入eventlet
eventlet.monkey_patch()#引入patch
with eventlet.Timeout(2,False):#設(shè)置超時(shí)時(shí)間為2秒
? ? time.sleep(1)
? ? print('1')
print('2')

當(dāng)我們修改time.sleep為1小于2時(shí)，并沒(méi)有超時(shí)，此時(shí)應(yīng)該輸出1、2：

當(dāng)我們想要調(diào)用子程序該函數(shù)是不起作用的。像這樣，該函數(shù)沒(méi)有效果，這時(shí)候我們需要使用第二種方法。

二、func-timeout

1.安裝

直接安裝就好了

pip install func-timeout

2.使用

一般是和try except一起用，超時(shí)則扔出錯(cuò)誤。

from func_timeout import func_set_timeout
import func_timeout
@func_set_timeout(1)#設(shè)定函數(shù)超時(shí)執(zhí)行時(shí)間
def task(i):
? ? time.sleep(2)
? ? print(i)
try:
? ? task(1)
except:
? ? print(2)

這時(shí)候sleep時(shí)間超過(guò)1應(yīng)該輸出2：

from func_timeout import func_set_timeout
import func_timeout
@func_set_timeout(3)#設(shè)定函數(shù)超時(shí)執(zhí)行時(shí)間
def task(i):
? ? time.sleep(2)
? ? print(i)
try:
? ? task(1)
except:
? ? print(2)