昨日内容回顾

进程

multiprocess

Process —— 进程 在python中创建一个进程的模块

  1. start
  2. daemon 守护进程
  3. join 等待子进程执行结束

锁 Lock

acquire release

锁是一个同步控制的工具

如果同一时刻有多个进程同时执行一段代码,

那么在内存中的数据是不会发生冲突的

但是,如果涉及到文件,数据库就会发生资源冲突的问题

我们就需要用锁来把这段代码锁起来

任意一个进程执行了 acquire 之后,

其他所有的进程都会在这里阻塞,等待一个 release

信号量 semaphore

锁 + 计数器

同一时间只能有指定个数的进程执行同一段代码

事件 Event

set clear is_set 控制对象的状态

wait 根据状态不同执行效果也不同

  1. 状态是 True ---> pass
  2. 状态是 False --> 阻塞

一般 wait 是和 set clear 放在不同的进程中

set/clear 负责控制状态

wait 负责感知状态

我可以在一个进程中控制另外一个或多个进程的运行情况

IPC 通信

队列 Queue

管道 PIPE

进程间通信(队列和管道)

判断队列是否为空

  1. from multiprocessing import Process,Queue
  2. q = Queue()
  3. print(q.empty())

执行输出:True

判断队列是否满了

  1. from multiprocessing import Process,Queue
  2. q = Queue()
  3. print(q.full())

执行输出:False

如果队列已满,再增加值的操作,会被阻塞,直到队列有空余的

  1. from multiprocessing import Process,Queue
  2. q = Queue(10) # 创建一个只能放 10 个 value 的队列
  3. for i in range(10):
  4. q.put(i) # 增加一个 value
  5. print(q.qsize()) # 返回队列中目前项目的正确数量
  6. print(q.full()) # 如果 q 已满,返回为 True
  7. q.put(111) # 再增加一个值
  8. print(q.empty())

执行输出:

10

True

从结果中,可以看出,下面的操作 q.put(111)之后的代码被阻塞了。

总结:

队列可以在创建的时候制定一个容量

如果在程序运行的过程中,队列已经有了足够的数据,再 put 就会发生**阻塞**

如果队列为空,在 get 就会发生阻塞

为什么要指向队列的长度呢?是为了防止内存爆炸。

一个队列,不能无限制的存储。毕竟,内存是有限制的。

  1. ![](https://cdn.nlark.com/yuque/0/2020/png/1484428/1597999263964-a0a7cc20-563c-41f1-9127-4d4e636ad5f9.png)

上面提到的** put,get,qsize,full,empty 都是不准的。**

因为在返回结果和在稍后程序中使用结果之间,队列中可能添加或删除了项目。在某些系统上,此方法可能引发 NotImplementedError 异常。

如果其他进程或线程正在往队列中添加项目,结果是不可靠的。也就是说,在返回和使用结果之间,队列中可能已经加入新的项目。

  1. import time
  2. from multiprocessing import Process,Queue
  3. def wahaha(q):
  4. print(q.get())
  5. q.put(2) # 增加数字 2
  6. if __name__ == '__main__':
  7. q = Queue()
  8. p = Process(target=wahaha,args=[q,])
  9. p.start()
  10. q.put(1) # 增加数字 1
  11. time.sleep(0.1)
  12. print(q.get())

执行输出:

1

2

先执行主进程的 q.get(),再执行子进程的 q.get()

在进程中使用队列可以完成双向通信

队列是进程安全的 内置了锁来保证队列中的每一个数据都不会被多个进程重复取

  1. ![](https://cdn.nlark.com/yuque/0/2020/png/1484428/1597999264053-eed1dbbd-6156-4491-8a4f-8d065d5923a9.png)

在同一时刻,只能有一个进程来取值,它内部有一个锁的机制。那么另外一个进程就会阻塞一会,但是阻塞的时间非常短

队列能保证数据安全,同一个数据,不能被多个进程获取。

生产者消费者模型

解决数据供需不平衡的情况

  1. from multiprocessing import Process,Queue
  2. def producer(q,name,food):
  3. for i in range(5):
  4. print('{}生产了{}{}'.format(name,food,i))
  5. if __name__ == '__main__':
  6. q = Queue()
  7. Process(target=producer,args=[q,'康师傅','红烧牛肉']).start()
  8. Process(target=producer,args=[q,'郑师傅','红烧鱼块']).start()

执行输出:

康师傅生产了红烧牛肉 0

康师傅生产了红烧牛肉 1

康师傅生产了红烧牛肉 2

康师傅生产了红烧牛肉 3

康师傅生产了红烧牛肉 4

郑师傅生产了红烧鱼块 0

郑师傅生产了红烧鱼块 1

郑师傅生产了红烧鱼块 2

郑师傅生产了红烧鱼块 3

郑师傅生产了红烧鱼块 4

增加一个消费者

  1. import time
  2. import random
  3. from multiprocessing import Process,Queue
  4. def producer(q,name,food):
  5. for i in range(5):
  6. time.sleep(random.random()) # 模拟生产时间
  7. print('{}生产了{}{}'.format(name,food,i))
  8. q.put('{}{}'.format(food,i)) # 放入队列
  9. def consumer(q,name):
  10. for i in range(10):
  11. food = q.get() # 获取队列
  12. time.sleep(random.random()) # 模拟吃的时间
  13. print('{}吃了{}'.format(name,food))
  14. if __name__ == '__main__':
  15. q = Queue()
  16. Process(target=producer,args=[q,'康师傅','红烧牛肉']).start()
  17. Process(target=producer,args=[q,'郑师傅','红烧鱼块']).start()
  18. Process(target=consumer,args=[q,'xiao']).start()

执行输出:

郑师傅生产了红烧鱼块 0

xiao 吃了红烧鱼块 0

康师傅生产了红烧牛肉 0

xiao 吃了红烧牛肉 0

康师傅生产了红烧牛肉 1

郑师傅生产了红烧鱼块 1

xiao 吃了红烧牛肉 1

康师傅生产了红烧牛肉 2

郑师傅生产了红烧鱼块 2

康师傅生产了红烧牛肉 3

郑师傅生产了红烧鱼块 3

康师傅生产了红烧牛肉 4

xiao 吃了红烧鱼块 1

郑师傅生产了红烧鱼块 4

xiao 吃了红烧牛肉 2

xiao 吃了红烧鱼块 2

xiao 吃了红烧牛肉 3

xiao 吃了红烧鱼块 3

xiao 吃了红烧牛肉 4

xiao 吃了红烧鱼块 4

消费者,必须是有的吃,才能吃。没有吃的,就等着。

一个消费者,明显消费不过来。再加一个消费者

  1. import time
  2. import random
  3. from multiprocessing import Process,Queue
  4. def producer(q,name,food):
  5. for i in range(5):
  6. time.sleep(random.random()) # 模拟生产时间
  7. print('{}生产了{}{}'.format(name,food,i))
  8. q.put('{}{}'.format(food,i)) # 放入队列
  9. def consumer(q,name):
  10. for i in range(5): # 修改为 5,因为有 2 个人
  11. food = q.get() # 获取队列
  12. time.sleep(random.random()) # 模拟吃的时间
  13. print('{}吃了{}'.format(name,food))
  14. if __name__ == '__main__':
  15. q = Queue()
  16. Process(target=producer,args=[q,'康师傅','红烧牛肉']).start()
  17. Process(target=producer,args=[q,'郑师傅','红烧鱼块']).start()
  18. Process(target=consumer,args=[q,'xiao']).start()
  19. Process(target=consumer, args=[q,'lin']).start()

执行输出:

康师傅生产了红烧牛肉 0

郑师傅生产了红烧鱼块 0

xiao 吃了红烧牛肉 0

郑师傅生产了红烧鱼块 1

康师傅生产了红烧牛肉 1

lin 吃了红烧鱼块 0

郑师傅生产了红烧鱼块 2

康师傅生产了红烧牛肉 2

郑师傅生产了红烧鱼块 3

xiao 吃了红烧鱼块 1

郑师傅生产了红烧鱼块 4

lin 吃了红烧牛肉 1

xiao 吃了红烧鱼块 2

康师傅生产了红烧牛肉 3

xiao 吃了红烧鱼块 3

lin 吃了红烧牛肉 2

xiao 吃了红烧鱼块 4

康师傅生产了红烧牛肉 4

lin 吃了红烧牛肉 3

lin 吃了红烧牛肉 4

注意:必须将消费者的 rang(10)修改为 5,否则程序会卡住。为什么呢?因为队列已经是空的,再取就会阻塞

这样才能解决供需平衡

那么问题来了,如果有一个消费者,吃的比较快呢?

再修改 range 值?太 Low 了

能者多劳嘛,

不能使用 q.empty(),它是不准确的

看下图,有可能一开始,队列就空了

下面的 0.1 更快

  1. ![](https://cdn.nlark.com/yuque/0/2020/png/1484428/1597999263915-8b2650cc-0523-4932-8929-f55b5ebb8559.png)

看下面的解决方案:

  1. import time
  2. import random
  3. from multiprocessing import Process,Queue
  4. def producer(q,name,food):
  5. for i in range(5):
  6. time.sleep(random.random()) # 模拟生产时间
  7. print('{}生产了{}{}'.format(name,food,i))
  8. q.put('{}{}'.format(food,i)) # 放入队列
  9. def consumer(q,name):
  10. while True:
  11. food = q.get() # 获取队列
  12. if food == 'done':break # 当获取的值为 done 时,结束循环
  13. time.sleep(random.random()) # 模拟吃的时间
  14. print('{}吃了{}'.format(name,food))
  15. if __name__ == '__main__':
  16. q = Queue() #创建队列对象,如果不提供 maxsize,则队列数无限制
  17. p1 = Process(target=producer,args=[q,'康师傅','红烧牛肉'])
  18. p2 = Process(target=producer,args=[q,'郑师傅','红烧鱼块'])
  19. p1.start() # 启动进程
  20. p2.start()
  21. Process(target=consumer,args=[q,'xiao']).start()
  22. Process(target=consumer, args=[q,'lin']).start()
  23. p1.join() # 保证子进程结束后再向下执行
  24. p2.join()
  25. q.put('done') # 向队列添加一个值 done
  26. q.put('done')

执行输出:

康师傅生产了红烧牛肉 0

郑师傅生产了红烧鱼块 0

康师傅生产了红烧牛肉 1

郑师傅生产了红烧鱼块 1

xiao 吃了红烧牛肉 0

xiao 吃了红烧牛肉 1

康师傅生产了红烧牛肉 2

康师傅生产了红烧牛肉 3

xiao 吃了红烧鱼块 1

lin 吃了红烧鱼块 0

郑师傅生产了红烧鱼块 2

lin 吃了红烧牛肉 3

xiao 吃了红烧牛肉 2

康师傅生产了红烧牛肉 4

xiao 吃了红烧牛肉 4

lin 吃了红烧鱼块 2

郑师傅生产了红烧鱼块 3

xiao 吃了红烧鱼块 3

郑师傅生产了红烧鱼块 4

lin 吃了红烧鱼块 4

为什么要有 2 个 done?因为有 2 个消费者

为什么要有 2 个 join?因为必须要等厨师做完菜才可以。

最后输出 2 个 done,表示通知 2 个顾客,菜已经上完了,顾客要结账了。

2 个消费者,都会执行 break。通俗的来讲,亲,您一共消费了 xx 元,请付款!

上面的解决方案,代码太长了,有一个消费者,就得 done 一次。

下面介绍 JoinableQueue

JoinableQueue([maxsize])

创建可连接的共享进程队列。这就像是一个 Queue 对象,但队列允许项目的使用者通知生产者项目已经被成功处理。通知进程是使用共享的信号和条件变量来实现的。

方法介绍

  1. JoinableQueue 的实例 p 除了与 Queue 对象相同的方法之外,还具有以下方法:
  2. q.task_done()
  3. 使用者使用此方法发出信号,表示 q.get()返回的项目已经被处理。如果调用此方法的次数大于从队列中删除的项目数量,将引发 ValueError 异常。
  4. q.join()
  5. 生产者将使用此方法进行阻塞,直到队列中所有项目均被处理。阻塞将持续到为队列中的每个项目均调用 q.task_done()方法为止。
  6. 下面的例子说明如何建立永远运行的进程,使用和处理队列上的项目。生产者将项目放入队列,并等待它们被处理。

JoinableQueue 队列实现消费之生产者模型

  1. import time
  2. import random
  3. from multiprocessing import Process,JoinableQueue
  4. def producer(q,name,food):
  5. for i in range(5):
  6. time.sleep(random.random()) # 模拟生产时间
  7. print('{}生产了{}{}'.format(name,food,i))
  8. q.put('{}{}'.format(food,i))
  9. q.join() # 等到所有的数据都被 task_done 才结束
  10. def consumer(q,name):
  11. while True:
  12. food = q.get() # 获取队列
  13. time.sleep(random.random()) # 模拟吃的时间
  14. print('{}吃了{}'.format(name,food))
  15. q.task_done() # 向 q.join()发送一次信号,证明一个数据已经被取走了
  16. if __name__ == '__main__':
  17. q = JoinableQueue() #创建可连接的共享进程队列
  18. # 生产者们:即厨师们
  19. p1 = Process(target=producer,args=[q,'康师傅','红烧牛肉'])
  20. p2 = Process(target=producer,args=[q,'郑师傅','红烧鱼块'])
  21. p1.start() # 启动进程
  22. p2.start()
  23. # 消费者们:即吃货们
  24. c1 = Process(target=consumer,args=[q,'xiao'])
  25. c2 = Process(target=consumer, args=[q,'lin'])
  26. c1.daemon = True # 设置守护进程
  27. c2.daemon = True
  28. c1.start() # 启动进程
  29. c2.start()
  30. p1.join() # 保证子进程结束后再向下执行
  31. p2.join()

执行输出:

康师傅生产了红烧牛肉 0

xiao 吃了红烧牛肉 0

郑师傅生产了红烧鱼块 0

康师傅生产了红烧牛肉 1

lin 吃了红烧鱼块 0

xiao 吃了红烧牛肉 1

郑师傅生产了红烧鱼块 1

康师傅生产了红烧牛肉 2

xiao 吃了红烧牛肉 2

lin 吃了红烧鱼块 1

郑师傅生产了红烧鱼块 2

康师傅生产了红烧牛肉 3

xiao 吃了红烧鱼块 2

lin 吃了红烧牛肉 3

康师傅生产了红烧牛肉 4

郑师傅生产了红烧鱼块 3

xiao 吃了红烧牛肉 4

lin 吃了红烧鱼块 3

郑师傅生产了红烧鱼块 4

xiao 吃了红烧鱼块 4

总结:

  1. producer
  2. put
  3. 生产完全部的数据就没有其他工作了
  4. 在生产数据方 允许执行 q.join
  5. join 会发起一个阻塞,直到所有当前队列中的数据都被消费
  6. consumer
  7. get 获取到数据
  8. 处理数据
  9. q.task_done() 告诉 q,刚刚从 q 获取的数据已经处理完了
  10. consumer 每完成一个任务就会给 q 发送一个 taskdone
  11. producer 在所有的数据都生产完之后会执行 q.join()
  12. producer 会等待 consumer 消费完数据才结束
  13. 主进程中对 producer 进程进行 join
  14. 主进程中的代码会等待 producer 执行完才结束
  15. producer 结束就意味着主进程代码的结束
  16. consumer 作为守护进程结束
  17. 结束顺序:
  18. consumer queue 中的所有数据被消费
  19. producer join 结束
  20. 主进程的代码结束
  21. consumer 结束
  22. 主进程结束

管道(了解)

介绍

  1. #创建管道的类:
  2. Pipe([duplex]):在进程之间创建一条管道,并返回元组(conn1,conn2),其中 conn1conn2 表示管道两端的连接对象,强调一点:必须在产生 Process 对象之前产生管道
  3. #参数介绍:
  4. dumplex:默认管道是全双工的,如果将 duplex 射成 Falseconn1 只能用于接收,conn2 只能用于发送。
  5. #主要方法:
  6. conn1.recv():接收 conn2.send(obj)发送的对象。如果没有消息可接收,recv 方法会一直阻塞。如果连接的另外一端已经关闭,那么 recv 方法会抛出 EOFError
  7. conn1.send(obj):通过连接发送对象。obj 是与序列化兼容的任意对象
  8. #其他方法:
  9. conn1.close():关闭连接。如果 conn1 被垃圾回收,将自动调用此方法
  10. conn1.fileno():返回连接使用的整数文件描述符
  11. conn1.poll([timeout]):如果连接上的数据可用,返回 Truetimeout 指定等待的最长时限。如果省略此参数,方法将立即返回结果。如果将 timeout 射成 None,操作将无限期地等待数据到达。
  12. conn1.recv_bytes([maxlength]):接收 c.send_bytes()方法发送的一条完整的字节消息。maxlength 指定要接收的最大字节数。如果进入的消息,超过了这个最大值,将引发 IOError 异常,并且在连接上无法进行进一步读取。如果连接的另外一端已经关闭,再也不存在任何数据,将引发 EOFError 异常。
  13. conn.send_bytes(buffer [, offset [, size]]):通过连接发送字节数据缓冲区,buffer 是支持缓冲区接口的任意对象,offset 是缓冲区中的字节偏移量,而 size 是要发送字节数。结果数据以单条消息的形式发出,然后调用 c.recv_bytes()函数进行接收
  14. conn1.recv_bytes_into(buffer [, offset]):接收一条完整的字节消息,并把它保存在 buffer 对象中,该对象支持可写入的缓冲区接口(即 bytearray 对象或类似的对象)。offset 指定缓冲区中放置消息处的字节位移。返回值是收到的字节数。如果消息长度大于可用的缓冲区空间,将引发 BufferTooShort 异常。

pipe 初使用

  1. from multiprocessing import Process, Pipe
  2. def f(conn):
  3. conn.send("Hello The_Third_Wave")
  4. conn.close()
  5. if __name__ == '__main__':
  6. parent_conn, child_conn = Pipe()
  7. p = Process(target=f, args=(child_conn,))
  8. p.start()
  9. print(parent_conn.recv())
  10. p.join()

应该特别注意管道端点的正确管理问题。如果是生产者或消费者中都没有使用管道的某个端点,就应将它关闭。这也说明了为何在生产者中关闭了管道的输出端,在消费者中关闭管道的输入端。如果忘记执行这些步骤,程序可能在消费者中的 recv()操作上挂起。管道是由操作系统进行引用计数的,必须在所有进程中关闭管道后才能生成 EOFError 异常。因此,在生产者中关闭管道不会有任何效果,除非消费者也关闭了相同的管道端点。

  1. ![](https://cdn.nlark.com/yuque/0/2020/png/1484428/1597999263927-2ef68982-2308-4d4b-a269-0804c384d02a.png)
  1. from multiprocessing import Pipe
  2. left,right = Pipe()
  3. left.send('1234')
  4. print(right.recv())

执行输出:1234

管道实例化之后,形成 2 端。默认情况下,管道是双向的

左边 send,右边 recv

一端 send 和 recv,会阻塞

它不是走 TCP 和 UDP

它是一台机器的多个进程

引发 EOFError,程序卡住

  1. from multiprocessing import Process, Pipe
  2. def f(parent_conn,child_conn):
  3. parent_conn.close() #不写 close 将不会引发 EOFError
  4. while True:
  5. try:
  6. print(child_conn.recv())
  7. except EOFError:
  8. child_conn.close()
  9. break
  10. if __name__ == '__main__':
  11. # 在进程之间创建一条管道,并返回元组(conn1,conn2),其中 conn1 和 conn2 是表示管道两端的 Connection 对象
  12. parent_conn, child_conn = Pipe()
  13. p = Process(target=f, args=(parent_conn,child_conn,))
  14. p.start()
  15. child_conn.close() #关闭连接
  16. parent_conn.send('hello')
  17. parent_conn.send('hello')
  18. parent_conn.send('hello')
  19. parent_conn.close()
  20. p.join() # 等待子进程结束

执行输出:

hello

hello

hello

pipe 实现生产者消费者模型

  1. from multiprocessing import Process,Pipe
  2. def consumer(p,name):
  3. produce, consume=p
  4. produce.close()
  5. while True:
  6. try:
  7. baozi=consume.recv()
  8. print('%s 收到包子:%s' %(name,baozi))
  9. except EOFError:
  10. break
  11. def producer(seq,p):
  12. produce, consume=p
  13. consume.close()
  14. for i in seq:
  15. produce.send(i)
  16. if __name__ == '__main__':
  17. produce,consume=Pipe()
  18. c1=Process(target=consumer,args=((produce,consume),'c1'))
  19. c1.start()
  20. seq=(i for i in range(10))
  21. producer(seq,(produce,consume))
  22. produce.close()
  23. consume.close()
  24. c1.join()
  25. print('主进程')

多个消费之之间的竞争问题带来的数据不安全问题

  1. from multiprocessing import Process,Pipe,Lock
  2. def consumer(p,name,lock):
  3. produce, consume=p
  4. produce.close()
  5. while True:
  6. lock.acquire()
  7. baozi=consume.recv()
  8. lock.release()
  9. if baozi:
  10. print('%s 收到包子:%s' %(name,baozi))
  11. else:
  12. consume.close()
  13. break
  14. def producer(p,n):
  15. produce, consume=p
  16. consume.close()
  17. for i in range(n):
  18. produce.send(i)
  19. produce.send(None)
  20. produce.send(None)
  21. produce.close()
  22. if __name__ == '__main__':
  23. produce,consume=Pipe()
  24. lock = Lock()
  25. c1=Process(target=consumer,args=((produce,consume),'c1',lock))
  26. c2=Process(target=consumer,args=((produce,consume),'c2',lock))
  27. p1=Process(target=producer,args=((produce,consume),10))
  28. c1.start()
  29. c2.start()
  30. p1.start()
  31. produce.close()
  32. consume.close()
  33. c1.join()
  34. c2.join()
  35. p1.join()
  36. print('主进程')

进程之间的数据共享

展望未来,基于消息传递的并发编程是大势所趋

即便是使用线程,推荐做法也是将程序设计为大量独立的线程集合,通过消息队列交换数据。

这样极大地减少了对使用锁定和其他同步手段的需求,还可以扩展到分布式系统中。

但进程间应该尽量避免通信,即便需要通信,也应该选择进程安全的工具来避免加锁带来的问题。

以后我们会尝试使用数据库来解决现在进程之间的数据共享问题。

Manager模块介绍

  1. 进程间数据是独立的,可以借助于队列或管道实现通信,二者都是基于消息传递的
  2. 虽然进程间数据独立,但可以通过Manager实现数据共享,事实上Manager的功能远不止于此
  3. A manager object returned by Manager() controls a server process which holds Python objects and allows other processes to manipulate them using proxies.
  4. A manager returned by Manager() will support types list, dict, Namespace, Lock, RLock, Semaphore, BoundedSemaphore, Condition, Event, Barrier, Queue, Value and Array.

Manager 是一种较为高级的多进程通信方式,它能支持 Python 支持的的任何数据结构。

它的原理是:先启动一个 ManagerServer 进程,这个进程是阻塞的,它监听一个 socket,然后其他进程(ManagerClient)通过 socket 来连接到 ManagerServer,实现通信。

  1. from multiprocessing import Manager,Process
  2. def func(dic):
  3. print(dic)
  4. if __name__ == '__main__':
  5. m = Manager() # 创建一个server进程
  6. dic = m.dict({'count':100}) #这是一个特殊的字典
  7. p = Process(target=func,args=[dic,])
  8. p.start()
  9. p.join()

执行输出:{‘count’: 100}

修改字典的值

  1. from multiprocessing import Manager,Process
  2. def func(dic):
  3. dic['count'] = dic['count'] -1
  4. print(dic)
  5. if __name__ == '__main__':
  6. m = Manager() # 创建一个server进程
  7. dic = m.dict({'count':100}) #这是一个特殊的字典
  8. p = Process(target=func,args=[dic,])
  9. p.start()
  10. p.join()

输出:{‘count’: 99}

循环修改

  1. from multiprocessing import Manager,Process
  2. def func(dic):
  3. dic['count'] = dic['count'] -1 # 每次减1
  4. if __name__ == '__main__':
  5. m = Manager() # 创建一个server进程
  6. dic = m.dict({'count':100}) #这是一个特殊的字典
  7. p_lst = [] # 定义一个空列表
  8. for i in range(100): # 启动100个进程
  9. p = Process(target=func,args=[dic,])
  10. p_lst.append(p) # 进程追加到列表中
  11. p.start() # 启动进程
  12. for p in p_lst:p.join() # 等待100个进程全部结束
  13. print(dic) # 打印dic的值

重复执行 5 次,输出

{‘count’: 0}或者{‘count’: 1}或者{‘count’: 3}

发现数据已经出现错乱了,为什么呢?因为同一个时间内有多个进程操作 dic,就会发生数据错乱

为了解决这个问题,需要加锁

  1. from multiprocessing import Manager,Process,Lock
  2. def func(dic,lock):
  3. lock.acquire() #取得锁
  4. dic['count'] = dic['count'] -1 # 每次减1
  5. lock.release() #释放锁
  6. if __name__ == '__main__':
  7. m = Manager() # 创建一个server进程
  8. lock = Lock() #创建锁
  9. dic = m.dict({'count':100}) #这是一个特殊的字典
  10. p_lst = [] # 定义一个空列表
  11. for i in range(100): # 启动100个进程
  12. p = Process(target=func,args=[dic,lock])
  13. p_lst.append(p) # 进程追加到列表中
  14. p.start() # 启动进程
  15. for p in p_lst:p.join() # 等待100个进程全部结束
  16. print(dic) # 打印dic的值

重复执行 5 次,输出结果为:

{‘count’: 0}

另外一种写法,使用上下文管理

  1. from multiprocessing import Manager,Process,Lock
  2. def func(dic,lock):
  3. with lock: # 上下文管理 :必须有一个开始动作 和 一个结束动作的时候
  4. dic['count'] = dic['count'] -1 # 每次减1
  5. if __name__ == '__main__':
  6. m = Manager() # 创建一个server进程
  7. lock = Lock() #创建锁
  8. dic = m.dict({'count':100}) #这是一个特殊的字典
  9. p_lst = [] # 定义一个空列表
  10. for i in range(100): # 启动100个进程
  11. p = Process(target=func,args=[dic,lock])
  12. p_lst.append(p) # 进程追加到列表中
  13. p.start() # 启动进程
  14. for p in p_lst:p.join() # 等待100个进程全部结束
  15. print(dic) # 打印dic的值

重复执行,效果同上。

之前学到的文件管理,有用到上下文管理。这里也可以使用上下文管理。有 2 个必要条件

  1. 提供了 with 方法。

  2. 必须有一个开始和结束动作。

这里的开始和结束动作,分别指的是 acquire 和 release

同一台机器上 : 使用 Queue

在不同台机器上 :使用消息中间件

进程池和 multiprocess.Pool 模块

进程池

为什么要有进程池?进程池的概念。

在程序实际处理问题过程中,忙时会有成千上万的任务需要被执行,闲时可能只有零星任务。那么在成千上万个任务需要被执行的时候,我们就需要去创建成千上万个进程么?首先,创建进程需要消耗时间,销毁进程也需要消耗时间。第二即便开启了成千上万的进程,操作系统也不能让他们同时执行,这样反而会影响程序的效率。因此我们不能无限制的根据任务开启或者结束进程。那么我们要怎么做呢?

在这里,要给大家介绍一个进程池的概念,定义一个池子,在里面放上固定数量的进程,有需求来了,就拿一个池中的进程来处理任务,等到处理完毕,进程并不关闭,而是将进程再放回进程池中继续等待任务。如果有很多任务需要执行,池中的进程数量不够,任务就要等待之前的进程执行任务完毕归来,拿到空闲进程才能继续执行。也就是说,池中进程的数量是固定的,那么同一时间最多有固定数量的进程在运行。这样不会增加操作系统的调度难度,还节省了开闭进程的时间,也一定程度上能够实现并发效果。

multiprocess.Pool 模块

概念介绍

  1. Pool([numprocess [,initializer [, initargs]]]):创建进程池

进程池,是很重要的知识点

参数介绍

  1. 1 numprocess:要创建的进程数,如果省略,将默认使用cpu_count()的值
  2. 2 initializer:是每个工作进程启动时要执行的可调用对象,默认为None
  3. 3 initargs:是要传给initializer的参数组

主要方法

  1. p.apply(func [, args [, kwargs]]):在一个池工作进程中执行func(*args,**kwargs),然后返回结果。
  2. '''需要强调的是:此操作并不会在所有池工作进程中并执行func函数。如果要通过不同参数并发地执行func函数,必须从不同线程调用p.apply()函数或者使用p.apply_async()'''
  3. p.apply_async(func [, args [, kwargs]]):在一个池工作进程中执行func(*args,**kwargs),然后返回结果。
  4. '''此方法的结果是AsyncResult类的实例,callback是可调用对象,接收输入参数。当func的结果变为可用时,将理解传递给callback。callback禁止执行任何阻塞操作,否则将接收其他异步操作中的结果。'''
  5. p.close():关闭进程池,防止进一步操作。如果所有操作持续挂起,它们将在工作进程终止前完成
  6. P.jion():等待所有工作进程退出。此方法只能在close()或teminate()之后调用

其他方法了解

  1. 方法apply_async()和map_async()的返回值是AsyncResul的实例obj。实例具有以下方法
  2. obj.get():返回结果,如果有必要则等待结果到达。timeout是可选的。如果在指定时间内还没有到达,将引发一场。如果远程操作中引发了异常,它将在调用此方法时再次被引发。
  3. obj.ready():如果调用完成,返回True
  4. obj.successful():如果调用完成且没有引发异常,返回True,如果在结果就绪之前调用此方法,引发异常
  5. obj.wait([timeout]):等待结果变为可用。
  6. obj.terminate():立即终止所有工作进程,同时不执行任何清理或结束任何挂起工作。如果p被垃圾回收,将自动调用此函数

代码实例

import time

from multiprocessing import Pool

def fc(i):

  1. time.sleep(0.5)
  2. print('func%s'%i)

if name == ‘main‘:

  1. p = Pool(5)
  2. p.apply(func=fc,args=(1,))

执行输出:func1

import time

from multiprocessing import Pool

def fc(i):

  1. time.sleep(0.5)
  2. print('func%s'%i)

if name == ‘main‘:

  1. p = Pool(5)
  2. for i in range(5):
  3. p.apply(func=fc,args=(1,)) # 同步调用
  4. #ret = p.apply_async(func=fc, args=(1,)) # 异步调用

执行输出:

func1

func1

func1

func1

func1

import time

import random

from multiprocessing import Pool

def fc(i):

  1. print('func%s' % i)
  2. time.sleep(random.randint(1, 3))
  3. return i ** 2

if name == ‘main‘:

  1. p = Pool(5) # 创建拥有5个进程数量的进程池
  2. ret_1 = []
  3. for i in range(5):
  4. #p.apply(func=fc,args=(1,)) # 同步调用
  5. ret = p.apply_async(func=fc, args=(i,)) # 异步调用
  6. ret_1.append(ret)
  7. for ret in ret_1:print(ret.get()) # 打印返回结果

执行输出:

func0

func1

func2

func3

func4

0

1

4

9

16

后面的结果都是 i 的平方值

进程池的同步调用

import os,time

from multiprocessing import Pool

def work(n):

  1. print('%s run' %os.getpid())
  2. time.sleep(3)
  3. return n**2

if name == ‘main‘:

  1. p=Pool(3) #进程池中从无到有创建三个进程,以后一直是这三个进程在执行任务
  2. res_l=[]
  3. for i in range(10):
  4. res=p.apply(work,args=(i,)) # 同步调用,直到本次任务执行完毕拿到res,等待任务work执行的过程中可能有阻塞也可能没有阻塞
  5. # 但不管该任务是否存在阻塞,同步调用都会在原地等着
  6. print(res

进程池的异步调用

import os

import time

import random

from multiprocessing import Pool

def work(n):

  1. print('%s run' %os.getpid())
  2. time.sleep(random.random())
  3. return n**2

if name == ‘main‘:

  1. p=Pool(3) #进程池中从无到有创建三个进程,以后一直是这三个进程在执行任务
  2. res_l=[]
  3. for i in range(10):
  4. res=p.apply_async(work,args=(i,)) # 异步运行,根据进程池中有的进程数,每次最多3个子进程在异步执行
  5. # 返回结果之后,将结果放入列表,归还进程,之后再执行新的任务
  6. # 需要注意的是,进程池中的三个进程不会同时开启或者同时结束
  7. # 而是执行完一个就释放一个进程,这个进程就去接收新的任务。
  8. res_l.append(res)
  9. # 异步apply_async用法:如果使用异步提交的任务,主进程需要使用jion,等待进程池内任务都处理完,然后可以用get收集结果
  10. # 否则,主进程结束,进程池可能还没来得及执行,也就跟着一起结束了
  11. p.close()
  12. p.join()
  13. for res in res_l:
  14. print(res.get()) #使用get来获取apply_aync的结果,如果是apply,则没有get方法,因为apply是同步执行,立刻获取结果,也根本

练习

server:进程池版socket并发聊天

Pool内的进程数默认是cpu核数,假设为4(查看方法os.cpu_count())

开启6个客户端,会发现2个客户端处于等待状态

在每个进程内查看pid,会发现pid使用为4个,即多个客户端公用4个进程

from socket import *

from multiprocessing import Pool

import os

server=socket(AF_INET,SOCK_STREAM)

server.setsockopt(SOL_SOCKET,SO_REUSEADDR,1)

server.bind((‘127.0.0.1’,8080))

server.listen(5)

def talk(conn):

  1. print('进程pid: %s' %os.getpid())
  2. while True:
  3. try:
  4. msg=conn.recv(1024)
  5. if not msg:break
  6. conn.send(msg.upper())
  7. except Exception:
  8. break

if name == ‘main‘:

  1. p=Pool(4)
  2. while True:
  3. conn,*_=server.accept()
  4. p.apply_async(talk,args=(conn,))
  5. # p.apply(talk,args=(conn,client_addr)) #同步的话,则同一时间只有一个客

client

from socket import *

client=socket(AF_INET,SOCK_STREAM)

client.connect((‘127.0.0.1’,8080))

while True:

  1. msg=input('>>: ').strip()
  2. if not msg:continue
  3. client.send(msg.encode('utf-8'))
  4. msg=client.recv(1024)
  5. print(msg.decode('utf-8'

发现:并发开启多个客户端,服务端同一时间只有 4 个不同的 pid,只能结束一个客户端,另外一个客户端才会进来.

信号量和进程池的区别:

  1. ![](https://cdn.nlark.com/yuque/0/2020/png/1484428/1597999264004-7c1541b6-8931-411f-8972-99cbb6a86345.png)

回调函数

需要回调函数的场景:进程池中任何一个任务一旦处理完了,就立即告知主进程:我好了额,你可以处理我的结果了。主进程则调用一个函数去处理该结果,该函数即回调函数

我们可以把耗时间(阻塞)的任务放到进程池中,然后指定回调函数(主进程负责执行),这样主进程在执行回调函数时就省去了I/O的过程,直接拿到的是任务的结果。

进程池的数量一般为 CPU 的个数加 1

简单爬虫例子:

import os

from urllib.request import urlopen

from multiprocessing import Pool

def get_url(url):

  1. print('-->',url,os.getpid(),'get_url进程')
  2. ret = urlopen(url) # 打开url
  3. content = ret.read() # 读取网页内容
  4. return content

def call(url): # 回调函数

  1. #分析
  2. print(url,os.getpid(),'回调函数')

if name == ‘main‘:

  1. print(os.getpid(),'主进程') # 主进程id
  2. l = [
  3. 'http://www.baidu.com',
  4. 'http://www.sina.com',
  5. 'http://www.sohu.com',
  6. 'http://www.sogou.com',
  7. 'http://www.qq.com',
  8. 'http://www.bilibili.com',
  9. ]
  10. p = Pool(5)
  11. ret_l = []
  12. for url in l:
  13. ret = p.apply_async(func=get_url,args=[url,],callback=call) # 异步
  14. ret_l.append(ret) # 将进程追加到列表中
  15. for ret in ret_l:ret.get() # 获取进程返回值

执行输出:

/www.sohu.com/a/231538578_115362” target=”_blank”

输出了一堆内容,但是 get_url 函数并没有 print,那么由谁输出的呢?

是由 call 打印的

回调函数

在进程池中,起了一个任务,这个任务对应的函数在执行完毕之后

的返回值会自动作为参数返回给回调函数

回调函数就根据返回值再进行相应的处理

回调函数 是在主进程执行的

看下图

  1. ![](https://cdn.nlark.com/yuque/0/2020/png/1484428/1597999263968-132ec858-8421-47bb-b427-a6d0114b5a41.png)

有 6 个 url,每一个 url 的响应时间是不一样的。假设第一个 url 访问很慢,那么必须等待任务结束,才能执行分析函数。如果使用回调函数,谁最快范围,优先执行回调函数。那么最慢的 url,最后执行。

这样效率就提升了很多。

回调函数是瞬间执行的,网络延时才是最耗最长的。

回调函数是主 进程执行的,不是子进程执行的。

怎么证明呢?修改 get_url 的 return 值

import os

from urllib.request import urlopen

from multiprocessing import Pool

def get_url(url):

  1. print('-->',url,os.getpid(),'get_url进程')
  2. ret = urlopen(url) # 打开url
  3. content = ret.read() # 读取网页内容
  4. return url

def call(url): # 回调函数

  1. #分析
  2. print(url,os.getpid(),'回调函数')

if name == ‘main‘:

  1. print(os.getpid(),'主进程') # 主进程id
  2. l = [
  3. 'http://www.baidu.com',
  4. 'http://www.sina.com',
  5. 'http://www.sohu.com',
  6. 'http://www.sogou.com',
  7. 'http://www.qq.com',
  8. 'http://www.bilibili.com',
  9. ]
  10. p = Pool(5)
  11. ret_l = []
  12. for url in l:
  13. ret = p.apply_async(func=get_url,args=[url,],callback=call) # 异步
  14. ret_l.append(ret) # 将进程追加到列表中
  15. for ret in ret_l:ret.get() # 获取进程返回值

执行输出:

13240 主进程

—> http://www.baidu.com 4164 get_url 进程

—> http://www.sina.com 9144 get_url 进程

—> http://www.sohu.com 17984 get_url 进程

—> http://www.sogou.com 10348 get_url 进程

—> http://www.qq.com 7180 get_url 进程

—> http://www.bilibili.com 4164 get_url 进程

http://www.baidu.com 13240 回调函数

http://www.sohu.com 13240 回调函数

http://www.qq.com 13240 回调函数

http://www.sogou.com 13240 回调函数

http://www.bilibili.com 13240 回调函数

http://www.sina.com 13240 回调函数

执行回调函数的进程 id 都是 13240,这个进程正好是主进程。

使用多进程请求多个url来减少网络等待浪费的时间

from multiprocessing import Pool

import requests

import json

import os

def get_page(url):

  1. print('<进程%s> get %s' %(os.getpid(),url))
  2. respone=requests.get(url)
  3. if respone.status_code == 200:
  4. return {'url':url,'text':respone.text}

def pasrse_page(res):

  1. print('<进程%s> parse %s' %(os.getpid(),res['url']))
  2. parse_res='url:<%s> size:[%s]\n' %(res['url'],len(res['text']))
  3. with open('db.txt','a') as f:
  4. f.write(parse_res)

if name == ‘main‘:

  1. urls=[
  2. 'https://www.baidu.com',
  3. 'https://www.python.org',
  4. 'https://www.openstack.org',
  5. 'https://help.github.com/',
  6. 'http://www.sina.com.cn/'
  7. ]
  8. p=Pool(3)
  9. res_l=[]
  10. for url in urls:
  11. res=p.apply_async(get_page,args=(url,),callback=pasrse_page)
  12. res_l.append(res)
  13. p.close()
  14. p.join()
  15. print([res.get() for res in res_l]) #拿到的是get_page的结果,其实完全没必要拿该结果,该结果已经传给回调函数处理了

‘’’

打印结果:

<进程3388> get https://www.baidu.com

<进程3389> get https://www.python.org

<进程3390> get https://www.openstack.org

<进程3388> get https://help.github.com/

<进程3387> parse https://www.baidu.com

<进程3389> get http://www.sina.com.cn/

<进程3387> parse https://www.python.org

<进程3387> parse https://help.github.com/

<进程3387> parse http://www.sina.com.cn/

<进程3387> parse https://www.openstack.org

[{‘url’: ‘https://www.baidu.com‘, ‘text’: ‘<!DOCTYPE html>\r\n.

爬虫实例

import re

from urllib.request import urlopen

from multiprocessing import Pool

def get_page(url,pattern):

  1. response=urlopen(url).read().decode('utf-8')
  2. return pattern,response

def parse_page(info):

  1. pattern,page_content=info
  2. res=re.findall(pattern,page_content)
  3. for item in res:
  4. dic={
  5. 'index':item[0].strip(),
  6. 'title':item[1].strip(),
  7. 'actor':item[2].strip(),
  8. 'time':item[3].strip(),
  9. }
  10. print(dic)

if name == ‘main‘:

  1. regex = r'<dd>.*?<.*?class="board-index.*?>(\d+)</i>.*?title="(.*?)".*?class="movie-item-info".*?<p class="star">(.*?)</p>.*?<p class="releasetime">(.*?)</p>'
  2. pattern1=re.compile(regex,re.S)
  3. url_dic={
  4. 'http://maoyan.com/board/7':pattern1,
  5. }
  6. p=Pool()
  7. res_l=[]
  8. for url,pattern in url_dic.items():
  9. res=p.apply_async(get_page,args=(url,pattern),callback=parse_page)
  10. res_l.append(res)
  11. for i in res_l:

如果在主进程中等待进程池中所有任务都执行完毕后,再统一处理结果,则无需回调函数

无需回调函数

from multiprocessing import Pool

import time,random,os

def work(n):

  1. time.sleep(1)
  2. return n**2

if name == ‘main‘:

  1. p=Pool()
  2. res_l=[]
  3. for i in range(10):
  4. res=p.apply_async(work,args=(i,))
  5. res_l.append(res)
  6. p.close()
  7. p.join() #等待进程池中所有进程执行完毕
  8. nums=[]
  9. for res in res_l:
  10. nums.append(res.get()) #拿到所有结果
  11. print(nums) #主进程拿到所有的处理结果,可以在主进程中进行统一进

进程池的其他实现方式:https://docs.python.org/dev/library/concurrent.futures.html

参考资料

http://www.cnblogs.com/linhaifeng/articles/6817679.html

https://www.jianshu.com/p/1200fd49b583

https://www.jianshu.com/p/aed6067eeac

明日默写:

import time

import random

from multiprocessing import Process,Queue

def consumer(q,name):

  1. while True:
  2. food = q.get()
  3. if food == 'done':break
  4. time.sleep(random.random())
  5. print('%s吃了%s'%(name,food))

def producer(q,name,food):

  1. for i in range(10):
  2. time.sleep(random.random())
  3. print('%s生产了%s%s'%(name,food,i))
  4. q.put('%s%s'%(food,i))

if name == ‘main‘:

  1. q = Queue()
  2. p1 = Process(target=producer,args=[q,'Egon','泔水'])
  3. p2 = Process(target=producer,args=[q,'Yuan','骨头鱼刺'])
  4. p1.start()
  5. p2.start()
  6. Process(target=consumer,args=[q,'alex']).start()
  7. Process(target=consumer,args=[q,'wusir']).start()
  8. p1.join()
  9. p2.join()
  10. q.put('done')
  11. q.put('done')