code-server/src/node/routes/health.ts

29 lines
668 B
TypeScript

import { Router } from "express"
import { wss, Router as WsRouter } from "../wsRouter"
export const router = Router()
router.get("/", (req, res) => {
res.json({
status: req.heart.alive() ? "alive" : "expired",
lastHeartbeat: req.heart.lastHeartbeat,
})
})
export const wsRouter = WsRouter()
wsRouter.ws("/", async (req) => {
wss.handleUpgrade(req, req.ws, req.head, (ws) => {
ws.addEventListener("message", () => {
ws.send(
JSON.stringify({
event: "health",
status: req.heart.alive() ? "alive" : "expired",
lastHeartbeat: req.heart.lastHeartbeat,
}),
)
})
req.ws.resume()
})
})