@empyria/restate 0.4.16 → 0.4.18

package/AGENTS.md

@@ -79,6 +79,12 @@ may observe an already-mocked value. See `test/Admin.test.js`.
 same address registers again, but only `force: true` updates its handler list. Check both against a live server when
 touching them.
 
+`setupRestate` takes ONE complete `serviceURL` (scheme and explicit port included): it is registered as given and the
+endpoint listens on its port. There is deliberately no `host` or `port` option — never assemble a URL from parts, the
+scheme and any mapping are the caller's to state. `parseServiceURL` validates it (unit tested).
+`createRestateAdmin().setupRestate` must forward every option to `setupRestate` — it once dropped
+`keepDeploymentOnShutdown`/`forceRegistration`/`registrationCallbackFn`.
+
 `setupRestate` (in `lib/Admin.js`) is not unit tested: it binds real HTTP/2 + health-check
 ports, installs process-wide `SIGTERM`/`SIGINT` handlers, and its returned `forceClose` calls
 `process.exit()` — none of which are safe to exercise in-process in a test run.

package/README.md

@@ -122,9 +122,14 @@ endpoint, then create jobs with either:
 Cron expressions are `croner` patterns (five fields, or six with seconds for tests). The target handler is sent the
 `payload` as its input, or no input when there is none.
 
-### An endpoint that does not register itself
+### Serving and registering an endpoint
 
-`setupRestate` serves, registers the deployment, and deletes it again on SIGINT/SIGTERM. Invocations in flight are
+`setupRestate({ restateAdminURL, serviceURL: 'https://agent-service.internal:9080', services })` takes one complete URL, the
+address Restate reaches the endpoint at. It is registered as given and the endpoint listens on its port, so the scheme and
+an explicit port are required (there is no separate host or port option); the health endpoint listens on the next port
+unless `healthPort` is set.
+
+It serves, registers the deployment, and deletes it again on SIGINT/SIGTERM. Invocations in flight are
 pinned to their deployment, so deleting it strands them: a restarted process cannot finish them and they can only be
 killed. For a process restarted at the same address, pass `keepDeploymentOnShutdown: true` and `forceRegistration: true`:
 Restate keeps the deployment ID when the same address registers again (checked against Restate 1.7), running
@@ -132,8 +137,7 @@ invocations resume on the new process, and `force` updates the services and hand
 re-registration keeps the old handler list). Only compatible changes are safe this way: an invocation of a removed or
 renamed handler cannot resume.
 
-When something else owns
-registration (a supervisor that registers a process's address once and restarts the process), use
+To serve without registering: when something else owns registration (a supervisor that registers a process's address once and restarts the process), use
 `serveEndpoint({port, services})` instead: it only serves. Invocations already running on a deployment are pinned to its
 address, so the deployment must survive the process stopping and come back at the same address.
 

package/lib/Admin.js

@@ -550,7 +550,7 @@ export function sendMessageAsyncWithDiscovery({ restateAdminURL }) {
 }
 
 /**
- * @typedef {import('node:http2').Http2Server & {deploymentId: string, services: string[], forceClose: () => void}} RestateServer
+ * @typedef {import('node:http2').Http2Server & {deploymentId: string|undefined, services: string[]|undefined, forceClose: () => void}} RestateServer
  */
 
 /**
@@ -592,8 +592,9 @@ export function defineObject(def) {
  * that was removed or renamed cannot resume.
  * @param {Object} params
  * @param {string} params.restateAdminURL
- * @param {string} params.host
- * @param {number} params.port
+ * @param {string} params.serviceURL Complete URL Restate uses to reach this endpoint, scheme and port included
+ *   (e.g. `https://agent-service.internal:9080`). It is registered as given, and the endpoint listens on its port, so
+ *   the port must be explicit. Scheme and port are the caller's to state, never guessed here.
  * @param {number} [params.healthPort] - Defaults to `port + 1`.
  * @param {Parameters<typeof restate.createEndpointHandler>[0]['services']} [params.services]
  * @param {() => void} [params.shutdownCallbackFn]
@@ -605,15 +606,17 @@ export function defineObject(def) {
  */
 export async function setupRestate({
 	restateAdminURL,
-	host,
-	port,
-	healthPort = port + 1,
+	serviceURL: serviceURLParam,
+	healthPort: healthPortParam,
 	services = [],
 	shutdownCallbackFn,
 	registrationCallbackFn,
 	keepDeploymentOnShutdown = false,
 	forceRegistration = false,
 }) {
+	const { serviceURL, port } = parseServiceURL(serviceURLParam)
+	const healthPort = healthPortParam ?? port + 1
+
 	const handler = restate.createEndpointHandler({ services })
 
 	const server = http2.createServer(handler)
@@ -684,101 +687,121 @@ export async function setupRestate({
 
 	server.listen(port, () => console.log(`Server running on ${port}`))
 
-	if (host && port) {
-		const serviceURL = `http://${host}:${port}`
-
-		/**
-		 * @param {Object} params
-		 * @param {string} params.restateAdminURL
-		 * @param {string} params.serviceURL
-		 * @param {number} [params.attempts=5]
-		 * @param {number} [params.baseDelayMs=3000]
-		 * @returns {Promise<RegisterDeploymentResponse>}
-		 */
-		async function registerDeploymentWithRetry({
-			restateAdminURL,
-			serviceURL,
-			attempts = 5,
-			baseDelayMs = 3_000,
-		}) {
-			let lastErr
-
-			for (let attempt = 1; attempt <= attempts; attempt++) {
+	/**
+	 * @param {Object} params
+	 * @param {string} params.restateAdminURL
+	 * @param {string} params.serviceURL
+	 * @param {number} [params.attempts=5]
+	 * @param {number} [params.baseDelayMs=3000]
+	 * @returns {Promise<RegisterDeploymentResponse>}
+	 */
+	async function registerDeploymentWithRetry({
+		restateAdminURL,
+		serviceURL,
+		attempts = 5,
+		baseDelayMs = 3_000,
+	}) {
+		let lastErr
+
+		for (let attempt = 1; attempt <= attempts; attempt++) {
+			try {
+				const result = await registerDeployment({
+					restateAdminURL,
+					serviceURL,
+					force: forceRegistration,
+				})
+
 				try {
-					const result = await registerDeployment({
-						restateAdminURL,
-						serviceURL,
-						force: forceRegistration,
-					})
-
-					try {
-						registrationCallbackFn?.(result.id)
-					} catch (err) {
-						console.error(
-							`Error in registration callback for deployment ${result.id}: ${
-								err instanceof Error ? err.message : String(err)
-							}`,
-						)
-					}
-
-					return result
+					registrationCallbackFn?.(result.id)
 				} catch (err) {
-					lastErr = err
-
-					// Deterministic — retrying won't help. Bail immediately.
-					if (err instanceof Error && err.message.includes('META0004')) {
-						throw err
-					}
-
-					if (attempt < attempts) {
-						const delay = Math.min(baseDelayMs * 2 ** (attempt - 1), DELAY_CAP)
-						console.log(
-							`Registration attempt ${attempt}/${attempts} failed, retrying in ${delay}ms: ${
-								err instanceof Error ? err.message : String(err)
-							}`,
-						)
-						await sleep(delay)
-					}
+					console.error(
+						`Error in registration callback for deployment ${result.id}: ${
+							err instanceof Error ? err.message : String(err)
+						}`,
+					)
 				}
-			}
 
-			throw new Error(
-				`Failed to register deployment after ${attempts} attempts: ${
-					lastErr instanceof Error ? lastErr.message : String(lastErr)
-				}`,
-			)
-		}
-		const deployment = await registerDeploymentWithRetry({
-			restateAdminURL,
-			serviceURL,
-		}).catch(console.error)
+				return result
+			} catch (err) {
+				lastErr = err
 
-		isHealthy = !!deployment // Only healthy once registered
+				// Deterministic — retrying won't help. Bail immediately.
+				if (err instanceof Error && err.message.includes('META0004')) {
+					throw err
+				}
 
-		return Object.assign(server, {
-			deploymentId: deployment?.id,
-			services: deployment?.services?.map((s) => s.name),
-			forceClose,
-		})
-	} else {
-		isHealthy = true
+				if (attempt < attempts) {
+					const delay = Math.min(baseDelayMs * 2 ** (attempt - 1), DELAY_CAP)
+					console.log(
+						`Registration attempt ${attempt}/${attempts} failed, retrying in ${delay}ms: ${
+							err instanceof Error ? err.message : String(err)
+						}`,
+					)
+					await sleep(delay)
+				}
+			}
+		}
+
+		throw new Error(
+			`Failed to register deployment after ${attempts} attempts: ${
+				lastErr instanceof Error ? lastErr.message : String(lastErr)
+			}`,
+		)
 	}
+	const deployment = await registerDeploymentWithRetry({
+		restateAdminURL,
+		serviceURL,
+	}).catch(console.error)
+
+	isHealthy = !!deployment // Only healthy once registered
 
 	return Object.assign(server, {
-		deploymentId: 'NA',
-		services: [],
+		deploymentId: deployment?.id,
+		services: deployment?.services?.map((s) => s.name),
 		forceClose,
 	})
 }
 
 /**
- * Whether `setupRestate`'s shutdown removes the deployment: only a real one (registered, not `'NA'`), and not when the
- * caller keeps it for the next start.
+ * Validates the URL `setupRestate` registers with Restate and returns it with the port the endpoint listens on.
+ * @param {string} serviceURL Complete http(s) URL including an explicit port
+ * @returns {{serviceURL: string, port: number}} `serviceURL` unchanged
+ * @throws {Error} If it is missing, not an absolute http(s) URL, or has no explicit port (the scheme's default would
+ *   be a guess)
+ */
+export function parseServiceURL(serviceURL) {
+	if (!serviceURL) {
+		throw new Error(
+			'setupRestate needs `serviceURL`: the complete URL (scheme and port) Restate reaches this endpoint at',
+		)
+	}
+	let url
+	try {
+		url = new URL(serviceURL)
+	} catch {
+		throw new Error(
+			`serviceURL must be a complete URL including http:// or https://, got '${serviceURL}'`,
+		)
+	}
+	if (url.protocol !== 'http:' && url.protocol !== 'https:') {
+		throw new Error(`serviceURL must use http or https, got '${serviceURL}'`)
+	}
+	if (!url.port) {
+		throw new Error(
+			`serviceURL '${serviceURL}' has no explicit port: the endpoint listens on the port in the URL`,
+		)
+	}
+	return { serviceURL, port: Number(url.port) }
+}
+
+/**
+ * Whether `setupRestate`'s shutdown removes the deployment: only one that registered (`deploymentId` is undefined when registration
+ * failed), and not when the caller keeps it for the next start.
  * @param {{deploymentId: string|undefined, keepDeploymentOnShutdown?: boolean}} params
  * @returns {boolean}
  */
 export function shouldDeleteOnShutdown({ deploymentId, keepDeploymentOnShutdown = false }) {
-	return Boolean(deploymentId) && deploymentId !== 'NA' && !keepDeploymentOnShutdown
+	return Boolean(deploymentId) && !keepDeploymentOnShutdown
 }
 
 /**
@@ -885,15 +908,7 @@ export function createRestateAdmin({ restateAdminURL, restateURL }) {
 		listHandlers: ({ name }) => listHandlers({ restateAdminURL, name }),
 
 		// Management queries
-		setupRestate: ({ port, healthPort, host, services, shutdownCallbackFn }) =>
-			setupRestate({
-				restateAdminURL,
-				host,
-				port,
-				healthPort: healthPort ?? port + 1,
-				services,
-				shutdownCallbackFn,
-			}),
+		setupRestate: (params) => setupRestate({ ...params, restateAdminURL }),
 		registerDeployment: ({ serviceURL }) => registerDeployment({ restateAdminURL, serviceURL }),
 
 		submitWorkflow: ({ name, key, payload }) =>

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@empyria/restate",
-	"version": "0.4.16",
+	"version": "0.4.18",
 	"description": "Restate.dev helpers for the Empyria nanoservice framework",
 	"license": "MIT",
 	"author": "Imre Fazekas <imre.fazekas@icloud.com>",

package/test/Admin.test.js

@@ -21,6 +21,7 @@ import {
 	sendMessageAsyncWithDiscovery,
 	registerDeployment,
 	shouldDeleteOnShutdown,
+	parseServiceURL,
 	sendMessageAsynSafe,
 	defineWorkflow,
 	defineService,
@@ -542,6 +543,35 @@ describe('registerDeployment', () => {
 	})
 })
 
+describe('parseServiceURL', () => {
+	test('returns the URL unchanged with the port it carries', () => {
+		expect(parseServiceURL('http://agent-service:9080')).toEqual({
+			serviceURL: 'http://agent-service:9080',
+			port: 9080,
+		})
+		expect(parseServiceURL('https://svc.example.com:9443/base')).toEqual({
+			serviceURL: 'https://svc.example.com:9443/base',
+			port: 9443,
+		})
+	})
+
+	test('requires a serviceURL', () => {
+		expect(() => parseServiceURL(undefined)).toThrow(/needs `serviceURL`/)
+		expect(() => parseServiceURL('')).toThrow(/needs `serviceURL`/)
+	})
+
+	test('rejects a URL without a scheme or with a non-http scheme', () => {
+		expect(() => parseServiceURL('agent-service:9080')).toThrow(/http/)
+		expect(() => parseServiceURL('not a url')).toThrow(/complete URL/)
+		expect(() => parseServiceURL('ftp://svc:21')).toThrow(/http or https/)
+	})
+
+	test('refuses to guess the scheme default when the URL has no explicit port', () => {
+		expect(() => parseServiceURL('https://svc.example.com')).toThrow(/no explicit port/)
+		expect(() => parseServiceURL('http://svc:80')).toThrow(/no explicit port/)
+	})
+})
+
 describe('shouldDeleteOnShutdown', () => {
 	test('deletes a registered deployment by default', () => {
 		expect(shouldDeleteOnShutdown({ deploymentId: 'dep-1' })).toBe(true)
@@ -554,7 +584,7 @@ describe('shouldDeleteOnShutdown', () => {
 	})
 
 	test('never deletes without a real deployment', () => {
-		for (const deploymentId of [undefined, '', 'NA']) {
+		for (const deploymentId of [undefined, '']) {
 			expect(shouldDeleteOnShutdown({ deploymentId })).toBe(false)
 		}
 	})