Python futures.wait() 函数详解：异步任务等待利器

在Python中，futures.wait()函数是一个用于等待一组Future对象完成的函数。\n\nFuture对象表示一个异步操作的结果。当一个Future对象被创建时，它会被放入一个待完成的任务列表中。wait()函数用于等待这个任务列表中的所有Future对象完成。\n\nfutures.wait(fs, timeout=None, return_when=ALL_COMPLETED)函数接受三个参数：\n\n- fs：一个包含Future对象的可迭代对象。这些Future对象代表了要等待的异步任务。\n- timeout：等待的最长时间（以秒为单位）。如果设置了超时时间，那么wait()函数将在超时时间到达时返回。\n- return_when：确定wait()函数在返回之前等待的条件。它可以取以下值：\n - FIRST_COMPLETED：只要有一个Future对象完成，wait()函数就会返回。\n - FIRST_EXCEPTION：只要有一个Future对象抛出异常，wait()函数就会返回。\n - ALL_COMPLETED：所有的Future对象都完成时，wait()函数才会返回。\n\nwait()函数会阻塞当前线程，直到满足指定的等待条件。一旦满足条件，它将返回一个具有两个元素的元组。第一个元素是已完成的Future对象的集合，第二个元素是未完成的Future对象的集合。\n\n以下是一个使用futures.wait()函数的示例：\n\npython\nimport concurrent.futures\n\ndef worker(n):\n return n * n\n\nwith concurrent.futures.ThreadPoolExecutor() as executor:\n futures = [executor.submit(worker, i) for i in range(5)]\n done, not_done = concurrent.futures.wait(futures, return_when=concurrent.futures.ALL_COMPLETED)\n \n for future in done:\n print(future.result())\n\n\n在上面的示例中，我们使用ThreadPoolExecutor创建了一个线程池，并使用executor.submit()方法提交了5个任务。然后，我们使用wait()函数等待所有任务完成，并通过done集合获取已完成的Future对象。最后，我们通过result()方法获取每个已完成Future对象的结果并打印出来。\n\n注意，futures.wait()函数是一个阻塞函数，它会阻塞当前线程直到满足指定的等待条件。如果你想非阻塞地等待一组Future对象完成，可以使用concurrent.futures.as_completed()函数配合yield from语法来实现。