python使用threading獲取線程函數返回值的實現方法

這篇文章主要介紹了python使用threading獲取線程函數返回值的實現方法,需要的朋友可以參考下

threading用于提供線程相關的操作，線程是應用程序中工作的最小單元。python當前版本的多線程庫沒有實現優先級、線程組，線程也不能被停止、暫停、恢復、中斷。

threading模塊提供的類：

　　Thread, Lock, Rlock, Condition, [Bounded]Semaphore, Event, Timer, local。

threading 模塊提供的常用方法：

　　threading.currentThread(): 返回當前的線程變量。

　　threading.enumerate(): 返回一個包含正在運行的線程的list。正在運行指線程啟動后、結束前，不包括啟動前和終止后的線程。

　　threading.activeCount(): 返回正在運行的線程數量，與len(threading.enumerate())有相同的結果。

threading 模塊提供的常量：

　　threading.TIMEOUT_MAX 設置threading全局超時時間。
好了，正文開始：

最近需要用python寫一個環境搭建工具，多線程并行對環境各個部分執行一些操作，并最終知道這些并行執行的操作是否都執行成功了，也就是判斷這些操作函數的返回值是否為0。但是threading并沒有顯式的提供獲取各個線程函數返回值的方法，只好自己動手，下面就介紹一下自己的實現方式。

一開始考慮到執行的操作可能有很多，而且后續會不斷補充，因此先寫了一個通用的多線程執行類，封裝線程操作的基本方法，如下：    
import threading
class MyThread(object):
  def __init__(self, func_list=None):
  #所有線程函數的返回值匯總，如果最后為0，說明全部成功
    self.ret_flag = 0
    self.func_list = func_list
    self.threads = []
  def set_thread_func_list(self, func_list):
    """
    @note: func_list是一個list，每個元素是一個dict，有func和args兩個參數
    """
    self.func_list = func_list
  def start(self):
    """
    @note: 啟動多線程執行，并阻塞到結束
    """
    self.threads = []
    self.ret_flag = 0
    for func_dict in self.func_list:
      if func_dict["args"]:
        t = threading.Thread(target=func_dict["func"], args=func_dict["args"])
      else:
        t = threading.Thread(target=func_dict["func"])
      self.threads.append(t)
    for thread_obj in self.threads:
      thread_obj.start()
    for thread_obj in self.threads:
      thread_obj.join()
  def ret_value(self):
    """
    @note: 所有線程函數的返回值之和，如果為0那么表示所有函數執行成功
    """
    return self.ret_flag

MyThread類會接受一個func_list參數，每個元素是一個dict，有func和args兩個key，func是真正要執行的函數引用，args是函數的參數。其中最主要的方法是start方法，會多線程執行每個func，然后一直等到所有線程都執行結束后退出。接下來的關鍵就是如何對self.ret_flag設置正確的值，以判斷所有的線程函數是否都返回0了。

我的實現是，在MyThread class中寫一個方法trace_func，作為直接的線程函數，這個trace_func中執行真正需要執行的函數，從而可以獲取到該函數的返回值，設置給self.ret_flag。

這個trace_func的第一參數是要執行的func引用，后面是這個func的參數，具體代碼如下：    
def start(self):
  """
  @note: 啟動多線程執行，并阻塞到結束
  """
  self.threads = []
  self.ret_flag = 0
  for func_dict in self.func_list:
    if func_dict["args"]:
      new_arg_list = []
      new_arg_list.append(func_dict["func"])
      for arg in func_dict["args"]:
        new_arg_list.append(arg)
      new_arg_tuple = tuple(new_arg_list)
      t = threading.Thread(target=self.trace_func, args=new_arg_tuple)
    else:
      t = threading.Thread(target=self.trace_func, args=(func_dict["func"],))
    self.threads.append(t)
  for thread_obj in self.threads:
    thread_obj.start()
  for thread_obj in self.threads:
    thread_obj.join()

這樣能夠成功獲得返回值了,實驗:    
def func1(ret_num):
  print "func1 ret:%d" % ret_num
  return ret_num
def func2(ret_num):
  print "func2 ret:%d" % ret_num
  return ret_num
def func3():
  print "func3 ret:100"
  return 100
mt = MyThread()
g_func_list = []
g_func_list.append({"func":func1,"args":(1,)})
g_func_list.append({"func":func2,"args":(2,)})
g_func_list.append({"func":func3,"args":None})
mt.set_thread_func_list(g_func_list)
mt.start()
print "all thread ret : %d" % mt.ret_flag

最后的輸出結果
    
func1 ret:1
func2 ret:2
func3 ret:100
all thread ret : 103
總結
以上所述是小編給大家介紹的python使用threading獲取線程函數返回值的實現方法，希望對大家有所幫助