Compare commits
No commits in common. 'e979f0233afcfcab60e54392104e62f0160c1129' and 'd78b7dabd514a629d14a46191ee5337c5644f240' have entirely different histories.
e979f0233a
...
d78b7dabd5
@ -1,217 +0,0 @@ |
|||||||
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