python并行编程|python并行编程 - 线程篇

目录[1]

介绍篇
线程篇
进程篇(待写)
异步篇(待写)
GPU篇(待写)
基本使用 python线程使用的两个模块为: _thread (不推荐再使用)threading
(查看threading的源码可以发现,threading实际是对_thread进一步的封装,官方将其称为 Low-level threading API,下面简单尝试使用_thread)
调用start_new_thread()函数生成新线程
函数声明:_thread.start_new_thread(function, args[, kwargs])
function: 子线程所执行的函数
args: 传递的参数,参数类型必须是元组
kwargs:可选参数

示例:
#!usr/bin/env python #coding=utf-8import _thread import timedef func(t_name): time.sleep(1) print(t_name, 'end')_thread.start_new_thread(func, ('my_thread_1',))# 传递的参数必须是元组类型 print('main thread end') time.sleep(2)# 暂停一下等待子线程,避免主线程结束后直接退出,看不到子线程的输出

【python并行编程|python并行编程 - 线程篇】输出
main thread end
my_thread_1 end
更多:_thread — Low-level threading API
threading模块 需要 import threading
threading模块提供了比_thread模块更加高级别的方法,如下:
  • threading.active_count(): 返回当前运行的线程数量
  • threading.current_thread(): 返回当前运行的线程对象
  • threading.get_ident(): 返回当前运行的线程标识码
  • threading.enumerate(): 获取运作着的线程对象的列表(包含设置了daemon属性的后台线程)
  • threading.main_thread(): 获取主线程对象
threading模块包含Thread类来处理线程
函数声明:class threading.Thread(group=None, target=None, name=None, args=(), kwargs={}, *, daemon=None)
(group: 官方预留的参数)
target: 子线程要执行的函数
name: 给子线程命名
args: 传递参数到要执行的函数中 (类型为元组)
daemon: 将线程设置为后台线程[2]
Thread类包含的方法:
  • start(): 开始线程,它会安排在单独的控制线程中使该对象的run()方法被调用 (invoked)[3] (如果多次调用,会raise RuntimeError
  • run(): 你可以在子类中重写这个方法,标准的run()方法会在构造器传递了target参数后调用它
  • join(timeout=None): 阻塞当前线程,直到等待调用了join()的线程结束,或到达设置的超时timeout的参数为止 (如果尝试加入当前线程[4],因为会发生死锁,join()会raise RuntimeError。在线程启动前调用join()也会报相同的错误)
  • name: 线程名
  • ident: 线程标识码 (如果线程未start(),则为None。实测线程结束后,ident值还存在)
  • is_alive(): 判断线程是否在运行
  • daemon: 是否为后台线程的属性值
  • isDaemon(): 判断是否为后台线程
更多:threading — Thread-based parallelism
函数形式
使用threading.Thread类实例化对象,再调用start()方法运行
示例:
#!usr/bin/env python #coding=utf-8import threading import timedef func(): print(threading.current_thread().name, ' start') time.sleep(1) print(threading.current_thread().name, ' end')t1 = threading.Thread(target=func)# 创建线程 t2 = threading.Thread(target=func)t1.start()# 开始线程 t2.start()# t1.join()# 等待该线程结束后,再往下执行 # t2.join()print('main thread end')# 使用threading模块Thread类的线程,程序需要等待全部线程执行完后才退出

输出
Thread-1 start
Thread-2 start
main thread end
Thread-1 end
Thread-2 end
继承类的形式
通过继承threading.Thread类,可以重写run()方法,再实例化该类,调用start()方法运行
(继承Thread类,并不是非要重写run()
示例:
#!usr/bin/env python #coding=utf-8import threading import timeclass MyThread(threading.Thread): def __init__(self, name): threading.Thread.__init__(self) self.name = namedef run(self): print(self.name + ' start') time.sleep(1) print(self.name + ' end')t1 = MyThread('thread1') t2 = MyThread('thread2')t1.start() t2.start()# t1.join() # t2.join()print('main thread end')

输出
thread1 start
thread2 start
main thread end
thread1 end
thread2 end
线程同步 当属于并发线程的多个操作尝试访问共享内存,并且至少有一个操作能修改数据的状态时,这时如果没有恰当的同步机制,可能会发生意外情况或bug。使用可以解决此问题。
当一个线程想要访问共享内存的某一部分区域时,它必须再使用前获取到该部分的锁。并在操作完后,要释放掉之前获取到的锁。
注意! 要避免死锁[5]的情况发生
使用Lock实现线程同步 使用threading.Lock()实例化Lock锁对象
在共享资源操作的部分,调用Lock的方法acquire()获取锁
结束操作后,调用Lock的方法release()释放锁,以便于其它线程使用该资源
(函数声明:
acquire(blocking=True, timeout=-1)
获取锁,并阻塞其它线程访问这部分资源
blocking[6]: 设置为False的线程不会被阻塞 (并且timeout设置为默认值-1时,失去同步效果。设置为非-1值时,被设置为True的线程阻塞,则False立即返回。这2种情况都会提示错误信息)
timeout: 设置等待的超时值,-1为无限等待,超时后无视阻塞
返回值为True成功获取锁定,False反之(例如超时到期)
release()
在未锁的资源上调用释放锁方法,会引发RuntimeError

示例:
#!usr/bin/env python #coding=utf-8import threading import timelock = threading.Lock()# 创建Lock锁 num = 0# 累加这个变量,观察不同步的情况出现def func(): lock.acquire()# 获取锁 global num for i in range(1000000):# 如果未出现不同步,是由于运算太快,加大循环值 num += 1 lock.release()# 释放锁t1 = threading.Thread(target=func) t2 = threading.Thread(target=func)t1.start() t2.start()t1.join() t2.join()print(num)

有锁情况下,输出
2000000
无锁情况下,不同步,输出
第一次输出
1253312
第二次输出
1227567
第三次输出
1309097
注! 书中并不提倡使用锁来解决,因为可能会导致死锁情况发生,也会对代码的可读性产生影响,调试困难
使用RLock实现线程同步 可重入锁(reentrant lock)
操作方式同Lock锁
与Lock的区别:RLock在 同一个线程中可以多次acquire()获取锁而不发生阻塞 (这是为了解决一些特殊场景的使用)
# 部分代码 lock = threading.RLock()# 创建RLock锁def func(): lock.acquire()# 获取锁 lock.acquire() global num for i in range(1000000): num += 1 lock.release()# 释放锁 lock.release()

注意! acquire()需要成对使用
使用信号量实现线程同步 信号量的提出,首次用在操作系统中。它是一个操作系统管理的抽象数据类型,用于同步多个线程对共享资源与数据的访问
(本质上,信号量是由一个内部变量构成的,它标识出了对其所关联的资源的并发访问量)
使用threading.Semaphore()创建对象
在线程模块中,信号量的操作基于acquire()release()
当一个线程想使用一个资源,它需要调用acquire(),会判断信号量内部变量值_value,如果为0则阻塞线程,并且进行timeout超时处理,如果_value不为0,线程运行 (由于信号量的初始值为非负数,故设计中不存在负数情况的代码)
当一个线程使用完一个资源后,它需要调用release(),该操作会增加信号量的内部变量值,并通知等待的线程
( 注! 书中的描述和threading模块的源码不符,重新按源码的理解写)
注意! acquire()release()并不需要放在某段代码的前后,来锁住某段资源
示例:
(由于书中给的示例代码,感觉很符合理解信号量的特点,这里也采用生产者和消费者的关系编写代码)
#!usr/bin/env python #coding=utf-8import threading import time import random# 可选参数为内部变量_value赋初值,默认为1 # 如果赋的值小于0,会raise ValueError异常 sem = threading.Semaphore(0)def producer(): global item time.sleep(1) item = random.randint(0, 1000) print('producer: produced', item) sem.release()# 释放信号量,将内部_value加1,并通知其它等待的线程def consumer(): print('consumer is waiting') sem.acquire()# 获取信号量,值等于0则阻塞线程,否则内部_value减1,并继续运行 print('consumer: sonsumed', item)t1 = threading.Thread(target=producer) t2 = threading.Thread(target=consumer)t1.start() t2.start()

输出
consumer is waiting
producer: produced 295
consumer: sonsumed 295
分析:多执行几次,逻辑上可以发现消费者总需要等待生产者生产出产品后,才能消费
可以看出信号量很适合这样的场景,下面可以测试,生产者可以多生产几个,消费者再消费
>>> import threading >>> sem = threading.Semaphore() >>> sem.acquire()# 获取初始化的信号量 True >>> sem.release()# 信号量+1 >>> sem.release()# 信号量+1 >>> sem.release()# 信号量+1 >>> sem.acquire()# 信号量-1 True >>> sem.acquire()# 信号量-1 True >>> sem.acquire()# 信号量-1 True >>> sem.acquire()# 由于信号量=0 # 所以陷入了阻塞

使用条件实现线程同步 使用threading.Condition()创建对象
查看Condition的源码,发现可以传入一个锁作为初始化参数。如果不传,默认会赋值RLock锁来进行后续的锁的操作 acquire()release()
# Condition类的初始化部分源码 def __init__(self, lock=None): if lock is None: lock = RLock() self._lock = lock # Export the lock's acquire() and release() methods self.acquire = lock.acquire self.release = lock.release ...

示例:
(下例还是使用生产者和消费者的关系编写示例代码,用items作为存储容器,以满了(10个)就不能再生产作为条件,以没了(0个)就不能再消费作为条件)
#!usr/bin/env python #coding=utf-8import threading import timecondition = threading.Condition()# 创建条件 items = []# 作为产品的存储容器,设达到10个为满了,就不能再生产了def producer(): global items condition.acquire()# 获取锁 if len(items) == 10: print('producer: stop produce') condition.wait()# 等待(items达到10,等待消费者消费) items.append('') print('producer: produced', len(items)) condition.notify()# 通知等待的线程 condition.release()# 释放锁def consumer(): global items condition.acquire()# 获取锁 if len(items) == 0: print('consumer: waiting') condition.wait()# 等待(items为0,等待生产者生产) items.pop()# 注! 注释掉这行,会发现,等待wait()在接收到通知notify()后,并没有再次判断条件,直接就接着运行了 print('consumer: sonsumed', len(items)) condition.notify()# 通知等待的线程 condition.release()# 释放锁# producer_loop()和consumer_loop()用来多次循环运行,为了达到items为0或10的情况 def producer_loop(): for i in range(20): time.sleep(1) producer()def consumer_loop(): for i in range(20): time.sleep(4) consumer()t1 = threading.Thread(target=producer_loop) t2 = threading.Thread(target=consumer_loop)t1.start() t2.start()

输出
producer: produced 1
producer: produced 2
producer: produced 3
consumer: sonsumed 2
producer: produced 3
producer: produced 4
producer: produced 5
producer: produced 6
consumer: sonsumed 5
producer: produced 6
producer: produced 7
producer: produced 8
producer: produced 9
consumer: sonsumed 8
producer: produced 9
producer: produced 10
producer: stop produce
consumer: sonsumed 9
producer: produced 10
producer: stop produce
consumer: sonsumed 9
producer: produced 10
producer: stop produce
consumer: sonsumed 9
producer: produced 10
producer: stop produce
consumer: sonsumed 9
producer: produced 10
producer: stop produce
consumer: sonsumed 9
producer: produced 10
producer: stop produce
consumer: sonsumed 9
producer: produced 10
producer: stop produce
consumer: sonsumed 9
producer: produced 10
consumer: sonsumed 9
consumer: sonsumed 8
consumer: sonsumed 7
consumer: sonsumed 6
consumer: sonsumed 5
consumer: sonsumed 4
consumer: sonsumed 3
consumer: sonsumed 2
consumer: sonsumed 1
consumer: sonsumed 0
分析:观察可以发现,生产者生产满了10个就会进入等待wait(),直到消费者通知notify()
为了观察消费者消费到0个的情况,可以将生产者和消费者循环的等待时间做调整

使用事件实现线程同步 查看Event类的源码,发现内部使用的是条件Condition类实现,并传入了Lock锁。事件通过对内部的标志_flag进行管理来实现线程同步
使用set()方法可以将标志设为True
使用clear()方法将其重置为False
使用wait()方法阻塞线程
# Event类的初始化源码 def __init__(self): self._cond = Condition(Lock()) self._flag = False

注意! 并不存在set()clear()放在某段代码的前后,来锁住某段资源
示例:
(还是采用生产者和消费者的关系编写,生产者做完工作,调用set()设置_flag标志,并通知wait()等待的消费者线程运行;再使用clear()清除_flag标志,以便后面的消费者能正确的进入等待。这个过程类似于事件的触发)
#!usr/bin/env python #coding=utf-8import threading import time import randomevent = threading.Event()# 创建事件 items = [1,2,3] # 设置个初值,便于后面注释event.clear()后的测试(可以发现,消费者不等待了)def producer(): global items print('producer: start') items.append(random.randint(0,100)) event.set()# 将内部标志_flag设为True,并通知所有等待的线程(类似于触发事件) print('producer: notify') event.clear()# 将内部标志_flag设为False(只有清除了_flag,消费者下一次的wait()操作才会正常进入等待)(clear()操作也可以交给消费者调用,不过为了简化消费者的操作,让消费者只需要等待通知即可) print('producer: end')def consumer(): global items print('consumer: waiting') event.wait()# 等待(等待生产者通知,根据_flag标志判断是否进入等待) print('consumer:', items.pop())# producer_loop()和consumer_loop()用来多次循环运行 def producer_loop(): for i in range(3): time.sleep(1) producer()def consumer_loop(): while True: # 消费者有wait()等待,就不用线程休眠了,以免错过生产者的set()通知 consumer()t1 = threading.Thread(target=producer_loop) t2 = threading.Thread(target=consumer_loop)t1.start() t2.start()

输出
consumer: waiting
producer: start
producer: notify
consumer: 57
consumer: waiting
producer: end
producer: start
producer: notify
consumer: 10
producer: end
consumer: waiting
producer: start
producer: notify
consumer: 24
producer: end
consumer: waiting
如果注释掉代码中的# event.clear()一行,会出现如下输出
consumer: waiting
producer: start
producer: notify
consumer: 19
producer: end
consumer: waiting
consumer: 3
consumer: waiting
consumer: 2
consumer: waiting
consumer: 1
consumer: waiting
Exception in thread Thread-2:
Traceback (most recent call last):
File "D:\app\Python\Python37\lib\threading.py", line 917, in _bootstrap_inner
self.run()
File "D:\app\Python\Python37\lib\threading.py", line 865, in run
self._target(*self._args, **self._kwargs)
File "g:/tmp/code.py", line 35, in consumer_loop
consumer()
File "g:/tmp/code.py", line 24, in consumer
print('consumer:', items.pop())
IndexError: pop from empty list
producer: start
producer: notify
producer: end
producer: start
producer: notify
producer: end
分析:会发现,消费者没有等待(打印出的consumer: waiting,只是给自己的提示,实际没有等待),直接进行了列表的pop()操作,直到后面列表为空再弹出时报错为止
附:其它 使用with语句[1] (由于没有理解部分,这一部分基本就是书中原文)
with: 是Python 2.5中引入的。当有两个相关的操作需要对一个代码块承兑执行时,with语句的作用就彰显出来了。它可以再自动精确的分配或释放资源 (因此也被称为上下文管理器)。如线程模块中,使用到acquire()release()方法的地方,都可以采用with语句块,如下:
  • Lock
  • RLock
  • 条件
  • 信号量 (感觉这用了with就不太灵活了)
示例:
(会测试上述列表中的with操作)
#!usr/bin/env python #coding=utf-8import threadingdef func_with(statement): with statement: # 会自动进行acquire()和release() print('//todo1')def func_not_with(statement): statement.acquire() try:# 为了避免出现异常,导致没有release()释放 print('//todo2') finally: statement.release()lock = threading.Lock() rlock = threading.RLock() condition = threading.Condition() sem = threading.Semaphore(1)# 采用with,需要初始至少有一个信号量值(因为需要先acquire())li = [lock, rlock, condition, sem]for statement in li: t1 = threading.Thread(target=func_with, args=(statement,)) t2 = threading.Thread(target=func_not_with, args=(statement,))t1.start() t2.start()

使用队列实现线程通信 虽然python线程模块提供了很多同步原语 信号量条件事件,但有时候,在使用场景中,可能采用队列模块会是个最佳选择。它使得线程编程变得更加容易和安全
使用队列Queue的方式处理,尽管它不属于threading模块,但查看其源码,发现队列的功能实现有用到threading模块
# Queue的初始化源码 def __init__(self, maxsize=0): self.maxsize = maxsize self._init(maxsize)# mutex must be held whenever the queue is mutating.All methods # that acquire mutex must release it before returning.mutex # is shared between the three conditions, so acquiring and # releasing the conditions also acquires and releases mutex. self.mutex = threading.Lock()# Notify not_empty whenever an item is added to the queue; a # thread waiting to get is notified then. self.not_empty = threading.Condition(self.mutex)# Notify not_full whenever an item is removed from the queue; # a thread waiting to put is notified then. self.not_full = threading.Condition(self.mutex)# Notify all_tasks_done whenever the number of unfinished tasks # drops to zero; thread waiting to join() is notified to resume self.all_tasks_done = threading.Condition(self.mutex) self.unfinished_tasks = 0

Queue会用到如下方法:
  • put(): 添加一个项目到队列
  • get(): 从队列中取出一个项目
  • task_done(): 每处理完一个项目,需要调用该方法
  • join(): 阻塞线程,等待全部的任务完成
附:查看源代码,分析可得Queue内部的操作是这样的:(这段可以不用看)
  • 调用put() -> with队列满(条件锁not_full) -> 能阻塞?(参数block) & 能超时?(参数timeout) & 需要等待?(条件锁not_full.wait()) -> 添加数据(内部方法_put()) -> 任务计数加1(内部计数变量unfinished_tasks += 1) -> 发起队列非空的通知(条件锁not_empty.notify())
  • 调用get() -> with队列空(条件锁not_empty) -> 能阻塞?(参数block) & 能超时?(参数timeout) & 需要等待?(条件锁not_empty.wait()) -> 取出数据(内部方法_get()) -> 发起队列非满的通知(条件锁not_full.notify()) ( 注意! get()put()操作相比并没有对任务计数操作,需要通过后面task_done()完成任务方法来减少任务数)
  • *调用task_done() -> with任务完成(条件锁all_tasks_done) -> 判断全部任务完成了?是的话发起通知(条件锁all_tasks_done.notify_all()) -> 任务计数减1(内部计数变量unfinished_tasks -= 1) *
  • 调用join() -> with任务完成(条件锁all_tasks_done) -> 循环未完成的任务计数变量(内部计数变量unfinished_tasks) -> 还有没完成的任务,等待(条件锁all_tasks_done.wait() | 全部完成,退出循环,解除线程阻塞)
示例:
#!usr/bin/env python #coding=utf-8import threading from queue import Queue import time import randomqueue = Queue()def producer(): for i in range(5): item = random.randint(0, 100) queue.put(item) print('producer:', 'put', item) time.sleep(1)def consumer(): while True: item = queue.get() print('consumer:', 'get', item) queue.task_done()t1 = threading.Thread(target=producer) t2 = threading.Thread(target=consumer)t1.start() t2.start()

输出
producer: put 47
consumer: get 47
producer: put 71
consumer: get 71
producer: put 99
consumer: get 99
producer: put 30
consumer: get 30
producer: put 75
consumer: get 75
  1. 参考书籍:《Python并行编程手册》 ? ?
  2. 后台线程在主线程停止后就直接停止运行。他们资源(如打开的文件,数据库事务等)可能不会被正确的释放。如果你想要你的线程优美的停止,让他们不要变为后台和使用一个合适的信号机制如事件Event ?
  3. 官方文档中使用invoke一词,我并没有更好的翻译,因为其它语言中invoke反射是一种技术手段,但Google的翻译中,将此解释为调用 (是我想多了) ?
  4. 加入线程?不理解如何能加入线程,并且官方文档说会死锁。实测,创建2个线程互相join(),虽然陷入死循环,但并没抛出错误 // TODO: 不知理解有偏差没有 ?
  5. 多个对象,互持对方所需资源的锁,导致都无法访问 ?
  6. 锁的acquire()方法的参数,有点难理解,文中所写是结合官方文档和实测的结果描述所得。不过一般,我们都不用改变它的默认值 // TODO: 没有从源码分析(找不到源码) ?

    推荐阅读