历程间通讯

表明

•         设置启动方法：
  
  
  
  • 使用 set_start_method 来指定历程启动的方法。这行代码应该尽大概早地实行，而且只能被调用一次。对于 macOS 和 Linux 体系，默认是 fork，而对于 Windows 则默认是 spawn。这里我们实验设置为 fork，但假如有辩说则跳过（比方当脚本已经被导入时）。
    
  
  
•         界说工作者函数：
  
  
  
  • worker 函数代表了每个子历程要实行的任务。它起首实验从队列中获取一条消息，然后模拟一些工作（在这里是简朴的 time.sleep(1)），末了将一条新消息放入队列。
    
  
  
•         主步伐逻辑：
  
  
  
  • 在主步伐中，我们起首创建了一个 Queue 实例用于历程间的通讯。
    
  • 接着，我们向队列中放入了一条初始消息，以供第一个子历程读取。
    
  • 然后，我们创建了多少个子历程，并启动它们。每个子历程都将调用 worker 函数。
    
  • 使用 join() 方法等待全部子历程完成。
    
  • 末了，主步伐遍历队列中的全部消息，并打印出来，以展示全部子历程的工作结果。
    
  
  

这种布局使得代码更加模块化和易读，同时也明确了各个部分的功能。

2. import time
3. from multiprocessing import Process, Queue, set_start_method
5. def worker(queue, worker_id):
6.     """
7.     工作进程的主函数。
8.    
9.     参数:
10.     - queue: 进程间通信的队列。
11.     - worker_id: 工作进程的标识符。
12.    
13.     返回值: 无
14.     """
15.     # 尝试从队列中获取一条消息（如果有的话）
16.     if not queue.empty():
17.         msg = queue.get()
18.         print(f"Worker {worker_id}: Received message '{msg}'")
20.     # 模拟工作负载
21.     time.sleep(1)
23.     # 向队列中放入一条消息
24.     response_msg = f"Message from Worker {worker_id}"
25.     queue.put(response_msg)
26.     print(f"Worker {worker_id}: Sent message '{response_msg}'")
28. if __name__ == "__main__":
29.     # 设置启动方法为 "fork" 或 "spawn"（根据操作系统选择），并且必须在创建任何进程前调用
30.     try:
31.         set_start_method('fork')  # 对于 Unix/Linux/macOS 系统
32.     except RuntimeError:
33.         pass  # 如果已经设置了启动方法，则忽略错误
35.     # 创建一个队列对象
36.     queue = Queue()
38.     # 初始化进程列表
39.     processes = []
40.     num_workers = 5  # 定义多少个子进程
42.     # 向队列添加初始消息
43.     initial_message = "Start Message"
44.     queue.put(initial_message)
45.     print(f"Main: Sent initial message '{initial_message}'")
47.     # 创建并启动子进程
48.     for i in range(num_workers):
49.         p = Process(target=worker, args=(queue, i))
50.         p.start()
51.         processes.append(p)
53.     # 等待所有子进程完成
54.     for p in processes:
55.         p.join()
57.     print("Main: All workers have finished.")
59.     print("############# Final Messages #############")
60.     # 收集并打印队列中的剩余消息
61.     while not queue.empty():
62.         print(f"Main: Received final message '{queue.get()}'")

复制代码

2. Main: Sent initial message 'Start Message'
3. Worker 0: Received message 'Start Message'
4. Worker 0: Sent message 'Message from Worker 0'
5. Worker 1: Sent message 'Message from Worker 1'Worker 2: Sent message 'Message from Worker 2'
7. Worker 3: Sent message 'Message from Worker 3'
8. Worker 4: Sent message 'Message from Worker 4'
9. Main: All workers have finished.
10. ############# Final Messages #############
11. Main: Received final message 'Message from Worker 0'
12. Main: Received final message 'Message from Worker 1'
13. Main: Received final message 'Message from Worker 2'
14. Main: Received final message 'Message from Worker 3'
15. Main: Received final message 'Message from Worker 4'

复制代码