Class YGraphStorage

packages/y-graph-storage/src/YGraphStorage.ts:30–419

export const $OutEKey = makeInternalKey("outE");

export class YGraphStorage implements GraphStorage {
  #doc: Y.Doc;
  #config: YGraphStorageConfig;
  #vertexIdentities: WeakMap<Y.Map<unknown>, StoredVertex>;
  #edgeIdentities: WeakMap<Y.Map<unknown>, StoredEdge>;

  public constructor(doc: Y.Doc, config: YGraphStorageConfig) {
    this.#doc = doc;
    this.#config = config;
    this.#vertexIdentities = new WeakMap();
    this.#edgeIdentities = new WeakMap();
  }

  public get doc(): Y.Doc {
    return this.#doc;
  }

  public getVertexCollectionMap(label: string): Y.Map<Y.Map<unknown>> {
    if (!(label in this.#config.schema.vertices)) {
      throw new LabelNotFoundError(label);
    }
    return this.#doc.getMap(`V:${label}`);
  }

  public getEdgeCollectionMap(label: string): Y.Map<Y.Map<unknown>> {
    if (!(label in this.#config.schema.edges)) {
      throw new LabelNotFoundError(label);
    }
    return this.#doc.getMap(`E:${label}`);
  }

  protected getVertexProperties(label: string, uuid: string, map: Y.Map<unknown>): object {
    const schema = this.#config.schema.vertices[label];
    if (schema == null) {
      throw new LabelNotFoundError(label);
    }
    return createLazyPropertyDictionary(schema, map);
  }

  protected getEdgeProperties(label: string, uuid: string, map: Y.Map<unknown>): object {
    const schema = this.#config.schema.edges[label];
    if (schema == null) {
      throw new LabelNotFoundError(label);
    }
    return createLazyPropertyDictionary(schema, map);
  }

  public getVertexById(id: ElementId): StoredVertex | undefined {
    const [label, uuid] = parseElementId(id);
    const collection = this.getVertexCollectionMap(label);
    const data = collection.get(uuid);
    if (data == null) {
      return undefined;
    }
    const existing = this.#vertexIdentities.get(data);
    if (existing != null) {
      return existing;
    }