TimeChunkedFlow.kt.diff

launch {
    while (true) {
+       try {
            delay(duration)
            mutex.withLock {
                // If the upstream flow has been completed and there are no values
                // pending to emit in the collector, just break this loop.
                if (isFlowCompleted && values.isEmpty()) {
                    return@launch
                }
                collector.emit(values.toList())
                values.clear()
            }
+       } catch (e: CancellationException) {
+           collector.emit(values.toList())
+           return@launch
+       }
    }
}