@genroc/eval-node 0.0.0-edge.ea911c5 → 0.0.0-edge.eadd43a

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/README.md CHANGED
@@ -90,6 +90,23 @@ queue worker rather than the HTTP sidecar it used to be:
90
90
  that genroc had to be able to reach. A worker only needs outbound access, so it can live
91
91
  anywhere — behind NAT, in another trust zone.
92
92
 
93
+ The cost of that inversion is that genroc cannot reach a running worker, so anything it needs
94
+ to say has to ride the renewal the worker already makes. **Renewing is therefore mandatory,
95
+ not an optimisation** — a worker that stops renewing is indistinguishable from one that died.
96
+ The response answers per token:
97
+
98
+ | list | what it means | what this worker does |
99
+ |---|---|---|
100
+ | `renewed` | still yours | carry on |
101
+ | `lost` | already someone else's | stop; do **not** release — that would bump the new holder's claim |
102
+ | `cancelled` | still yours, nobody wants it | abort the script and release the claim |
103
+
104
+ `cancelled` is how an operator's `genctl cancel` reaches work already running: the evaluation
105
+ is aborted through an `AbortSignal`, its realm is terminated, and no outcome is submitted —
106
+ the process is stopping, so there is nothing left to answer. `renew_before_ms` on the claim
107
+ says how long the worker may wait before renewing again, so the interval is a value it reads
108
+ rather than one it guesses.
109
+
93
110
  ## The task input
94
111
 
95
112
  The `input` of the external task IS the evaluation request:
@@ -169,7 +186,7 @@ no retry policy to write.
169
186
 
170
187
  - id: script_failed
171
188
  switch:
172
- - case: 'error.data.name == "LimitExceeded"'
189
+ - case: 'last_error.data.name == "LimitExceeded"'
173
190
  raise: { code: limit_exceeded, message: "the script rejected the amount" }
174
191
  - raise: { code: script_failed, message: "the script failed" }
175
192
  ```
package/dist/eval.js CHANGED
@@ -20,13 +20,31 @@ class RealmFault extends Error {
20
20
  this.name = "RealmFault";
21
21
  }
22
22
  }
23
- export async function evaluate(req) {
23
+ /** Thrown when `signal` aborts: the work was cancelled server-side, so there is no outcome to
24
+ * report and worker.ts must release rather than answer. Distinct from RealmFault because the
25
+ * runner did not fault -- nothing is wrong, the answer is simply no longer wanted. */
26
+ export class Cancelled extends Error {
27
+ constructor() {
28
+ super("cancelled");
29
+ this.name = "Cancelled";
30
+ }
31
+ }
32
+ export async function evaluate(req, signal) {
24
33
  const budget = typeof req.timeout_ms === "number" ? req.timeout_ms : DEFAULT_TIMEOUT_MS;
25
34
  const worker = new Worker(REALM_URL);
26
35
  let timer;
36
+ let onAbort;
27
37
  try {
28
38
  return await new Promise((resolve, reject) => {
29
39
  timer = setTimeout(() => resolve(timedOut(budget)), budget);
40
+ // The abort is wired to the same promise as the budget, so both settle through the one
41
+ // finally below -- which is what guarantees the thread is gone before either returns.
42
+ if (signal) {
43
+ if (signal.aborted)
44
+ reject(new Cancelled());
45
+ onAbort = () => reject(new Cancelled());
46
+ signal.addEventListener("abort", onAbort, { once: true });
47
+ }
30
48
  worker.once("message", (reply) => resolve(reply));
31
49
  // A script may end its own realm (`process.exit()`), which is not a throw and would
32
50
  // otherwise present as a hang until the budget expired. Our own terminate() raises
@@ -38,6 +56,8 @@ export async function evaluate(req) {
38
56
  }
39
57
  finally {
40
58
  clearTimeout(timer);
59
+ if (signal && onAbort)
60
+ signal.removeEventListener("abort", onAbort);
41
61
  // Awaited, and the whole point: on the timeout path a thread is still burning a core, and
42
62
  // resolving before it is gone would report an evaluation the machine is still running.
43
63
  await worker.terminate();
package/dist/import.js CHANGED
@@ -7,9 +7,10 @@
7
7
  // code; "build" typechecks and bundles. A separate types hook would mean a second `tsc`
8
8
  // over the same project.
9
9
  import { spawn } from "node:child_process";
10
+ import { existsSync } from "node:fs";
10
11
  import { access, mkdir, writeFile } from "node:fs/promises";
11
12
  import { builtinModules } from "node:module";
12
- import { dirname, join, relative } from "node:path";
13
+ import { dirname, join, relative, resolve } from "node:path";
13
14
  import { fileURLToPath } from "node:url";
14
15
  import commonjs from "@rollup/plugin-commonjs";
15
16
  import json from "@rollup/plugin-json";
@@ -112,21 +113,14 @@ function deref(s, defs) {
112
113
  }
113
114
  return cur;
114
115
  }
115
- /** The manifest's `input` is the type of the whole ACTION input — for /eval that is
116
- * `{code, input, timeout_ms, …}`, and only its `input` field is bound as the script's
117
- * parameter. genroc cannot know that; this file owns the evaluator's wire contract, so
118
- * the navigation belongs here rather than in genctl. */
119
- function scriptInput(site, defs) {
120
- const action = deref(site.input, defs);
121
- const bound = action?.properties?.input;
122
- return bound ?? site.input;
123
- }
124
116
  /** Emits one named type per reachable $def rather than inlining: a task output may
125
117
  * reference itself (specs/recursive-type-inference.md) and inlining would not terminate. */
126
- function declarations(site, defs) {
118
+ function declarations(at) {
119
+ const { site, where } = at;
120
+ const defs = where.$defs ?? {};
127
121
  const used = new Set();
128
- const input = tsType(scriptInput(site, defs), used);
129
- const output = tsType(site.output, used);
122
+ const input = tsType(site.types?.Input, used);
123
+ const output = tsType(site.types?.Output, used);
130
124
  const emitted = [];
131
125
  const done = new Set();
132
126
  while (true) {
@@ -139,7 +133,7 @@ function declarations(site, defs) {
139
133
  }
140
134
  return [
141
135
  "// Generated by genroc. Do not edit - regenerate with `genctl types`.",
142
- `// ${site.process}${site.task ? ` / ${site.task}` : ""} (${site.pointer})`,
136
+ `// ${where.name} (${address(site.pointer)})`,
143
137
  "",
144
138
  ...emitted,
145
139
  emitted.length ? "" : "",
@@ -167,7 +161,7 @@ async function nearestTsconfig(from, root) {
167
161
  return null;
168
162
  }
169
163
  }
170
- async function typecheck(root, sites) {
164
+ async function typecheck(sites) {
171
165
  // NOT `.genroc`: that is the project config FILE, and a directory of the same name cannot
172
166
  // coexist with it. The suffix is what keeps the scratch area out of its way.
173
167
  const dir = join(root, ".genroc-cache");
@@ -175,13 +169,13 @@ async function typecheck(root, sites) {
175
169
  // One tsc per distinct base config: `extends` takes a single base, so merging two would
176
170
  // check each script under the other author's options.
177
171
  const groups = new Map();
178
- for (const site of sites) {
179
- const base = (await nearestTsconfig(dirname(site.path), root)) ?? "";
172
+ for (const at of sites) {
173
+ const base = (await nearestTsconfig(dirname(at.file), root)) ?? "";
180
174
  const group = groups.get(base);
181
175
  if (group)
182
- group.push(site);
176
+ group.push(at);
183
177
  else
184
- groups.set(base, [site]);
178
+ groups.set(base, [at]);
185
179
  }
186
180
  let n = 0;
187
181
  for (const [base, group] of groups) {
@@ -203,8 +197,8 @@ async function typecheck(root, sites) {
203
197
  ...(base ? {} : { types: [] }),
204
198
  },
205
199
  files: group.flatMap((s) => [
206
- relative(dir, s.path),
207
- relative(dir, typesPathFor(s.path)),
200
+ relative(dir, s.file),
201
+ relative(dir, typesPathFor(s.file)),
208
202
  ]),
209
203
  // `files` overrides the base's, but a base `include` survives beside it and would
210
204
  // drag the author's whole tree in, to be checked under the worker lib.
@@ -287,15 +281,16 @@ function tsResolve(configPath) {
287
281
  /** Bundles to a self-contained ES module, which is what the evaluator imports: the default
288
282
  * export it calls is the author's own, so nothing wraps or rewrites the code between the two.
289
283
  * Bundling is entirely the importer's job, so a definition version pins its code forever. */
290
- async function bundle(site, root) {
284
+ async function bundle(at) {
285
+ const site = at.site;
291
286
  // Builtins are EXTERNALISED as imports the realm resolves natively. Anything else
292
287
  // unresolved is a REFUSAL, not an external: rollup's default is to leave it as an import
293
288
  // of a module that will not be there, which bundles clean and fails at runtime.
294
289
  const built = await rollup({
295
- input: site.path,
290
+ input: at.file,
296
291
  external: (id) => BUILTIN.has(id),
297
292
  plugins: [
298
- tsResolve(await nearestTsconfig(dirname(site.path), root)),
293
+ tsResolve(await nearestTsconfig(dirname(at.file), root)),
299
294
  nodeResolve({ extensions: [".ts", ".tsx", ".mjs", ".js", ".json"] }),
300
295
  commonjs(),
301
296
  // A `.json` import is a data file inlined at build time, which the previous bundler
@@ -305,10 +300,10 @@ async function bundle(site, root) {
305
300
  ],
306
301
  onwarn(warning) {
307
302
  if (warning.code === "UNRESOLVED_IMPORT") {
308
- die(`${site.path}: cannot resolve ${warning.exporter ?? "an import"} — is it installed?`);
303
+ die(`${at.file}: cannot resolve ${warning.exporter ?? "an import"} — is it installed?`);
309
304
  }
310
305
  },
311
- }).catch((e) => die(`${site.path}: ${e instanceof Error ? e.message : String(e)}`));
306
+ }).catch((e) => die(`${at.file}: ${e instanceof Error ? e.message : String(e)}`));
312
307
  const { output } = await built.generate({
313
308
  format: "es",
314
309
  inlineDynamicImports: true,
@@ -317,7 +312,7 @@ async function bundle(site, root) {
317
312
  // Refused here rather than in the realm: the evaluator can only report it against a running
318
313
  // instance, and the file it names is on this machine.
319
314
  if (!output[0].exports.includes("default")) {
320
- die(`${site.path}: a script must \`export default\` the function to run`);
315
+ die(`${at.file}: a script must \`export default\` the function to run`);
321
316
  }
322
317
  return output[0].code;
323
318
  }
@@ -330,32 +325,73 @@ const stdin = await new Promise((resolve, reject) => {
330
325
  process.stdin.on("error", reject);
331
326
  });
332
327
  const manifest = JSON.parse(stdin);
333
- if (!manifest || !Array.isArray(manifest.sites))
328
+ // genctl runs a resolver with the project root as its cwd, so the manifest need not say so.
329
+ const root = process.cwd();
330
+ if (!manifest || !Array.isArray(manifest.processes))
334
331
  die("stdin is not a genroc resolver manifest");
332
+ /** A pointer as the address it is, so a generated comment or an error can be pasted into
333
+ * `genctl schema`. A key no identifier can spell is bracketed, as the grammar spells it. */
334
+ function address(pointer) {
335
+ return pointer
336
+ .map((seg) => typeof seg === "string" && /^[A-Za-z_][A-Za-z0-9_]*$/.test(seg)
337
+ ? `.${seg}`
338
+ : `[${JSON.stringify(seg)}]`)
339
+ .join("")
340
+ .replace(/^\./, "");
341
+ }
342
+ const located = manifest.processes.flatMap((where) => where.sites.map((site) => ({
343
+ site,
344
+ where,
345
+ file: resolve(where.dir, site.argument),
346
+ })));
347
+ // genctl is agnostic about what a script is for; the contract that an evaluation request carries
348
+ // its module in `code` is THIS resolver's, so it is the one that checks a directive landed there.
349
+ // `child` is the shape the scaffold uses — a call to a process that forwards to the evaluator —
350
+ // and `external` is the same request made directly.
351
+ for (const at of located) {
352
+ // Two halves, from the two places that carry them: the action's KIND is a field, and WHERE in
353
+ // it the directive sits is the pointer. A child call is the shape the scaffold generates — to a
354
+ // process that forwards to the evaluator — and an external task is the same request made
355
+ // directly.
356
+ const kind = at.site.level === "action" ? (at.site.action ?? "") : "";
357
+ const slot = at.site.pointer.slice(-2).join(".");
358
+ if ((kind !== "child" && kind !== "external") || slot !== "input.code") {
359
+ die(`${join(at.where.dir, at.where.file)}: ${address(at.site.pointer)}: an evaluated script ` +
360
+ "belongs in the `code` field of a child or external task's input, and this is " +
361
+ `${kind ? `a ${kind} task's ` : ""}\`${slot}\`.`);
362
+ }
363
+ }
364
+ // genctl no longer stats the argument — it does not know it is a file — so the resolver that
365
+ // does is the one that must say when it is not there.
366
+ for (const at of located) {
367
+ if (!existsSync(at.file)) {
368
+ die(`${join(at.where.dir, at.where.file)}: ${address(at.site.pointer)}: ` +
369
+ `"$import: ${at.site.argument}" names no file (looked at ${at.file})`);
370
+ }
371
+ }
335
372
  // One script at two sites with different input types is a refusal, not a union: the union
336
373
  // is sound and would typecheck a body that is wrong at one of the sites.
337
374
  const byPath = new Map();
338
- for (const site of manifest.sites) {
339
- const seen = byPath.get(site.path);
340
- const defsOf = (x) => (manifest.schemas[x.process]?.$defs ?? {});
375
+ for (const at of located) {
376
+ const seen = byPath.get(at.file);
341
377
  if (seen &&
342
- JSON.stringify(scriptInput(seen, defsOf(seen))) !==
343
- JSON.stringify(scriptInput(site, defsOf(site)))) {
344
- die(`${site.path} is imported at ${seen.pointer} and ${site.pointer} with different input types.\n` +
378
+ JSON.stringify(seen.site.types) !== JSON.stringify(at.site.types)) {
379
+ die(`${at.file} is imported at ${address(seen.site.pointer)} and ` +
380
+ `${address(at.site.pointer)} with different input types.\n` +
345
381
  "Split it into two scripts, or make the two call sites pass the same shape.");
346
382
  }
347
- byPath.set(site.path, site);
383
+ byPath.set(at.file, at);
348
384
  }
349
- for (const site of byPath.values()) {
350
- const defs = (manifest.schemas[site.process]?.$defs ?? {});
351
- await write(typesPathFor(site.path), declarations(site, defs));
385
+ for (const at of byPath.values()) {
386
+ await write(typesPathFor(at.file), declarations(at));
352
387
  }
353
388
  if (manifest.mode === "types") {
389
+ console.log(JSON.stringify(manifest, null, 2));
354
390
  process.exit(0);
355
391
  }
356
- await typecheck(manifest.root, [...byPath.values()]);
392
+ await typecheck([...byPath.values()]);
357
393
  const code = [];
358
- for (const site of manifest.sites) {
359
- code.push(await bundle(site, manifest.root));
394
+ for (const at of located) {
395
+ code.push(await bundle(at));
360
396
  }
361
397
  process.stdout.write(JSON.stringify({ code }));
package/dist/worker.js CHANGED
@@ -5,7 +5,7 @@
5
5
  //
6
6
  // See README.md for the contract, and specs/external-task-queue.md for the queue itself.
7
7
  import { readFileSync } from "node:fs";
8
- import { evaluate } from "./eval.js";
8
+ import { Cancelled, evaluate } from "./eval.js";
9
9
  const SERVER = (process.env.GENROC_SERVER ?? "http://localhost:8448").replace(/\/$/, "");
10
10
  const WORKER_ID = process.env.WORKER_ID ?? `evaluator-${process.pid}`;
11
11
  // The credential, when the server runs with --auth token. A worker needs exactly the `worker`
@@ -185,7 +185,7 @@ async function answer(token, outcome) {
185
185
  console.error(`genroc refused the outcome for ${token}: ${JSON.stringify(data)}`);
186
186
  await release(token);
187
187
  }
188
- async function run(job) {
188
+ async function run(job, signal) {
189
189
  let resolved;
190
190
  try {
191
191
  resolved = await resolveObjects(job);
@@ -206,9 +206,17 @@ async function run(job) {
206
206
  }
207
207
  let result;
208
208
  try {
209
- result = await evaluate(req);
209
+ result = await evaluate(req, signal);
210
210
  }
211
211
  catch (err) {
212
+ // Cancelled is not a fault: the process was stopped while this ran. The claim still goes
213
+ // back — the row is terminal, so nothing re-claims it, and the release is what stops it
214
+ // waiting out a lease nobody is serving.
215
+ if (err instanceof Cancelled) {
216
+ console.log(`cancelled: ${job.token}`);
217
+ await release(job.token);
218
+ return;
219
+ }
212
220
  // The RUNNER faulted, not the script — the one class where a retry can help. There is no
213
221
  // error code for it on purpose: releasing the claim is how a queue spells "retryable", and
214
222
  // it puts the task in front of a different worker instead of burning the definition's
@@ -245,11 +253,22 @@ async function renewLoop() {
245
253
  tokens,
246
254
  lease_ms: LEASE_MS,
247
255
  });
248
- // A short count means a claim lapsed and was taken over. Nothing to do about it — the work
249
- // continues and its answer will be refused — but say so, because it is the signal that
250
- // LEASE_MS is too short for what these scripts actually take.
251
- if (ok && data?.renewed < tokens.length) {
252
- console.error(`renewed ${data.renewed}/${tokens.length} claims; a lease lapsed under load`);
256
+ if (!ok)
257
+ continue;
258
+ // Cancelled: the process was stopped, so the evaluation is abandoned and the claim handed
259
+ // back. Aborting first is the point of the list — waiting for the script to finish would
260
+ // keep burning a core on work an operator has already stopped. The release rides run()'s
261
+ // own catch, so nothing is released twice.
262
+ for (const token of (data?.cancelled ?? [])) {
263
+ inFlight.get(token)?.abort.abort();
264
+ }
265
+ // Lost: someone else holds this claim now. The work continues and its answer will be
266
+ // refused — that is not fixable here — but say so, because it is the signal that LEASE_MS
267
+ // is too short for what these scripts actually take. Deliberately NOT released: the claim
268
+ // is the new holder's, and releasing would bump the epoch out from under it.
269
+ const lost = (data?.lost ?? []);
270
+ if (lost.length) {
271
+ console.error(`lost ${lost.length}/${tokens.length} claims; a lease lapsed under load`);
253
272
  }
254
273
  }
255
274
  }
@@ -258,8 +277,9 @@ async function pollLoop() {
258
277
  const free = CONCURRENCY - inFlight.size;
259
278
  const jobs = free > 0 ? await claim(free) : [];
260
279
  for (const job of jobs) {
261
- inFlight.set(job.token, job);
262
- void run(job).finally(() => inFlight.delete(job.token));
280
+ const abort = new AbortController();
281
+ inFlight.set(job.token, { job, abort });
282
+ void run(job, abort.signal).finally(() => inFlight.delete(job.token));
263
283
  }
264
284
  // Only idle when there was nothing to take: a full queue should be drained at the speed the
265
285
  // realms allow, not at the poll interval.
package/eval.ts CHANGED
@@ -57,14 +57,32 @@ class RealmFault extends Error {
57
57
  }
58
58
  }
59
59
 
60
- export async function evaluate(req: EvalRequest): Promise<EvalResult> {
60
+ /** Thrown when `signal` aborts: the work was cancelled server-side, so there is no outcome to
61
+ * report and worker.ts must release rather than answer. Distinct from RealmFault because the
62
+ * runner did not fault -- nothing is wrong, the answer is simply no longer wanted. */
63
+ export class Cancelled extends Error {
64
+ constructor() {
65
+ super("cancelled");
66
+ this.name = "Cancelled";
67
+ }
68
+ }
69
+
70
+ export async function evaluate(req: EvalRequest, signal?: AbortSignal): Promise<EvalResult> {
61
71
  const budget = typeof req.timeout_ms === "number" ? req.timeout_ms : DEFAULT_TIMEOUT_MS;
62
72
 
63
73
  const worker = new Worker(REALM_URL);
64
74
  let timer: ReturnType<typeof setTimeout> | undefined;
75
+ let onAbort: (() => void) | undefined;
65
76
  try {
66
77
  return await new Promise<EvalResult>((resolve, reject) => {
67
78
  timer = setTimeout(() => resolve(timedOut(budget)), budget);
79
+ // The abort is wired to the same promise as the budget, so both settle through the one
80
+ // finally below -- which is what guarantees the thread is gone before either returns.
81
+ if (signal) {
82
+ if (signal.aborted) reject(new Cancelled());
83
+ onAbort = () => reject(new Cancelled());
84
+ signal.addEventListener("abort", onAbort, { once: true });
85
+ }
68
86
  worker.once("message", (reply: WorkerReply) => resolve(reply));
69
87
  // A script may end its own realm (`process.exit()`), which is not a throw and would
70
88
  // otherwise present as a hang until the budget expired. Our own terminate() raises
@@ -75,6 +93,7 @@ export async function evaluate(req: EvalRequest): Promise<EvalResult> {
75
93
  });
76
94
  } finally {
77
95
  clearTimeout(timer);
96
+ if (signal && onAbort) signal.removeEventListener("abort", onAbort);
78
97
  // Awaited, and the whole point: on the timeout path a thread is still burning a core, and
79
98
  // resolving before it is gone would report an evaluation the machine is still running.
80
99
  await worker.terminate();
package/import.ts CHANGED
@@ -8,9 +8,10 @@
8
8
  // over the same project.
9
9
 
10
10
  import { spawn } from "node:child_process";
11
+ import { existsSync } from "node:fs";
11
12
  import { access, mkdir, writeFile } from "node:fs/promises";
12
13
  import { builtinModules } from "node:module";
13
- import { dirname, join, relative } from "node:path";
14
+ import { dirname, join, relative, resolve } from "node:path";
14
15
  import { fileURLToPath } from "node:url";
15
16
 
16
17
  import commonjs from "@rollup/plugin-commonjs";
@@ -22,20 +23,37 @@ import ts from "typescript";
22
23
  type Schema = Record<string, any>;
23
24
 
24
25
  type Site = {
25
- resolver: string;
26
- process: string;
26
+ // Which namespace the directive sits in: `process`, `task` or `action`.
27
+ level: "process" | "task" | "action";
27
28
  task?: string;
28
- pointer: string;
29
- path: string;
30
- input?: Schema;
31
- output?: Schema;
29
+ // What the site IS, at action level: the action's type, and the process a child call names.
30
+ action?: string;
31
+ child?: string;
32
+ // The slot as a path of keys and indices, so nothing has to unescape a JSON Pointer.
33
+ pointer: (string | number)[];
34
+ // Everything after `$<resolver>:`, verbatim. genctl does not read it as a path — that is
35
+ // this resolver's reading of it.
36
+ argument: string;
37
+ // What .genroc asked genctl to type, keyed by the name it chose. This resolver reads Input
38
+ // and Output; which ADDRESS each came from is the config's business, not ours.
39
+ types?: Record<string, Schema>;
40
+ };
41
+
42
+ type ManifestProcess = {
43
+ name: string;
44
+ // The definition's own location, split because a relative argument is relative to the
45
+ // DIRECTORY — joining is ours to do, and this is the base.
46
+ dir: string;
47
+ file: string;
48
+ sites: Site[];
49
+ // Only what this process's fragments reach — a `$ref` in `types` points here.
50
+ $defs?: Record<string, Schema>;
32
51
  };
33
52
 
34
53
  type Manifest = {
35
54
  mode: "types" | "build";
36
55
  root: string;
37
- schemas: Record<string, Schema>;
38
- sites: Site[];
56
+ processes: ManifestProcess[];
39
57
  };
40
58
 
41
59
  function die(message: string): never {
@@ -150,25 +168,14 @@ function deref(
150
168
  return cur;
151
169
  }
152
170
 
153
- /** The manifest's `input` is the type of the whole ACTION input — for /eval that is
154
- * `{code, input, timeout_ms, …}`, and only its `input` field is bound as the script's
155
- * parameter. genroc cannot know that; this file owns the evaluator's wire contract, so
156
- * the navigation belongs here rather than in genctl. */
157
- function scriptInput(
158
- site: Site,
159
- defs: Record<string, Schema>,
160
- ): Schema | undefined {
161
- const action = deref(site.input, defs);
162
- const bound = action?.properties?.input;
163
- return bound ?? site.input;
164
- }
165
-
166
171
  /** Emits one named type per reachable $def rather than inlining: a task output may
167
172
  * reference itself (specs/recursive-type-inference.md) and inlining would not terminate. */
168
- function declarations(site: Site, defs: Record<string, Schema>): string {
173
+ function declarations(at: Located): string {
174
+ const { site, where } = at;
175
+ const defs = where.$defs ?? {};
169
176
  const used = new Set<string>();
170
- const input = tsType(scriptInput(site, defs), used);
171
- const output = tsType(site.output, used);
177
+ const input = tsType(site.types?.Input, used);
178
+ const output = tsType(site.types?.Output, used);
172
179
 
173
180
  const emitted: string[] = [];
174
181
  const done = new Set<string>();
@@ -182,7 +189,7 @@ function declarations(site: Site, defs: Record<string, Schema>): string {
182
189
 
183
190
  return [
184
191
  "// Generated by genroc. Do not edit - regenerate with `genctl types`.",
185
- `// ${site.process}${site.task ? ` / ${site.task}` : ""} (${site.pointer})`,
192
+ `// ${where.name} (${address(site.pointer)})`,
186
193
  "",
187
194
  ...emitted,
188
195
  emitted.length ? "" : "",
@@ -215,7 +222,7 @@ async function nearestTsconfig(
215
222
  }
216
223
  }
217
224
 
218
- async function typecheck(root: string, sites: Site[]): Promise<void> {
225
+ async function typecheck(sites: Located[]): Promise<void> {
219
226
  // NOT `.genroc`: that is the project config FILE, and a directory of the same name cannot
220
227
  // coexist with it. The suffix is what keeps the scratch area out of its way.
221
228
  const dir = join(root, ".genroc-cache");
@@ -223,12 +230,12 @@ async function typecheck(root: string, sites: Site[]): Promise<void> {
223
230
 
224
231
  // One tsc per distinct base config: `extends` takes a single base, so merging two would
225
232
  // check each script under the other author's options.
226
- const groups = new Map<string, Site[]>();
227
- for (const site of sites) {
228
- const base = (await nearestTsconfig(dirname(site.path), root)) ?? "";
233
+ const groups = new Map<string, Located[]>();
234
+ for (const at of sites) {
235
+ const base = (await nearestTsconfig(dirname(at.file), root)) ?? "";
229
236
  const group = groups.get(base);
230
- if (group) group.push(site);
231
- else groups.set(base, [site]);
237
+ if (group) group.push(at);
238
+ else groups.set(base, [at]);
232
239
  }
233
240
 
234
241
  let n = 0;
@@ -251,8 +258,8 @@ async function typecheck(root: string, sites: Site[]): Promise<void> {
251
258
  ...(base ? {} : { types: [] }),
252
259
  },
253
260
  files: group.flatMap((s) => [
254
- relative(dir, s.path),
255
- relative(dir, typesPathFor(s.path)),
261
+ relative(dir, s.file),
262
+ relative(dir, typesPathFor(s.file)),
256
263
  ]),
257
264
  // `files` overrides the base's, but a base `include` survives beside it and would
258
265
  // drag the author's whole tree in, to be checked under the worker lib.
@@ -351,15 +358,16 @@ function tsResolve(configPath: string | null): Plugin {
351
358
  /** Bundles to a self-contained ES module, which is what the evaluator imports: the default
352
359
  * export it calls is the author's own, so nothing wraps or rewrites the code between the two.
353
360
  * Bundling is entirely the importer's job, so a definition version pins its code forever. */
354
- async function bundle(site: Site, root: string): Promise<string> {
361
+ async function bundle(at: Located): Promise<string> {
362
+ const site = at.site;
355
363
  // Builtins are EXTERNALISED as imports the realm resolves natively. Anything else
356
364
  // unresolved is a REFUSAL, not an external: rollup's default is to leave it as an import
357
365
  // of a module that will not be there, which bundles clean and fails at runtime.
358
366
  const built = await rollup({
359
- input: site.path,
367
+ input: at.file,
360
368
  external: (id) => BUILTIN.has(id),
361
369
  plugins: [
362
- tsResolve(await nearestTsconfig(dirname(site.path), root)),
370
+ tsResolve(await nearestTsconfig(dirname(at.file), root)),
363
371
  nodeResolve({ extensions: [".ts", ".tsx", ".mjs", ".js", ".json"] }),
364
372
  commonjs(),
365
373
  // A `.json` import is a data file inlined at build time, which the previous bundler
@@ -370,12 +378,12 @@ async function bundle(site: Site, root: string): Promise<string> {
370
378
  onwarn(warning) {
371
379
  if (warning.code === "UNRESOLVED_IMPORT") {
372
380
  die(
373
- `${site.path}: cannot resolve ${warning.exporter ?? "an import"} — is it installed?`,
381
+ `${at.file}: cannot resolve ${warning.exporter ?? "an import"} — is it installed?`,
374
382
  );
375
383
  }
376
384
  },
377
385
  }).catch((e: unknown) =>
378
- die(`${site.path}: ${e instanceof Error ? e.message : String(e)}`),
386
+ die(`${at.file}: ${e instanceof Error ? e.message : String(e)}`),
379
387
  );
380
388
 
381
389
  const { output } = await built.generate({
@@ -386,7 +394,7 @@ async function bundle(site: Site, root: string): Promise<string> {
386
394
  // Refused here rather than in the realm: the evaluator can only report it against a running
387
395
  // instance, and the file it names is on this machine.
388
396
  if (!output[0].exports.includes("default")) {
389
- die(`${site.path}: a script must \`export default\` the function to run`);
397
+ die(`${at.file}: a script must \`export default\` the function to run`);
390
398
  }
391
399
  return output[0].code;
392
400
  }
@@ -401,45 +409,99 @@ const stdin: string = await new Promise((resolve, reject) => {
401
409
  process.stdin.on("error", reject);
402
410
  });
403
411
  const manifest = JSON.parse(stdin) as Manifest;
404
- if (!manifest || !Array.isArray(manifest.sites))
412
+ // genctl runs a resolver with the project root as its cwd, so the manifest need not say so.
413
+ const root = process.cwd();
414
+ if (!manifest || !Array.isArray(manifest.processes))
405
415
  die("stdin is not a genroc resolver manifest");
406
416
 
417
+ /** Every site with the process it sits in and the file its argument names, in the manifest's own
418
+ * order — which is the order `code` must answer in. genctl passes the argument verbatim, so
419
+ * joining it to the definition's directory is this resolver's business: only it knows the
420
+ * argument is a path at all. */
421
+ type Located = { site: Site; where: ManifestProcess; file: string };
422
+
423
+ /** A pointer as the address it is, so a generated comment or an error can be pasted into
424
+ * `genctl schema`. A key no identifier can spell is bracketed, as the grammar spells it. */
425
+ function address(pointer: (string | number)[]): string {
426
+ return pointer
427
+ .map((seg) =>
428
+ typeof seg === "string" && /^[A-Za-z_][A-Za-z0-9_]*$/.test(seg)
429
+ ? `.${seg}`
430
+ : `[${JSON.stringify(seg)}]`,
431
+ )
432
+ .join("")
433
+ .replace(/^\./, "");
434
+ }
435
+ const located: Located[] = manifest.processes.flatMap((where) =>
436
+ where.sites.map((site) => ({
437
+ site,
438
+ where,
439
+ file: resolve(where.dir, site.argument),
440
+ })),
441
+ );
442
+
443
+ // genctl is agnostic about what a script is for; the contract that an evaluation request carries
444
+ // its module in `code` is THIS resolver's, so it is the one that checks a directive landed there.
445
+ // `child` is the shape the scaffold uses — a call to a process that forwards to the evaluator —
446
+ // and `external` is the same request made directly.
447
+ for (const at of located) {
448
+ // Two halves, from the two places that carry them: the action's KIND is a field, and WHERE in
449
+ // it the directive sits is the pointer. A child call is the shape the scaffold generates — to a
450
+ // process that forwards to the evaluator — and an external task is the same request made
451
+ // directly.
452
+ const kind = at.site.level === "action" ? (at.site.action ?? "") : "";
453
+ const slot = at.site.pointer.slice(-2).join(".");
454
+ if ((kind !== "child" && kind !== "external") || slot !== "input.code") {
455
+ die(
456
+ `${join(at.where.dir, at.where.file)}: ${address(at.site.pointer)}: an evaluated script ` +
457
+ "belongs in the `code` field of a child or external task's input, and this is " +
458
+ `${kind ? `a ${kind} task's ` : ""}\`${slot}\`.`,
459
+ );
460
+ }
461
+ }
462
+
463
+ // genctl no longer stats the argument — it does not know it is a file — so the resolver that
464
+ // does is the one that must say when it is not there.
465
+ for (const at of located) {
466
+ if (!existsSync(at.file)) {
467
+ die(
468
+ `${join(at.where.dir, at.where.file)}: ${address(at.site.pointer)}: ` +
469
+ `"$import: ${at.site.argument}" names no file (looked at ${at.file})`,
470
+ );
471
+ }
472
+ }
473
+
407
474
  // One script at two sites with different input types is a refusal, not a union: the union
408
475
  // is sound and would typecheck a body that is wrong at one of the sites.
409
- const byPath = new Map<string, Site>();
410
- for (const site of manifest.sites) {
411
- const seen = byPath.get(site.path);
412
- const defsOf = (x: Site) =>
413
- (manifest.schemas[x.process]?.$defs ?? {}) as Record<string, Schema>;
476
+ const byPath = new Map<string, Located>();
477
+ for (const at of located) {
478
+ const seen = byPath.get(at.file);
414
479
  if (
415
480
  seen &&
416
- JSON.stringify(scriptInput(seen, defsOf(seen))) !==
417
- JSON.stringify(scriptInput(site, defsOf(site)))
481
+ JSON.stringify(seen.site.types) !== JSON.stringify(at.site.types)
418
482
  ) {
419
483
  die(
420
- `${site.path} is imported at ${seen.pointer} and ${site.pointer} with different input types.\n` +
484
+ `${at.file} is imported at ${address(seen.site.pointer)} and ` +
485
+ `${address(at.site.pointer)} with different input types.\n` +
421
486
  "Split it into two scripts, or make the two call sites pass the same shape.",
422
487
  );
423
488
  }
424
- byPath.set(site.path, site);
489
+ byPath.set(at.file, at);
425
490
  }
426
491
 
427
- for (const site of byPath.values()) {
428
- const defs = (manifest.schemas[site.process]?.$defs ?? {}) as Record<
429
- string,
430
- Schema
431
- >;
432
- await write(typesPathFor(site.path), declarations(site, defs));
492
+ for (const at of byPath.values()) {
493
+ await write(typesPathFor(at.file), declarations(at));
433
494
  }
434
495
 
435
496
  if (manifest.mode === "types") {
497
+ console.log(JSON.stringify(manifest, null, 2));
436
498
  process.exit(0);
437
499
  }
438
500
 
439
- await typecheck(manifest.root, [...byPath.values()]);
501
+ await typecheck([...byPath.values()]);
440
502
 
441
503
  const code: string[] = [];
442
- for (const site of manifest.sites) {
443
- code.push(await bundle(site, manifest.root));
504
+ for (const at of located) {
505
+ code.push(await bundle(at));
444
506
  }
445
507
  process.stdout.write(JSON.stringify({ code }));
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@genroc/eval-node",
3
- "version": "0.0.0-edge.ea911c5",
3
+ "version": "0.0.0-edge.eadd43a",
4
4
  "type": "module",
5
5
  "engines": {
6
6
  "node": ">=24"
package/worker.ts CHANGED
@@ -6,7 +6,7 @@
6
6
  // See README.md for the contract, and specs/external-task-queue.md for the queue itself.
7
7
 
8
8
  import { readFileSync } from "node:fs";
9
- import { evaluate, type EvalRequest, type FailureKind } from "./eval.ts";
9
+ import { Cancelled, evaluate, type EvalRequest, type FailureKind } from "./eval.ts";
10
10
 
11
11
  const SERVER = (process.env.GENROC_SERVER ?? "http://localhost:8448").replace(/\/$/, "");
12
12
  const WORKER_ID = process.env.WORKER_ID ?? `evaluator-${process.pid}`;
@@ -135,7 +135,12 @@ function asEvalRequest(input: unknown): EvalRequest | string {
135
135
  };
136
136
  }
137
137
 
138
- const inFlight = new Map<string, QueueTask>();
138
+ /** A claim this worker is currently serving. The controller is how the renewal loop reaches
139
+ * the evaluation: genroc cannot call us, so a cancellation arrives as an answer to our own
140
+ * heartbeat and has to be delivered inward from there. */
141
+ type Running = { job: QueueTask; abort: AbortController };
142
+
143
+ const inFlight = new Map<string, Running>();
139
144
  let running = true;
140
145
 
141
146
  // Whether the last claim reached genroc. A worker polls several times a second, so an
@@ -203,7 +208,7 @@ async function answer(token: string, outcome: Record<string, unknown>): Promise<
203
208
  await release(token);
204
209
  }
205
210
 
206
- async function run(job: QueueTask): Promise<void> {
211
+ async function run(job: QueueTask, signal: AbortSignal): Promise<void> {
207
212
  let resolved: unknown;
208
213
  try {
209
214
  resolved = await resolveObjects(job);
@@ -224,8 +229,16 @@ async function run(job: QueueTask): Promise<void> {
224
229
 
225
230
  let result;
226
231
  try {
227
- result = await evaluate(req);
232
+ result = await evaluate(req, signal);
228
233
  } catch (err) {
234
+ // Cancelled is not a fault: the process was stopped while this ran. The claim still goes
235
+ // back — the row is terminal, so nothing re-claims it, and the release is what stops it
236
+ // waiting out a lease nobody is serving.
237
+ if (err instanceof Cancelled) {
238
+ console.log(`cancelled: ${job.token}`);
239
+ await release(job.token);
240
+ return;
241
+ }
229
242
  // The RUNNER faulted, not the script — the one class where a retry can help. There is no
230
243
  // error code for it on purpose: releasing the claim is how a queue spells "retryable", and
231
244
  // it puts the task in front of a different worker instead of burning the definition's
@@ -263,11 +276,22 @@ async function renewLoop(): Promise<void> {
263
276
  tokens,
264
277
  lease_ms: LEASE_MS,
265
278
  });
266
- // A short count means a claim lapsed and was taken over. Nothing to do about it — the work
267
- // continues and its answer will be refused — but say so, because it is the signal that
268
- // LEASE_MS is too short for what these scripts actually take.
269
- if (ok && data?.renewed < tokens.length) {
270
- console.error(`renewed ${data.renewed}/${tokens.length} claims; a lease lapsed under load`);
279
+ if (!ok) continue;
280
+
281
+ // Cancelled: the process was stopped, so the evaluation is abandoned and the claim handed
282
+ // back. Aborting first is the point of the list — waiting for the script to finish would
283
+ // keep burning a core on work an operator has already stopped. The release rides run()'s
284
+ // own catch, so nothing is released twice.
285
+ for (const token of (data?.cancelled ?? []) as string[]) {
286
+ inFlight.get(token)?.abort.abort();
287
+ }
288
+ // Lost: someone else holds this claim now. The work continues and its answer will be
289
+ // refused — that is not fixable here — but say so, because it is the signal that LEASE_MS
290
+ // is too short for what these scripts actually take. Deliberately NOT released: the claim
291
+ // is the new holder's, and releasing would bump the epoch out from under it.
292
+ const lost = (data?.lost ?? []) as string[];
293
+ if (lost.length) {
294
+ console.error(`lost ${lost.length}/${tokens.length} claims; a lease lapsed under load`);
271
295
  }
272
296
  }
273
297
  }
@@ -277,8 +301,9 @@ async function pollLoop(): Promise<void> {
277
301
  const free = CONCURRENCY - inFlight.size;
278
302
  const jobs = free > 0 ? await claim(free) : [];
279
303
  for (const job of jobs) {
280
- inFlight.set(job.token, job);
281
- void run(job).finally(() => inFlight.delete(job.token));
304
+ const abort = new AbortController();
305
+ inFlight.set(job.token, { job, abort });
306
+ void run(job, abort.signal).finally(() => inFlight.delete(job.token));
282
307
  }
283
308
  // Only idle when there was nothing to take: a full queue should be drained at the speed the
284
309
  // realms allow, not at the poll interval.