Class Graph

packages/graph/src/Graph.ts:307–727

}

export class Graph<TSchema extends GraphSchema> implements GraphSource<TSchema> {
  #config: GraphConfig<TSchema>;

  #vertexIdentities: WeakMap<StoredVertex, Vertex<TSchema, any>>;
  #edgeIdentities: WeakMap<StoredEdge, Edge<TSchema, any>>;
  #indexManager: IndexManager<TSchema>;

  public constructor(config: GraphConfig<TSchema>) {
    this.#config = config;
    this.#vertexIdentities = new WeakMap();
    this.#edgeIdentities = new WeakMap();
    this.#indexManager = new IndexManager(config.schema);
  }

  /**
   * The index manager for the graph.
   * Use this to access indexes for query optimization.
   */
  public get indexManager(): IndexManager<TSchema> {
    return this.#indexManager;
  }

  /**
   * The schema of the graph.
   */
  public get schema(): TSchema {
    return this.#config.schema;
  }

  /**
   * The storage of the graph.
   */
  public get storage(): GraphStorage {
    return this.#config.storage;
  }
  /**
   * Generate a new unique identifier for an element.
   */
  public generateElementId(label: string): ElementId {
    const generateId = this.#config.generateId ?? defaultIdGenerator;
    return `${label}:${generateId()}`;
  }

  /**
   * Get an element by its id.
   * @param id The id of the element.
   * @returns The element.
   */
  public getElementById<TLabel extends VertexLabel<TSchema>>(
    id: ElementId<TLabel>,
  ): Vertex<TSchema, TLabel>;
  public getElementById<TLabel extends EdgeLabel<TSchema>>(
    id: ElementId<TLabel>,
  ): Edge<TSchema, TLabel>;
  public getElementById<TLabel extends AnyLabel<TSchema>>(id: ElementId<TLabel>) {
    const label = getLabelFromElementId<TLabel>(id);
    if (label in this.#config.schema.vertices) {
      return this.getVertexById<TLabel>(id);