Laurin Weger
4 hours ago
parent
615f872663
commit
b4bcbecaaf
@ -0,0 +1,217 @@ |
||||
import type { Diff as Patches, Scope } from "../types.ts"; |
||||
import { applyDiff } from "./applyDiff.ts"; |
||||
|
||||
import * as NG from "@ng-org/lib-wasm"; |
||||
|
||||
import { |
||||
deepSignal, |
||||
watch as watchDeepSignal, |
||||
batch, |
||||
} from "@ng-org/alien-deepsignals"; |
||||
import type { |
||||
DeepPatch, |
||||
DeepSignalObject, |
||||
WatchPatchCallback, |
||||
WatchPatchEvent, |
||||
} from "@ng-org/alien-deepsignals"; |
||||
import type { ShapeType, BaseType } from "@ng-org/shex-orm"; |
||||
|
||||
export class OrmConnection<T extends BaseType> { |
||||
// TODO: WeakMaps?
|
||||
private static idToEntry = new Map<string, OrmConnection<any>>(); |
||||
|
||||
private ng: typeof NG; |
||||
readonly shapeType: ShapeType<T>; |
||||
readonly scope: Scope; |
||||
readonly signalObject: DeepSignalObject<T | {}>; |
||||
private refCount: number; |
||||
/*** Identifier as a combination of shape type and scope. Prevents duplications. */ |
||||
private identifier: string; |
||||
ready: boolean; |
||||
sessionId: number; |
||||
suspendDeepWatcher: boolean; |
||||
readyPromise: Promise<void>; |
||||
// Promise that resolves once initial data has been applied.
|
||||
resolveReady!: () => void; |
||||
|
||||
// FinalizationRegistry to clean up connections when signal objects are GC'd.
|
||||
private static cleanupSignalRegistry = |
||||
typeof FinalizationRegistry === "function" |
||||
? new FinalizationRegistry<string>((connectionId) => { |
||||
// Best-effort fallback; look up by id and clean
|
||||
const entry = this.idToEntry.get(connectionId); |
||||
if (!entry) return; |
||||
entry.release(); |
||||
}) |
||||
: null; |
||||
|
||||
private constructor(shapeType: ShapeType<T>, scope: Scope, ng: typeof NG) { |
||||
this.shapeType = shapeType; |
||||
this.scope = scope; |
||||
this.ng = ng; |
||||
this.refCount = 0; |
||||
this.ready = false; |
||||
this.suspendDeepWatcher = false; |
||||
this.identifier = `${shapeType.shape}::${canonicalScope(scope)}`; |
||||
this.signalObject = deepSignal<T | {}>(new Set(), { |
||||
addIdToObjects: true, |
||||
idGenerator: this.generateSubjectIri, |
||||
}); |
||||
|
||||
// TODO:
|
||||
this.sessionId = 1; |
||||
|
||||
// Schedule cleanup of the connection when the signal object is GC'd.
|
||||
OrmConnection.cleanupSignalRegistry?.register( |
||||
this.signalObject, |
||||
this.identifier, |
||||
this.signalObject |
||||
); |
||||
|
||||
// Add listener to deep signal object to report changes back to wasm land.
|
||||
watchDeepSignal(this.signalObject as T, this.onSignalObjectUpdate); |
||||
|
||||
// Initialize per-entry readiness promise that resolves in setUpConnection
|
||||
this.readyPromise = new Promise<void>((resolve) => { |
||||
this.resolveReady = resolve; |
||||
}); |
||||
|
||||
// Establish connection to wasm land.
|
||||
ng.orm_start(scope, shapeType, this.sessionId, this.onBackendMessage); |
||||
} |
||||
|
||||
/** |
||||
* Get a connection which contains the ORM and lifecycle methods. |
||||
* @param shapeType |
||||
* @param scope |
||||
* @param ng |
||||
* @returns |
||||
*/ |
||||
public static getConnection<T extends BaseType>( |
||||
shapeType: ShapeType<T>, |
||||
scope: Scope, |
||||
ng: typeof NG |
||||
): OrmConnection<T> { |
||||
const scopeKey = canonicalScope(scope); |
||||
|
||||
// Unique identifier for a given shape type and scope.
|
||||
const identifier = `${shapeType.shape}::${scopeKey}`; |
||||
|
||||
// If we already have an object for this shape+scope,
|
||||
// return it and just increase the reference count.
|
||||
// Otherwise, create new one.
|
||||
const connection = |
||||
OrmConnection.idToEntry.get(identifier) ?? |
||||
new OrmConnection(shapeType, scope, ng); |
||||
|
||||
connection.refCount += 1; |
||||
|
||||
return connection; |
||||
} |
||||
|
||||
public release() { |
||||
if (this.refCount > 0) this.refCount--; |
||||
if (this.refCount === 0) { |
||||
OrmConnection.idToEntry.delete(this.identifier); |
||||
|
||||
OrmConnection.cleanupSignalRegistry?.unregister(this.signalObject); |
||||
} |
||||
} |
||||
|
||||
private onSignalObjectUpdate({ patches }: WatchPatchEvent<T>) { |
||||
if (this.suspendDeepWatcher || !this.ready || !patches.length) return; |
||||
|
||||
const ormPatches = deepPatchesToDiff(patches); |
||||
|
||||
this.ng.orm_update( |
||||
this.scope, |
||||
this.shapeType.shape, |
||||
ormPatches, |
||||
this.sessionId |
||||
); |
||||
} |
||||
|
||||
private onBackendMessage(...message: any) { |
||||
this.handleInitialResponse(message); |
||||
} |
||||
|
||||
private handleInitialResponse(...param: any) { |
||||
console.log("RESPONSE FROM BACKEND", param); |
||||
|
||||
// TODO: This will break, just provisionary.
|
||||
const wasmMessage: WasmMessage = param; |
||||
const { initialData } = wasmMessage; |
||||
|
||||
// Assign initial data to empty signal object without triggering watcher at first.
|
||||
this.suspendDeepWatcher = true; |
||||
batch(() => { |
||||
// Convert arrays to sets and apply to signalObject (we only have sets but can only transport arrays).
|
||||
Object.assign(this.signalObject, recurseArrayToSet(initialData)!); |
||||
}); |
||||
|
||||
queueMicrotask(() => { |
||||
this.suspendDeepWatcher = false; |
||||
// Resolve readiness after initial data is committed and watcher armed.
|
||||
this.resolveReady?.(); |
||||
}); |
||||
|
||||
this.ready = true; |
||||
} |
||||
private onBackendUpdate(...params: any) { |
||||
// Apply diff
|
||||
} |
||||
|
||||
/** Function to create random subject IRIs for newly created nested objects. */ |
||||
private generateSubjectIri(path: (string | number)[]): string { |
||||
// Generate random string.
|
||||
let b = Buffer.alloc(33); |
||||
crypto.getRandomValues(b); |
||||
const randomString = b.toString("base64url"); |
||||
|
||||
if (path.length > 0 && path[0].toString().startsWith("did:ng:o:")) { |
||||
// If the root is a nuri, use that as a base IRI.
|
||||
let rootNuri = path[0] as string; |
||||
|
||||
return rootNuri.substring(0, 9 + 44) + ":q:" + randomString; |
||||
} else { |
||||
// Else, just generate a random IRI.
|
||||
return "did:ng:q:" + randomString; |
||||
} |
||||
} |
||||
} |
||||
|
||||
//
|
||||
//
|
||||
|
||||
function escapePathSegment(segment: string): string { |
||||
return segment.replace("~", "~0").replace("/", "~1"); |
||||
} |
||||
|
||||
export function deepPatchesToDiff(patches: DeepPatch[]): Patches { |
||||
return patches.map((patch) => { |
||||
const path = |
||||
"/" + |
||||
patch.path.map((el) => escapePathSegment(el.toString())).join("/"); |
||||
return { ...patch, path }; |
||||
}) as Patches; |
||||
} |
||||
|
||||
const recurseArrayToSet = (obj: any): any => { |
||||
if (Array.isArray(obj)) { |
||||
return new Set(obj.map(recurseArrayToSet)); |
||||
} else if (obj && typeof obj === "object") { |
||||
for (const key of Object.keys(obj)) { |
||||
obj[key] = recurseArrayToSet(obj[key]); |
||||
} |
||||
return obj; |
||||
} else { |
||||
return obj; |
||||
} |
||||
}; |
||||
|
||||
function canonicalScope(scope: Scope | undefined): string { |
||||
if (scope == null) return ""; |
||||
return Array.isArray(scope) |
||||
? scope.slice().sort().join(",") |
||||
: String(scope); |
||||
} |
||||
Loading…
Reference in new issue