"""多線程多進(jìn)程模擬執(zhí)行效率"""

from multiprocessing import Pool
from threading import Thread
import time, math

def simulation_IO(a):
 """模擬IO操作"""
 time.sleep(3)

def simulation_compute(a):
 """模擬計算密集型任務(wù)"""
 for i in range(int(1e7)):
  math.sin(40) + math.cos(40)
 return

def normal_func(func):
 """普通方法執(zhí)行效率"""
 for i in range(6):
  func(i)
 return

def mp(func):
 """進(jìn)程池中的map方法"""
 with Pool(processes=6) as p:
  res = p.map(func, list(range(6)))
 return

def asy(func):
 """進(jìn)程池中的異步執(zhí)行"""
 with Pool(processes=6) as p:
  result = []
  for j in range(6):
a = p.apply_async(func, args=(j, ))
result.append(a)
  res = [j.get() for j in result]

def thread(func):
 """多線程方法"""
 threads = []
 for j in range(6):
  t = Thread(target=func, args=(j, ))
  threads.append(t)
  t.start()
 for t in threads:
  t.join()

def showtime(f, func, name):
 """
 計算并展示函數(shù)的運行時間
 :param f: 多進(jìn)程和多線程的方法
 :param func: 多進(jìn)程和多線程方法中需要傳入的函數(shù)
 :param name: 方法的名字
 :return:
 """
 start_time = time.time()
 f(func)
 print(f"{name} time: {time.time() - start_time:.4f}s")

def main(func):
 """
 運行程序的主函數(shù)
 :param func: 傳入需要計算時間的函數(shù)名
 """
 showtime(normal_func, func, "normal")
 print()
 print("------ 多進(jìn)程 ------")
 showtime(mp, func, "map")
 showtime(asy, func, "async")
 print()
 print("----- 多線程 -----")
 showtime(thread, func, "thread")

if __name__ == "__main__":
 print("------------ 計算密集型 ------------")
 func = simulation_compute
 main(func)
 print()
 print()
 print()
 print("------------ IO 密集型 ------------")
 func = simulation_IO
 main(func)