@fragno-dev/db 0.2.0 → 0.2.1

package/src/query/unit-of-work/unit-of-work.ts

@@ -917,7 +917,11 @@ export interface IUnitOfWork {
   getCreatedIds(): FragnoId[];
 
   // Parent-child relationships
-  restrict(): IUnitOfWork;
+  restrict(options?: { readyFor?: "mutation" | "retrieval" | "none" }): IUnitOfWork;
+
+  // Coordination for restricted UOWs
+  signalReadyForRetrieval(): void;
+  signalReadyForMutation(): void;
 
   // Reset for retry support
   reset(): void;
@@ -1275,8 +1279,15 @@ export class UnitOfWork<const TRawInput = unknown> implements IUnitOfWork {
    * Create a restricted child UOW that cannot execute phases.
    * The child shares the same operation storage but must signal readiness
    * before the parent can execute each phase.
+   *
+   * @param options.readyFor - Controls automatic readiness signaling:
+   *   - "mutation" (default): Signals ready for both retrieval and mutation immediately
+   *   - "retrieval": Signals ready for retrieval only
+   *   - "none": No automatic signaling, caller must signal manually
    */
-  restrict(): UnitOfWork<TRawInput> {
+  restrict(options?: { readyFor?: "mutation" | "retrieval" | "none" }): UnitOfWork<TRawInput> {
+    const readyFor = options?.readyFor ?? "mutation";
+
     const child = new UnitOfWork(
       this.#compiler,
       this.#executor,
@@ -1287,8 +1298,6 @@ export class UnitOfWork<const TRawInput = unknown> implements IUnitOfWork {
     );
     child.#coordinator.setAsRestricted(this, this.#coordinator);
 
-    // Share state with parent
-    child.#state = this.#state;
     child.#retrievalOps = this.#retrievalOps;
     child.#mutationOps = this.#mutationOps;
     child.#retrievalResults = this.#retrievalResults;
@@ -1301,10 +1310,13 @@ export class UnitOfWork<const TRawInput = unknown> implements IUnitOfWork {
 
     this.#coordinator.addChild(child);
 
-    // For synchronous usage (the common case), immediately signal readiness
-    // This allows services called directly from handlers to work without explicit signaling
-    child.signalReadyForRetrieval();
-    child.signalReadyForMutation();
+    // Signal readiness based on options
+    if (readyFor === "mutation" || readyFor === "retrieval") {
+      child.signalReadyForRetrieval();
+    }
+    if (readyFor === "mutation") {
+      child.signalReadyForMutation();
+    }
 
     return child;
   }
@@ -1375,7 +1387,7 @@ export class UnitOfWork<const TRawInput = unknown> implements IUnitOfWork {
   }
 
   get state(): UOWState {
-    return this.#state;
+    return this.#coordinator.parent?.state ?? this.#state;
   }
 
   get name(): string | undefined {
@@ -1502,7 +1514,9 @@ export class UnitOfWork<const TRawInput = unknown> implements IUnitOfWork {
       this.#state = "executed";
 
       if (result.success) {
-        this.#createdInternalIds = result.createdInternalIds;
+        // Mutate array in-place to preserve shared references with child UOWs
+        this.#createdInternalIds.length = 0;
+        this.#createdInternalIds.push(...result.createdInternalIds);
       }
 
       // Resolve the mutation phase promise to unblock waiting service methods
@@ -1536,9 +1550,9 @@ export class UnitOfWork<const TRawInput = unknown> implements IUnitOfWork {
    * Add a retrieval operation (used by TypedUnitOfWork)
    */
   addRetrievalOperation(op: RetrievalOperation<AnySchema>): number {
-    if (this.#state !== "building-retrieval") {
+    if (this.state !== "building-retrieval") {
       throw new Error(
-        `Cannot add retrieval operation in state ${this.#state}. Must be in building-retrieval state.`,
+        `Cannot add retrieval operation in state ${this.state}. Must be in building-retrieval state.`,
       );
     }
     this.#retrievalOps.push(op);
@@ -1550,7 +1564,7 @@ export class UnitOfWork<const TRawInput = unknown> implements IUnitOfWork {
    * Add a mutation operation (used by TypedUnitOfWork)
    */
   addMutationOperation(op: MutationOperation<AnySchema>): void {
-    if (this.#state === "executed") {
+    if (this.state === "executed") {
       throw new Error(`Cannot add mutation operation in executed state.`);
     }
     this.#mutationOps.push(op);
@@ -1565,9 +1579,9 @@ export class UnitOfWork<const TRawInput = unknown> implements IUnitOfWork {
    * @returns Array of FragnoIds in the same order as create() calls
    */
   getCreatedIds(): FragnoId[] {
-    if (this.#state !== "executed") {
+    if (this.state !== "executed") {
       throw new Error(
-        `getCreatedIds() can only be called after executeMutations(). Current state: ${this.#state}`,
+        `getCreatedIds() can only be called after executeMutations(). Current state: ${this.state}`,
       );
     }
 
@@ -1712,8 +1726,16 @@ export class TypedUnitOfWork<
     return this.#uow.executeMutations();
   }
 
-  restrict(): IUnitOfWork {
-    return this.#uow.restrict();
+  restrict(options?: { readyFor?: "mutation" | "retrieval" | "none" }): IUnitOfWork {
+    return this.#uow.restrict(options);
+  }
+
+  signalReadyForRetrieval(): void {
+    this.#uow.signalReadyForRetrieval();
+  }
+
+  signalReadyForMutation(): void {
+    this.#uow.signalReadyForMutation();
   }
 
   reset(): void {

package/src/schema/create.ts

@@ -1,5 +1,4 @@
 import { createId } from "../id";
-import { inspect } from "node:util";
 
 export type AnySchema = Schema<Record<string, AnyTable>>;
 
@@ -548,10 +547,6 @@ export class FragnoId {
   valueOf(): string {
     return this.#externalId;
   }
-
-  [inspect.custom](): string {
-    return `FragnoId { externalId: ${this.#externalId}, internalId: ${this.#internalId?.toString()} }`;
-  }
 }
 
 /**