46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
import timeit
|
||
|
|
from time import sleep
|
||
|
|
from random import randint
|
||
|
|
from concurrent.futures import ThreadPoolExecutor, Future
|
||
|
|
|
||
|
|
results: list[int] = []
|
||
|
|
|
||
|
|
def add_one(number: int) -> int:
|
||
|
|
sleep(randint(0,2))
|
||
|
|
result = number + 1
|
||
|
|
return result
|
||
|
|
|
||
|
|
|
||
|
|
def aggregate_results(future: Future):
|
||
|
|
results.append(future.result())
|
||
|
|
|
||
|
|
|
||
|
|
def first_method():
|
||
|
|
with ThreadPoolExecutor(32) as executor:
|
||
|
|
futures = [executor.submit(add_one, number) for number in range(10)]
|
||
|
|
for future in futures:
|
||
|
|
future.add_done_callback(aggregate_results)
|
||
|
|
|
||
|
|
|
||
|
|
def second_method():
|
||
|
|
futures = []
|
||
|
|
with ThreadPoolExecutor(32) as executor:
|
||
|
|
for number in range(10):
|
||
|
|
futures.append(executor.submit(add_one, number))
|
||
|
|
futures[-1].add_done_callback(aggregate_results)
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
print(timeit.timeit(first_method, number=1))
|
||
|
|
print(results)
|
||
|
|
print("All done!\n")
|
||
|
|
results.clear()
|
||
|
|
print(timeit.timeit(second_method, number=1))
|
||
|
|
print(results)
|
||
|
|
print("All done!")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|