CodePattern/B 高性能模式/010 多进程.py

import util
import multiprocessing


@util.measure_performance
def compute_task():
    processes = [
        multiprocessing.Process(target=util.compute_task) for _ in range(5)
    ]
    for process in processes:
        process.start()
    for process in processes:
        process.join()


@util.measure_performance
def io_task():
    processes = [
        multiprocessing.Process(target=util.fetch_url, args=(url, ))
        for url in util.urls
    ]
    for process in processes:
        process.start()
    for process in processes:
        process.join()


def main():
    compute_task()
    io_task()


if __name__ == "__main__":
    main()