1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68
|
/*
* Copyright 2016-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license.
*/
package kotlinx.coroutines.channels
import kotlinx.coroutines.selects.*
enum class TestChannelKind {
RENDEZVOUS {
override fun create(): Channel<Int> = Channel(Channel.RENDEZVOUS)
override fun toString(): String = "RendezvousChannel"
},
ARRAY_1 {
override fun create(): Channel<Int> = Channel(1)
override fun toString(): String = "ArrayChannel(1)"
},
ARRAY_10 {
override fun create(): Channel<Int> = Channel(10)
override fun toString(): String = "ArrayChannel(10)"
},
LINKED_LIST {
override fun create(): Channel<Int> = Channel(Channel.UNLIMITED)
override fun toString(): String = "LinkedListChannel"
},
CONFLATED {
override fun create(): Channel<Int> = Channel(Channel.CONFLATED)
override fun toString(): String = "ConflatedChannel"
override val isConflated: Boolean get() = true
},
ARRAY_BROADCAST_1 {
override fun create(): Channel<Int> = ChannelViaBroadcast(BroadcastChannel(1))
override fun toString(): String = "ArrayBroadcastChannel(1)"
},
ARRAY_BROADCAST_10 {
override fun create(): Channel<Int> = ChannelViaBroadcast(BroadcastChannel(10))
override fun toString(): String = "ArrayBroadcastChannel(10)"
},
CONFLATED_BROADCAST {
override fun create(): Channel<Int> = ChannelViaBroadcast(ConflatedBroadcastChannel<Int>())
override fun toString(): String = "ConflatedBroadcastChannel"
override val isConflated: Boolean get() = true
}
;
abstract fun create(): Channel<Int>
open val isConflated: Boolean get() = false
}
private class ChannelViaBroadcast<E>(
private val broadcast: BroadcastChannel<E>
): Channel<E>, SendChannel<E> by broadcast {
val sub = broadcast.openSubscription()
override val isClosedForReceive: Boolean get() = sub.isClosedForReceive
override val isEmpty: Boolean get() = sub.isEmpty
override suspend fun receive(): E = sub.receive()
override suspend fun receiveOrNull(): E? = sub.receiveOrNull()
override fun poll(): E? = sub.poll()
override fun iterator(): ChannelIterator<E> = sub.iterator()
override fun cancel(): Unit = sub.cancel()
override fun cancel(cause: Throwable?): Boolean = sub.cancel(cause)
override val onReceive: SelectClause1<E>
get() = sub.onReceive
override val onReceiveOrNull: SelectClause1<E?>
get() = sub.onReceiveOrNull
}
|