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

方法1 thread + time　

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

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

            
# coding=utf-8
import threading
import time


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


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

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

執行結果：

it's over

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

            
# coding=utf-8
import threading
import time


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


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

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


          

顯示結果：

myFunc執行了
it's over

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

方法 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)
          

執行結果：

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

方法3? 帶有返回值的超時異常，可以通過創建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)
          

顯示結果：

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

以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。