Last active
March 11, 2024 10:20
-
-
Save taion/1cfe1595a32b89ea4265679b38eee7b9 to your computer and use it in GitHub Desktop.
GraphQL subscription server with Socket.IO, backed by Redis Pub/Sub
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
const redisClient = redis.createClient(REDIS_URL); | |
const listeners = Object.create(null); | |
function addListener(channel, listener) { | |
if (!listeners[channel]) { | |
listeners[channel] = []; | |
redisClient.subscribe(channel); | |
} | |
listeners[channel].push(listener); | |
return () => { | |
listeners[channel] = listeners[channel].filter( | |
item => item !== listener | |
); | |
if (!listeners[channel].length) { | |
redisClient.unsubscribe(channel); | |
delete listeners[channel]; | |
} | |
}; | |
} | |
redisClient.on('message', (channel, message) => { | |
const channelListeners = listeners[channel]; | |
if (!channelListeners) { | |
return; | |
} | |
let body; | |
try { | |
body = JSON.parse(message); | |
} catch (e) { | |
console.error(`malformed message on ${channel}: ${message}`); | |
} | |
if (!body) { | |
return; | |
} | |
const data = getResponseData(body); | |
channelListeners.forEach((listener) => { | |
listener(data); | |
}); | |
}); | |
const io = new IoServer(server, { | |
serveClient: false, | |
}); | |
io.on('connection', (socket) => { | |
// Turn the request into an Express request for consistency. | |
const req = Object.create((express: any).request); | |
Object.assign(req, socket.request, { app }); | |
let authorization = null; | |
let credentials = null; | |
socket.on('authenticate', (token) => { | |
authorization = `Bearer ${token}`; | |
try { | |
const credentials = getCredentials(token); | |
} catch (e) { | |
credentials = null; | |
socket.emit('error', { code: 'invalid_token' }); | |
} | |
}); | |
const unsubscribeMap = Object.create(null); | |
socket.on('subscribe', async ({ id, query, variables }) => { | |
if (unsubscribeMap[id]) { | |
socket.emit('error', { | |
code: 'invalid_id.duplicate', | |
detail: id, | |
}); | |
return; | |
} | |
function subscribe(channel) { | |
unsubscribeMap[id] = addListener(channel, async (body) => { | |
if (!authorizeRead(body, credentials)) { | |
// User is not authorized to view this update. | |
return; | |
} | |
const result = await graphql( | |
schema, | |
query, | |
body, | |
createContext(req, authorization), | |
variables, | |
); | |
socket.emit('subscription update', { id, ...result }); | |
}); | |
} | |
const result = await graphqlSubscribe({ | |
schema, | |
query, | |
variables, | |
context: { subscribe }, | |
}); | |
if (result.errors) { | |
socket.emit('error', { | |
code: 'subscribe_failed', | |
detail: result.errors, | |
}); | |
} | |
}); | |
socket.on('unsubscribe', (id) => { | |
const unsubscribe = unsubscribeMap[id]; | |
if (!unsubscribe) { | |
return; | |
} | |
unsubscribe(); | |
delete unsubscribeMap[id]; | |
}); | |
socket.on('disconnect', () => { | |
Object.keys(unsubscribeMap).forEach((id) => { | |
unsubscribeMap[id](); | |
}); | |
}); | |
}); |
Where is graphqlSubscribe
defined? Does it come from a library?
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
That's what i'm looking for . Thanks man 👍