溫馨提示×

溫馨提示×

您好，登錄后才能下訂單哦！

密碼登錄×
登錄注冊×
其他方式登錄
點擊 登錄注冊 即表示同意《億速云用戶服務條款》

詳解python中自定義超時異常的幾種方法

發(fā)布時間：2020-09-07 00:18:59 來源：腳本之家 閱讀：789 作者：payneLi 欄目：開發(fā)技術(shù)

最近在項目中調(diào)用第三方接口時候，經(jīng)常會出現(xiàn)請求超時的情況，或者參數(shù)的問題導致調(diào)用異代碼異常。針對超時異常，查詢了python 相關(guān)文檔，沒有并發(fā)現(xiàn)完善的包來根據(jù)用戶自定義的時間來拋出超時異常的模塊。所以自己干脆自己來實現(xiàn)一個自定義的超時異常。目前找到了兩種方式來實現(xiàn)超時異常的功能（signal.alarm()、threading實現(xiàn)超時異常）

方法1 thread + time　

原理：將要調(diào)用的功能函數(shù)放入子線程，通過設定子線程的阻塞時間，超時則主線程并不會等待子線程的執(zhí)行。主線程退出，子線程就不存在了。

核心就是在程序中添加 join()方法，用于等待線程結(jié)束。join()的作用是，在子線程完成運行之前，這個子線程的父線程將會被一直阻塞.

# coding=utf-8
import threading
import time


def myFunc():
  time.sleep(4)
  print("myFunc執(zhí)行了")


if __name__ == '__main__':
  t = threading.Thread(target=myFunc)
  t.setDaemon(True)
  t.start()

  t.join(2)
  print("it's over")

執(zhí)行結(jié)果：

it's over

可以看出，當主線程執(zhí)行到2秒時候，結(jié)束退出。子線程還沒有結(jié)束，沒有執(zhí)行完及被強制退出

# coding=utf-8
import threading
import time


def myFunc():
  time.sleep(1)
  print("myFunc執(zhí)行了")


if __name__ == '__main__':
  t = threading.Thread(target=myFunc)
  t.setDaemon(True)
  t.start()

  t.join(2)
  print("it's over")

顯示結(jié)果：

myFunc執(zhí)行了
it's over

可以看出，子線程結(jié)束時，用時1秒，沒有超過主線程設定的3秒，所以主線程與子線程都被執(zhí)行了

方法 2  signal.alarm() ，注意兩點：一是signal信號機制要在linux上才能運行； 二是signal信號在主線程中才會會起作用

import signal
import time

# Define signal handler function
def myHandler(signum, frame):
  exit("TimeoutError")

def test_fun():
  # time.sleep(3)
  int("afsdf")
  a = 2 + 3
  return a

if __name__ == '__main__':
  try:
    signal.signal(signal.SIGALRM, myHandler)
    signal.alarm(2)
    test = test_fun()
    print(test)
    signal.alarm(0)
  except Exception as ret:
    print("msg:", ret)

執(zhí)行結(jié)果：

當 time.sleep(3) 時，會拋出TimeoutError的異常
當 test_fun 里面出現(xiàn) int("afsdf")時， 會拋出 ValueError("invalid literal for int()         with base 10: 'afsdf'",))
當test_fun函數(shù)執(zhí)行的時間小于2 秒時，就會返回函數(shù)對應的值

方法3  帶有返回值的超時異常，可以通過創(chuàng)建thread類的方式來進行捕捉

import threading
import sys
import time


class Dispacher(threading.Thread):
  def __init__(self, fun, args):
    threading.Thread.__init__(self)
    self.setDaemon(True)
    self.result = None
    self.error = None
    self.fun = fun
    self.args = args

    self.start()

  def run(self):
    try:
      self.result = self.fun(self.args)
    except:
      self.error = sys.exc_info()


def test_fun(i):
  # time.sleep(4)
  a = i*i
  # b 　　　
　　return a
def main_fun():
  c = Dispacher(test_fun, 2)
  c.join(2)

  if c.isAlive():
    return "TimeOutError"
  elif c.error:
    return c.error[1]
  t = c.result
  return t

if __name__ == '__main__':
  fun = main_fun()
  print(fun)

顯示結(jié)果：

test_fun 執(zhí)行時間大于設置的2秒時，會拋出TimeOutError
test_fun 執(zhí)行時間小于設置的2秒時，并且函數(shù)正常執(zhí)行時，顯示：4
test_fun 里面出現(xiàn)比如 “b” 時，會拋出 global name 'b' is not defined 的異常

以上就是本文的全部內(nèi)容，希望對大家的學習有所幫助，也希望大家多多支持億速云。

向AI問一下細節(jié)

免責聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點不代表本網(wǎng)站立場，如果涉及侵權(quán)請聯(lián)系站長郵箱：is@yisu.com進行舉報，并提供相關(guān)證據(jù)，一經(jīng)查實，將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI
助
手