Function as_completed

src/python/cli/src/mesos/futures.py:129–161

(futures, timeout=None)

    def as_completed(futures, timeout=None):
        # Record the start time in order to determine the remaining
        # timeout as futures are completed.
        start = time.time()

        # Use a queue to collect the done futures.
        queue = Queue()

        # Define a helper for the future "done callback".
        def done(future):
            queue.put(future)

        # Add a "done callback" for each future.
        for future in futures:
            future.add_done_callback(done)

        # Helper to determine the remaining timeout.
        def remaining():
            if timeout is None:
                return None
            end = start + timeout
            remaining = end - time.time()
            return remaining if remaining >= 0 else 0

        # Now wait until all the futures have completed or we timeout.
        finished = 0
        while finished < len(futures):
            try:
                yield queue.get(timeout=remaining())
            except Empty:
                raise TimeoutError()
            else:
                finished += 1


class ThreadingExecutor(Executor):