當你在同一個線程中調用一個函數時,它通常不會返回,直到完成。你打電話的功能必須首先被設計爲可中斷的。有很多方法可以達到這個目的,但具有不同程度的複雜性和通用性。
也許最簡單的方法是將時間限制傳遞給你的函數,並以小塊處理工作。處理完每個塊後,檢查經過的時間是否超過超時時間,如果超過,請提早提前退出。
下面的例子說明了這個想法,對工作採取每塊的一個隨機時間,其有時會完全有時超時:
import time
import random
import datetime
class TimeoutException(Exception):
def __init__(self, *args, **kwargs):
Exception.__init__(self, *args, **kwargs)
def busy_work():
# Pretend to do something useful
time.sleep(random.uniform(0.3, 0.6))
def train_loadbatch_from_lists(batch_size, timeout_sec):
time_start = datetime.datetime.now()
batch_xs = []
batch_ys = []
for i in range(0, batch_size+1):
busy_work()
batch_xs.append(i)
batch_ys.append(i)
time_elapsed = datetime.datetime.now() - time_start
print 'Elapsed:', time_elapsed
if time_elapsed > timeout_sec:
raise TimeoutException()
return batch_xs, batch_ys
def main():
timeout_sec = datetime.timedelta(seconds=5)
batch_size = 10
try:
print 'Processing batch'
batch_xs, batch_ys = train_loadbatch_from_lists(batch_size, timeout_sec)
print 'Completed successfully'
print batch_xs, batch_ys
except TimeoutException, e:
print 'Timeout after processing N records'
if __name__ == '__main__':
main()
另一種方式來實現,這是運行輔助功能一個單獨的線程,並且使用一個Event
來允許調用者發信號通知工作者函數應該儘早終止。
有些帖子(如上面鏈接的帖子)建議使用信號,但不幸的是信號可能會導致更多的複雜性,所以不推薦。
這裏有一些想法:http://stackoverflow.com/questions/2281850/timeout-function-if-it-takes-too-long-to-finish – Miikka
是的,我經歷了它。但我不確定我知道如何使用它。 所以,如果我有一個函數定義之前定義的@timeout(15),並且如果發生錯誤,我想再次調用它。你知道如何做到這一點嗎? –
我真的不會推薦使用信號。他們看起來像一個體面的解決方案,但有各種奇怪的副作用,角落案件等,並可能導致更多的問題比他們解決。 – gavinb