进程间的通信

Posted 2021-02-09 selling-fish-bears

 

使用 multiprocessing 里的 Queue()

import multiprocessing


def download_from_web(q):
    """下载数据"""

    # 模拟从网上下载的数据
    data = [11, 22, 33, 44]

    # 向队列中写入数据
    for temp in data:
        q.put(temp)

    print("下载器已经下载完了数据并且存入到队列中")


def analysis_data(q):
    """数据处理"""

    waiting_analysis_data = list()

    while True:
        # 从队列中获取数据
        data = q.get()
        waiting_analysis_data.append(data)

        # 判断队列是否为空
        if q.empty():
            break

    # 模拟数据处理(只有单核的话只打印1个数据)
    print(waiting_analysis_data)


def main():
    # 1.创建一个队列
    q = multiprocessing.Queue()
    # 2.创建多个进程，将队列的引用当做实参进行传递到里面
    p1 = multiprocessing.Process(target=download_from_web, args=(q,))
    p2 = multiprocessing.Process(target=analysis_data, args=(q,))
    p1.start()
    p2.start()


if __name__ == "__main__":
    main()