fix(session-query): contain sync cancellation failures

This commit is contained in:
Hypatia May
2026-07-11 10:59:54 +08:00
parent 6c6ce08a39
commit f4b0dba380
2 changed files with 66 additions and 2 deletions

View File

@@ -225,7 +225,12 @@ export class SessionProviderCoordinator {
private _syncLive(state: ProviderState, session: Session): Promise<void> {
const existing = state.liveSync.get(session.id)
if (existing !== undefined) return existing
const snapshot = this._snapshotLive(session)
let snapshot: ReturnType<SessionTextExtractors['buildSnapshot']>
try {
snapshot = this._snapshotLive(session)
} catch (error: unknown) {
return Promise.reject(this._synchronizationError(state, error))
}
const promise = this._enqueue(state, async () => {
/* v8 ignore next -- a provider can be disposed while queued behind an in-flight update */
if (!state.active) return
@@ -324,7 +329,12 @@ export class SessionProviderCoordinator {
function waitFor<T>(work: Promise<T>, signal: AbortSignal | undefined): Promise<T> {
if (signal === undefined) return work
if (signal.aborted) return Promise.reject(aborted())
if (signal.aborted) {
// Cancellation supersedes the caller's result, but shared work must still
// have a rejection observer when it has already failed synchronously.
void work.catch((_supersededError: unknown) => undefined)
return Promise.reject(aborted())
}
return new Promise<T>((resolve, reject) => {
const onAbort = () => { reject(aborted()) }
signal.addEventListener('abort', onAbort, { once: true })

View File

@@ -697,6 +697,60 @@ describe('provider selection and synchronization', () => {
expect(asError(thrown).message).toContain(`provider "${provider.id}"`)
expect(provider.sessionRequests).toEqual([])
})
it('observes synchronous synchronization failure when the caller is already aborted', async () => {
const ctx = await liveContext()
const session = ctx.sessions.create(SessionId('aborted-throwing-extractor'))
session.append('test/note', { note: 'unreachable' })
const provider = new FakeProvider()
ctx.sessionQuery.registerSearchProvider(provider)
ctx.sessionQuery.registerEventTextExtractor('test/note', {
version: 'aborted-throwing-v1',
extract: () => { throw new Error('superseded extraction failure') },
})
const controller = new AbortController()
controller.abort()
const unhandled: unknown[] = []
const onUnhandled = (reason: unknown) => { unhandled.push(reason) }
process.on('unhandledRejection', onUnhandled)
try {
await expect(ctx.sessionQuery.searchSessions({ query: 'x' }, { signal: controller.signal }))
.rejects.toThrow(expectCode('SESSION_QUERY_ABORTED'))
await new Promise<void>((resolve) => { setImmediate(resolve) })
expect(unhandled).toEqual([])
} finally {
process.off('unhandledRejection', onUnhandled)
}
})
it('types synchronous live-target extraction failures and leaves retries clean', async () => {
const ctx = await liveContext()
const session = ctx.sessions.create(SessionId('throwing-live-extractor'))
session.append('test/note', { note: 'unreachable' })
const provider = new FakeProvider()
ctx.sessionQuery.registerSearchProvider(provider)
const cause = new Error('live extractor failed')
const disposeExtractor = ctx.sessionQuery.registerEventTextExtractor('test/note', {
version: 'live-throwing-v1',
extract: () => { throw cause },
})
let thrown: unknown
try {
await ctx.sessionQuery.searchEvents({ sessionId: session.id, query: 'x' })
} catch (error: unknown) {
thrown = error
}
expect(thrown).toBeInstanceOf(SessionQueryError)
expect(thrown).toMatchObject({ code: 'SESSION_QUERY_INDEX_FAILED', cause })
expect(asError(thrown).message).toContain(`provider "${provider.id}"`)
expect(provider.eventRequests).toEqual([])
disposeExtractor()
await expect(ctx.sessionQuery.searchEvents({ sessionId: session.id, query: 'x' }))
.resolves.toMatchObject({ providerId: provider.id })
expect(provider.eventRequests).toHaveLength(1)
})
})
describe('semantic text extractors', () => {