discours.io/webapp
5
0
Fork 0
Code Issues Pull Requests Actions Projects Wiki Activity

connect-fox

Browse Source
This commit is contained in:
Untone 2024-05-18 20:43:20 +03:00
parent c8517c85c6
commit 95c4d777b2
1 changed files with 41 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

78
src/context/connect.tsx
View File

@ -30,53 +30,57 @@ const ConnectContext = createContext<ConnectContextType>()
export const ConnectProvider = (props: { children: JSX.Element }) => { export const ConnectProvider = (props: { children: JSX.Element }) => {
const [messageHandlers, setHandlers] = createSignal<MessageHandler[]>([]) const [messageHandlers, setHandlers] = createSignal<MessageHandler[]>([])
// const [messages, setMessages] = createSignal<Array<SSEMessage>>([]);
const [connected, setConnected] = createSignal(false) const [connected, setConnected] = createSignal(false)
const { session } = useSession() const { session } = useSession()
const [retried, setRetried] = createSignal<number>(0)
const addHandler = (handler: MessageHandler) => { const addHandler = (handler: MessageHandler) => {
setHandlers((hhh) => [...hhh, handler]) setHandlers((hhh) => [...hhh, handler])
} }
const [retried, setRetried] = createSignal<number>(0)
createEffect(async () => { createEffect(async () => {
const token = session()?.access_token const token = session()?.access_token
if (token && !connected()) { if (token && !connected() && retried() <= RECONNECT_TIMES) {
console.info('[context.connect] init SSE connection') console.info('[context.connect] init SSE connection')
await fetchEventSource('https://connect.discours.io', { try {
method: 'GET', await fetchEventSource('https://connect.discours.io', {
headers: { method: 'GET',
'Content-Type': 'application/json', headers: {
Authorization: token, 'Content-Type': 'application/json',
}, Authorization: token,
onmessage(event) { },
const m: SSEMessage = JSON.parse(event.data || '{}') onmessage(event) {
console.log('[context.connect] Received message:', m) const m: SSEMessage = JSON.parse(event.data || '{}')
console.log('[context.connect] Received message:', m)
// Iterate over all registered handlers and call them messageHandlers().forEach((handler) => handler(m))
messageHandlers().forEach((handler) => handler(m)) },
}, onopen: (response) => {
async onopen(response) { console.log('[context.connect] SSE connection opened', response)
console.log('[context.connect] SSE connection opened', response) if (response.ok && response.headers.get('content-type') === EventStreamContentType) {
if (response.ok && response.headers.get('content-type') === EventStreamContentType) { setConnected(true)
setConnected(true) setRetried(0)
} else if (response.status === 401) { return Promise.resolve()
throw new Error('SSE: cannot connect to real-time updates') }
} else { return Promise.reject(`SSE: cannot connect to real-time updates, status: ${response.status}`)
setRetried((r) => r + 1) },
throw new Error(`SSE: failed to connect ${retried()} times`) onclose() {
} console.log('[context.connect] SSE connection closed by server')
}, setConnected(false)
onclose() { if (retried() < RECONNECT_TIMES) {
console.log('[context.connect] SSE connection closed by server') setRetried((r) => r + 1)
setConnected(false) }
}, },
onerror(err) { onerror(err) {
if (err.message === 'unauthorized' || retried() > RECONNECT_TIMES) { console.error('[context.connect] SSE connection error:', err)
throw err // rethrow to stop the operation setConnected(false)
} if (retried() < RECONNECT_TIMES) {
}, setRetried((r) => r + 1)
}) } else throw Error(err)
},
})
} catch (error) {
console.error('[context.connect] SSE connection failed:', error)
}
} }
}) })