快捷導(dǎo)航

python使用threading獲取線程函數(shù)返回值的實(shí)現(xiàn)方法

更新時(shí)間：2017年11月15日 17:17:35 作者：QualitySong

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

threading用于提供線程相關(guān)的操作，線程是應(yīng)用程序中工作的最小單元。python當(dāng)前版本的多線程庫沒有實(shí)現(xiàn)優(yōu)先級(jí)、線程組，線程也不能被停止、暫停、恢復(fù)、中斷。

threading模塊提供的類：

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

threading 模塊提供的常用方法：

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

　　threading.enumerate(): 返回一個(gè)包含正在運(yùn)行的線程的list。正在運(yùn)行指線程啟動(dòng)后、結(jié)束前，不包括啟動(dòng)前和終止后的線程。

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

threading 模塊提供的常量：

　　threading.TIMEOUT_MAX 設(shè)置threading全局超時(shí)時(shí)間。

好了，正文開始：

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

一開始考慮到執(zhí)行的操作可能有很多，而且后續(xù)會(huì)不斷補(bǔ)充，因此先寫了一個(gè)通用的多線程執(zhí)行類，封裝線程操作的基本方法，如下：

import threading
class MyThread(object):
  def __init__(self, func_list=None):
  #所有線程函數(shù)的返回值匯總，如果最后為0，說明全部成功
    self.ret_flag = 0
    self.func_list = func_list
    self.threads = []
  def set_thread_func_list(self, func_list):
    """
    @note: func_list是一個(gè)list，每個(gè)元素是一個(gè)dict，有func和args兩個(gè)參數(shù)
    """
    self.func_list = func_list
  def start(self):
    """
    @note: 啟動(dòng)多線程執(zhí)行，并阻塞到結(jié)束
    """
    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: 所有線程函數(shù)的返回值之和，如果為0那么表示所有函數(shù)執(zhí)行成功
    """
    return self.ret_flag

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

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

這個(gè)trace_func的第一參數(shù)是要執(zhí)行的func引用，后面是這個(gè)func的參數(shù)，具體代碼如下：

def trace_func(self, func, *args, **kwargs):
  """
  @note:替代profile_func，新的跟蹤線程返回值的函數(shù)，對(duì)真正執(zhí)行的線程函數(shù)包一次函數(shù)，以獲取返回值
  """
  ret = func(*args, **kwargs)
  self.ret_flag += ret

這樣就需要修改start方法中Thread函數(shù)的設(shè)置，代碼如下：

def start(self):
  """
  @note: 啟動(dòng)多線程執(zhí)行，并阻塞到結(jié)束
  """
  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()

這樣能夠成功獲得返回值了,實(shí)驗(yàn):

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

最后的輸出結(jié)果

func1 ret:1
func2 ret:2
func3 ret:100
all thread ret : 103

總結(jié)

以上所述是小編給大家介紹的python使用threading獲取線程函數(shù)返回值的實(shí)現(xiàn)方法，希望對(duì)大家有所幫助，如果大家有任何疑問請(qǐng)給我留言，小編會(huì)及時(shí)回復(fù)大家的。在此也非常感謝大家對(duì)腳本之家網(wǎng)站的支持！

您可能感興趣的文章: