48 lines
1.1 KiB
TypeScript
48 lines
1.1 KiB
TypeScript
// This is a "side-effect" but we want a lifecycle cache map of
|
|
// peer statuses to prevent unnecessary fetches to the agent.
|
|
import type { LoaderFunctionArgs } from 'remix'
|
|
|
|
type Context = LoaderFunctionArgs['context']
|
|
const cache: { [nodeID: string]: unknown } = {}
|
|
|
|
export async function queryWS(context: Context, nodeIDs: string[]) {
|
|
const ws = context.ws
|
|
const firstClient = ws.clients.values().next().value
|
|
if (!firstClient) {
|
|
return cache
|
|
}
|
|
|
|
const cached = nodeIDs.map((nodeID) => {
|
|
const cached = cache[nodeID]
|
|
if (cached) {
|
|
return cached
|
|
}
|
|
})
|
|
|
|
// We only need to query the nodes that are not cached
|
|
const uncached = nodeIDs.filter((nodeID) => !cached.includes(nodeID))
|
|
if (uncached.length === 0) {
|
|
return cache
|
|
}
|
|
|
|
firstClient.send(JSON.stringify({ NodeIDs: uncached }))
|
|
await new Promise((resolve) => {
|
|
const timeout = setTimeout(() => {
|
|
resolve()
|
|
}, 3000)
|
|
|
|
firstClient.on('message', (message) => {
|
|
const data = JSON.parse(message.toString())
|
|
if (Object.keys(data).length === 0) {
|
|
resolve()
|
|
}
|
|
|
|
for (const [nodeID, status] of Object.entries(data)) {
|
|
cache[nodeID] = status
|
|
}
|
|
})
|
|
})
|
|
|
|
return cache
|
|
}
|