@empyria/restate 0.4.18 → 0.4.20

package/lib/Admin.js

@@ -578,6 +578,34 @@ export function defineObject(def) {
 	return restate.object(def)
 }
 
+/**
+ * The health check `setupRestate` uses unless given a `healthCheckFn`: `GET /health`, `/health/liveness` and
+ * `/health/readiness` answer 200 once the deployment is registered and 503 before, anything else 404.
+ * @param {import('node:http').IncomingMessage} req
+ * @param {import('node:http').ServerResponse} res
+ * @param {{isHealthy: boolean, deploymentId?: string, services?: string[]}} state
+ */
+export function defaultHealthCheck(req, res, { isHealthy, deploymentId, services }) {
+	if (
+		(req.url === '/health' ||
+			req.url === '/health/liveness' ||
+			req.url === '/health/readiness') &&
+		req.method === 'GET'
+	) {
+		res.writeHead(isHealthy ? 200 : 503, { 'Content-Type': 'application/json' })
+		res.end(
+			JSON.stringify({
+				status: isHealthy ? 'ok' : 'unavailable',
+				deployment: deploymentId,
+				services,
+			}),
+		)
+	} else {
+		res.writeHead(404)
+		res.end()
+	}
+}
+
 /**
  * Starts the HTTP/2 Restate endpoint, a health-check HTTP server, registers
  * the deployment with the admin API, and wires SIGTERM/SIGINT for graceful
@@ -599,6 +627,10 @@ export function defineObject(def) {
  * @param {Parameters<typeof restate.createEndpointHandler>[0]['services']} [params.services]
  * @param {() => void} [params.shutdownCallbackFn]
  * @param {(deploymentId: string) => void} [params.registrationCallbackFn]
+ * @param {(req: import('node:http').IncomingMessage, res: import('node:http').ServerResponse, state: {isHealthy: boolean, deploymentId?: string, services?: string[]}) => void | Promise<void>} [params.healthCheckFn]
+ *   Takes over every request to the health port: routing, status code, headers and body are the function's to set and
+ *   it must end the response. `state.isHealthy` is true once the deployment is registered. Without it, `/health`,
+ *   `/health/liveness` and `/health/readiness` answer 200 or 503 with a small JSON body, healthy once registered.
  * @param {boolean} [params.keepDeploymentOnShutdown=false] Leave the deployment registered on SIGINT/SIGTERM.
  * @param {boolean} [params.forceRegistration=false] Register with `force: true`, updating an existing deployment at
  *   the same address in place.
@@ -611,6 +643,7 @@ export async function setupRestate({
 	services = [],
 	shutdownCallbackFn,
 	registrationCallbackFn,
+	healthCheckFn,
 	keepDeploymentOnShutdown = false,
 	forceRegistration = false,
 }) {
@@ -624,25 +657,20 @@ export async function setupRestate({
 	// --- Health server ---
 	let isHealthy = false
 
+	const checkFn = healthCheckFn ?? defaultHealthCheck
 	const healthServer = createHttpServer((req, res) => {
-		if (
-			(req.url === '/health' ||
-				req.url === '/health/liveness' ||
-				req.url === '/health/readiness') &&
-			req.method === 'GET'
-		) {
-			const status = isHealthy ? 200 : 503
-			const body = JSON.stringify({
-				status: isHealthy ? 'ok' : 'unavailable',
-				deployment: server.deploymentId,
-				services: server.services,
-			})
-			res.writeHead(status, { 'Content-Type': 'application/json' })
-			res.end(body)
-		} else {
-			res.writeHead(404)
-			res.end()
+		const state = {
+			isHealthy,
+			deploymentId: server.deploymentId,
+			services: server.services,
 		}
+		Promise.resolve()
+			.then(() => checkFn(req, res, state))
+			.catch((err) => {
+				console.error('healthCheckFn failed', err)
+				if (!res.headersSent) res.writeHead(500)
+				res.end()
+			})
 	})
 
 	healthServer.listen(healthPort, () => console.log(`Health endpoint on :${healthPort}/health`))

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@empyria/restate",
-	"version": "0.4.18",
+	"version": "0.4.20",
 	"description": "Restate.dev helpers for the Empyria nanoservice framework",
 	"license": "MIT",
 	"author": "Imre Fazekas <imre.fazekas@icloud.com>",
@@ -36,10 +36,10 @@
 		"croner": "10.0.1"
 	},
 	"devDependencies": {
-		"@ai-sdk/openai-compatible": "3.0.63",
-		"@ai-sdk/provider": "4.0.22",
+		"@ai-sdk/openai-compatible": "3.0.65",
+		"@ai-sdk/provider": "4.0.24",
 		"@restatedev/restate-server": "1.7.13",
-		"ai": "7.0.128",
+		"ai": "7.0.130",
 		"oxfmt": "0.72.0",
 		"oxlint": "1.87.0"
 	},