GitHub/quartz
1
0
Fork 2
mirror of https://github.com/jackyzha0/quartz.git synced 2025-12-27 23:04:05 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add websocket keepalive

Browse Source
This commit is contained in:
Adam Sage 2024-09-15 16:24:56 -04:00 committed by GitHub
parent cd3bb25626
commit 858c1db159
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 21 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
quartz/cli/handlers.js
View File

@ -317,14 +317,10 @@ export async function handleBuild(argv) {
}
if (argv.serve) {
const connections = []
const clientRefresh = () => connections.forEach((conn) => conn.send("rebuild"))
if (argv.baseDir !== "" && !argv.baseDir.startsWith("/")) {
argv.baseDir = "/" + argv.baseDir
}
await build(clientRefresh)
const server = http.createServer(async (req, res) => {
if (argv.baseDir && !req.url?.startsWith(argv.baseDir)) {
console.log(
@ -410,8 +406,28 @@ export async function handleBuild(argv) {
return serve()
})
server.listen(argv.port)
function heartbeat() {
this.isAlive = true
}
const wss = new WebSocketServer({ port: argv.wsPort })
wss.on("connection", (ws) => connections.push(ws))
wss.on("connection", function connection(ws) {
ws.isAlive = true
ws.on("error", console.error)
ws.on("pong", heartbeat)
})
const interval = setInterval(function ping() {
wss.clients.forEach(function each(ws) {
if (ws.isAlive === false) return ws.terminate()
ws.isAlive = false
ws.ping()
})
}, 30000)
wss.on("close", function close() {
clearInterval(interval)
})
const clientRefresh = () => wss.clients.forEach((conn) => conn.send("rebuild"))
await build(clientRefresh)
console.log(
chalk.cyan(
`Started a Quartz server listening at http://localhost:${argv.port}${argv.baseDir}`,