快捷導(dǎo)航

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

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

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

前言

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

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

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

一、Eventlet

該庫可以直接下載：

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秒，超過了2秒，這樣會(huì)導(dǎo)致print('1')這條語句執(zhí)行跳過，直接輸出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í)，并沒有超時(shí)，此時(shí)應(yīng)該輸出1、2：

當(dāng)我們想要調(diào)用子程序該函數(shù)是不起作用的。像這樣，該函數(shù)沒有效果，這時(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í)間超過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)