Compare commits
3 Commits
d78b7dabd5
...
e979f0233a
| Author | SHA1 | Date |
|---|---|---|
Laurin Weger
|
e979f0233a | 21 hours ago |
|
Laurin Weger
|
b4bcbecaaf | 21 hours ago |
|
Laurin Weger
|
615f872663 | 1 day ago |
@ -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