fun <T> Flow<T>.timeout(durationInMillis: Long): Flow<T> =
    flow {
        coroutineScope {
            val outerScope = this
            launch {
                try {
                    delay(durationInMillis)
                    outerScope.cancel()
                } catch (e: CancellationException) {
                    outerScope.cancel(e) // cancel outer scope on cancellation exception, too
                }
            }
            collect { emit(it) }
        }
    }