Server-sent event with RTK query

[Dimatymoshchenko]

Hello guys, I'm facing a problem implementing SSE using RTQ Query. Everything works except one thing - closing the connection, can you tell me how to implement this in my case?

const sseRtkApi = createApi({
  reducerPath: 'rtkSseApi',
  baseQuery: RtkSseQuery(),
  endpoints: _ => ({}),
})

const RtkSseQuery = (
  { baseUrl }: { baseUrl: string } = { baseUrl: envConfiguration.studioApiHost },
): BaseQueryFn<
  {
    url: string
  }, // args
  unknown, // result
  RtkApiError // error
> => {
  let eventSource: EventSource | null = null

  const subscribeToSSE = (
    { url, token }: { url: string; token?: string },
    callback: (data: unknown, isError?: boolean) => void,
  ) => {
    eventSource = new (window as any).EventSourcePolyfill(url, {
      headers: {
        Authorization: `Basic ${token}`,
      },
    })

    if (!eventSource) {
      return
    }

    eventSource.onmessage = event => {
      const data = JSON.parse(event.data)
      callback(data)
    }

    eventSource.onerror = error => {
      eventSource?.close()
      callback(error, true)
    }
  }

  return async ({ url }, api) => {
    const currentUser = (api.getState() as RootState).user.currentUser
    return new Promise((resolve, reject) => {
      subscribeToSSE({ url: baseUrl + url, token: currentUser?.token }, (data, isError) => {
        if (isError) {
          reject({ error: data })
        }

        resolve({ data })
      })
    })
  }
}

const { data } = useGetAllDedicatedSessionsQuery()

How can I close the connection without access to the eventSource?

[Dimatymoshchenko]

Somehow callback triggers only once :(

[simon-lighthouse]

@Dimatymoshchenko did you resolve this?