Work Manager Replication thows on Replication complete (Beryllium)

Calling `cancel()` on the context to end the flow, causes `runBlocking` the throw an interrupted exception, and thus the failure of the worker.

Moving most of the logic from the `collect` call, to a `map` and introducing a `takeWhile` is more elegant, and does not generate the exception