fix(nests-reconnect): break the StateFlow collect + survive session swaps (T4 #2)
Two related bugs in connectReconnectingNestsListener:
1. Orchestrator never advanced past the first Failed. The reconnect
loop used `listener.state.collect { ... return@collect }` to
"exit" on a terminal state, but `return@collect` only returns
from the lambda — a StateFlow's collect keeps running. As a
result, after the first transport failure the orchestrator
re-entered the lambda for every subsequent emission and never
reached the outer `delay(delayMs)` / openOnce() that opens the
next session. Switch to `onEach { mirror } + first { terminal }`
so the upstream completes deterministically.
2. SubscribeHandle stopped emitting after a reconnect (the
long-deferred Tier-4 follow-up). Reissue the subscription on
every activeListener swap by feeding a wrapper-side
MutableSharedFlow from a `collectLatest` pump that re-calls
`listener.subscribeSpeaker(...)` against each fresh session.
The buffer (64 frames ≈ 1.3 s of Opus) covers a typical
re-handshake without dropping speech. Unsubscribing the
logical handle cancels the pump and forwards a best-effort
unsubscribe to the live underlying handle.
Tests: ReconnectingNestsListenerTest covers both happy-path
session swap (frames keep arriving) and the unsubscribe-before-swap
path. Uses real coroutines + Dispatchers.Default rather than
runTest because the test scheduler doesn't auto-advance the
StateFlow + delay chain reliably under UnconfinedTestDispatcher.
This commit is contained in:
Claude
+269
@@ -0,0 +1,269 @@
|
||||
/*
|
||||
* Copyright (c) 2025 Vitor Pamplona
|
||||
*
|
||||
* Permission is hereby granted, free of charge, to any person obtaining a copy of
|
||||
* this software and associated documentation files (the "Software"), to deal in
|
||||
* the Software without restriction, including without limitation the rights to use,
|
||||
* copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the
|
||||
* Software, and to permit persons to whom the Software is furnished to do so,
|
||||
* subject to the following conditions:
|
||||
*
|
||||
* The above copyright notice and this permission notice shall be included in all
|
||||
* copies or substantial portions of the Software.
|
||||
*
|
||||
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
|
||||
* FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
|
||||
* COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN
|
||||
* AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
||||
* WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||||
*/
|
||||
package com.vitorpamplona.nestsclient
|
||||
|
||||
import com.vitorpamplona.nestsclient.moq.MoqObject
|
||||
import com.vitorpamplona.nestsclient.moq.SubscribeHandle
|
||||
import com.vitorpamplona.nestsclient.moq.SubscribeOk
|
||||
import com.vitorpamplona.nestsclient.transport.WebTransportFactory
|
||||
import com.vitorpamplona.nestsclient.transport.WebTransportSession
|
||||
import com.vitorpamplona.quartz.nip01Core.crypto.KeyPair
|
||||
import com.vitorpamplona.quartz.nip01Core.signers.NostrSigner
|
||||
import com.vitorpamplona.quartz.nip01Core.signers.NostrSignerInternal
|
||||
import kotlinx.coroutines.CoroutineScope
|
||||
import kotlinx.coroutines.SupervisorJob
|
||||
import kotlinx.coroutines.async
|
||||
import kotlinx.coroutines.cancel
|
||||
import kotlinx.coroutines.flow.MutableSharedFlow
|
||||
import kotlinx.coroutines.flow.MutableStateFlow
|
||||
import kotlinx.coroutines.flow.StateFlow
|
||||
import kotlinx.coroutines.flow.asStateFlow
|
||||
import kotlinx.coroutines.flow.first
|
||||
import kotlinx.coroutines.flow.take
|
||||
import kotlinx.coroutines.flow.toList
|
||||
import kotlinx.coroutines.runBlocking
|
||||
import kotlinx.coroutines.withTimeout
|
||||
import kotlin.test.Test
|
||||
import kotlin.test.assertEquals
|
||||
import kotlin.test.assertTrue
|
||||
|
||||
class ReconnectingNestsListenerTest {
|
||||
private val room =
|
||||
NestsRoomConfig(
|
||||
authBaseUrl = "https://relay.example.com/api/v1/nests",
|
||||
endpoint = "https://relay.example.com/moq",
|
||||
hostPubkey = "0".repeat(64),
|
||||
roomId = "abc",
|
||||
)
|
||||
private val signer = NostrSignerInternal(KeyPair())
|
||||
|
||||
// The custom `connector` lambda below short-circuits the real
|
||||
// mintToken / WebTransport handshake, so these two sentinels
|
||||
// only satisfy the function signature — they're never invoked.
|
||||
private val httpClient =
|
||||
object : NestsClient {
|
||||
override suspend fun mintToken(
|
||||
room: NestsRoomConfig,
|
||||
publish: Boolean,
|
||||
signer: NostrSigner,
|
||||
): String = error("not invoked — connector overrides")
|
||||
}
|
||||
private val transport =
|
||||
object : WebTransportFactory {
|
||||
override suspend fun connect(
|
||||
authority: String,
|
||||
path: String,
|
||||
bearerToken: String?,
|
||||
): WebTransportSession = error("not invoked — connector overrides")
|
||||
}
|
||||
|
||||
/**
|
||||
* Scripted [NestsListener] that opens in [NestsListenerState.Connected]
|
||||
* by default and exposes a [MutableSharedFlow] to push frames into
|
||||
* the next [SubscribeHandle.objects] returned from [subscribeSpeaker].
|
||||
*/
|
||||
private class ScriptedListener(
|
||||
connectedRoom: NestsRoomConfig,
|
||||
moqVersion: Long = 1,
|
||||
) : NestsListener {
|
||||
private val mutableState =
|
||||
MutableStateFlow<NestsListenerState>(
|
||||
NestsListenerState.Connected(connectedRoom, moqVersion),
|
||||
)
|
||||
override val state: StateFlow<NestsListenerState> = mutableState.asStateFlow()
|
||||
val frames = MutableSharedFlow<MoqObject>(extraBufferCapacity = 16)
|
||||
|
||||
// Atomic counters — the wrapper's pump runs on Dispatchers.Default
|
||||
// while the test reads the count from the runBlocking thread, so
|
||||
// a plain `var subscribeCount` would risk stale reads.
|
||||
private val _subscribeCount =
|
||||
java.util.concurrent.atomic
|
||||
.AtomicInteger(0)
|
||||
private val _unsubscribeCount =
|
||||
java.util.concurrent.atomic
|
||||
.AtomicInteger(0)
|
||||
val subscribeCount: Int get() = _subscribeCount.get()
|
||||
val unsubscribeCount: Int get() = _unsubscribeCount.get()
|
||||
|
||||
override suspend fun subscribeSpeaker(speakerPubkeyHex: String): SubscribeHandle {
|
||||
val id = _subscribeCount.incrementAndGet().toLong()
|
||||
return SubscribeHandle(
|
||||
subscribeId = id,
|
||||
trackAlias = id,
|
||||
ok =
|
||||
SubscribeOk(
|
||||
subscribeId = id,
|
||||
expiresMs = 0L,
|
||||
groupOrder = 0x01,
|
||||
contentExists = false,
|
||||
),
|
||||
objects = frames,
|
||||
unsubscribeAction = { _unsubscribeCount.incrementAndGet() },
|
||||
)
|
||||
}
|
||||
|
||||
override suspend fun close() {
|
||||
mutableState.value = NestsListenerState.Closed
|
||||
}
|
||||
|
||||
fun fail(reason: String) {
|
||||
mutableState.value = NestsListenerState.Failed(reason)
|
||||
}
|
||||
}
|
||||
|
||||
private fun frame(payload: ByteArray): MoqObject =
|
||||
MoqObject(
|
||||
trackAlias = 1L,
|
||||
groupId = 0L,
|
||||
objectId = 0L,
|
||||
publisherPriority = 0,
|
||||
payload = payload,
|
||||
)
|
||||
|
||||
// The orchestrator + pump jobs are real coroutines on a real
|
||||
// dispatcher — runTest's virtual clock interacts badly with the
|
||||
// orchestrator's `delay(...)` reconnect loop (the test scheduler
|
||||
// doesn't auto-advance through the StateFlow + delay chain
|
||||
// under UnconfinedTestDispatcher). Use a dedicated supervisor
|
||||
// scope and real time; tests stay fast (single-digit ms) because
|
||||
// the policy's initialDelayMs is set to 1.
|
||||
@Test
|
||||
fun subscribeSpeaker_survives_session_swap() =
|
||||
runBlocking {
|
||||
// Plain SupervisorJob — NOT plus coroutineContext, otherwise
|
||||
// `scope.cancel()` in the finally block would propagate up
|
||||
// through the runBlocking parent and fail the test.
|
||||
val scope = CoroutineScope(SupervisorJob())
|
||||
try {
|
||||
val first = ScriptedListener(room)
|
||||
val second = ScriptedListener(room)
|
||||
val listenersInOrder = mutableListOf(first, second)
|
||||
|
||||
val reconnecting =
|
||||
connectReconnectingNestsListener(
|
||||
httpClient = httpClient,
|
||||
transport = transport,
|
||||
scope = scope,
|
||||
room = room,
|
||||
signer = signer,
|
||||
policy = NestsReconnectPolicy(initialDelayMs = 1L),
|
||||
connector = { listenersInOrder.removeAt(0) },
|
||||
)
|
||||
|
||||
withTimeout(5_000L) {
|
||||
reconnecting.state.first { it is NestsListenerState.Connected }
|
||||
}
|
||||
|
||||
val handle = reconnecting.subscribeSpeaker("speaker-pubkey")
|
||||
|
||||
val collected =
|
||||
scope.async {
|
||||
withTimeout(5_000L) {
|
||||
handle.objects.take(2).toList()
|
||||
}
|
||||
}
|
||||
|
||||
// Wait for the pump to actually subscribe to the first
|
||||
// session before we emit / fail. With Dispatchers.Default
|
||||
// the pumpJob's collectLatest doesn't run synchronously
|
||||
// off the launch site — yielding here lets it advance.
|
||||
withTimeout(5_000L) {
|
||||
while (first.subscribeCount == 0) kotlinx.coroutines.delay(5)
|
||||
}
|
||||
|
||||
first.frames.emit(frame(byteArrayOf(0x01)))
|
||||
|
||||
// Force a reconnect: fail the first listener, the
|
||||
// orchestrator opens the next one.
|
||||
first.fail("scripted-disconnect")
|
||||
// Both ScriptedListeners report the same `room` and
|
||||
// `negotiatedMoqVersion`, so we can't differentiate
|
||||
// first.Connected from second.Connected via state alone.
|
||||
// The actual postcondition we care about — the pump
|
||||
// re-issued the subscription against the new session —
|
||||
// is observable directly via second.subscribeCount.
|
||||
withTimeout(5_000L) {
|
||||
while (second.subscribeCount == 0) kotlinx.coroutines.delay(5)
|
||||
}
|
||||
|
||||
second.frames.emit(frame(byteArrayOf(0x02)))
|
||||
|
||||
val result = collected.await()
|
||||
assertEquals(2, result.size)
|
||||
assertEquals(0x01.toByte(), result[0].payload[0])
|
||||
assertEquals(0x02.toByte(), result[1].payload[0])
|
||||
|
||||
// Both sessions saw exactly one subscribeSpeaker call —
|
||||
// the wrapper re-issues against the new session, not the
|
||||
// old one.
|
||||
assertEquals(1, first.subscribeCount)
|
||||
assertEquals(1, second.subscribeCount)
|
||||
|
||||
handle.unsubscribe()
|
||||
assertEquals(1, second.unsubscribeCount)
|
||||
|
||||
reconnecting.close()
|
||||
} finally {
|
||||
scope.cancel()
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
fun unsubscribe_before_session_swap_releases_handle() =
|
||||
runBlocking {
|
||||
// Plain SupervisorJob — NOT plus coroutineContext, otherwise
|
||||
// `scope.cancel()` in the finally block would propagate up
|
||||
// through the runBlocking parent and fail the test.
|
||||
val scope = CoroutineScope(SupervisorJob())
|
||||
try {
|
||||
val first = ScriptedListener(room)
|
||||
val reconnecting =
|
||||
connectReconnectingNestsListener(
|
||||
httpClient = httpClient,
|
||||
transport = transport,
|
||||
scope = scope,
|
||||
room = room,
|
||||
signer = signer,
|
||||
connector = { first },
|
||||
)
|
||||
withTimeout(5_000L) {
|
||||
reconnecting.state.first { it is NestsListenerState.Connected }
|
||||
}
|
||||
|
||||
val handle = reconnecting.subscribeSpeaker("speaker-pubkey")
|
||||
|
||||
// Wait until the pump opened the underlying subscription —
|
||||
// collectLatest schedules the inner block on a child of
|
||||
// scope, so it runs after we yield.
|
||||
withTimeout(5_000L) {
|
||||
while (first.subscribeCount == 0) kotlinx.coroutines.yield()
|
||||
}
|
||||
assertTrue(first.subscribeCount >= 1, "pump should have opened the underlying sub")
|
||||
|
||||
handle.unsubscribe()
|
||||
assertEquals(1, first.unsubscribeCount)
|
||||
|
||||
reconnecting.close()
|
||||
} finally {
|
||||
scope.cancel()
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user