Only poll for graph schema when we dont have too many labels/relTypes (#646)

Author: anderson4j

.changeset/curvy-geckos-bow.md

@@ -0,0 +1,5 @@
+---
+'@neo4j-cypher/query-tools': patch
+---
+
+Workaround bug: Running db.schema.visualization on db with too large schema can crash db

packages/query-tools/src/metadataPoller.ts

@@ -250,10 +250,18 @@ export class ConnectedMetadataPoller extends MetadataPoller {
     await Promise.allSettled([
       this.databases.refetch(),
       this.dataSummary.refetch(),
-      this.graphSchema.refetch(),
       ...Object.values(this.procedures).map((version) => version?.refetch()),
       ...Object.values(this.functions).map((version) => version?.refetch()),
     ]);
+    if (
+      this.dbSchema.labels &&
+      this.dbSchema.relationshipTypes &&
+      this.dbSchema.labels.length + this.dbSchema.relationshipTypes.length < 200
+    ) {
+      await this.graphSchema.refetch();
+    } else {
+      this.dbSchema.graphSchema = undefined;
+    }
     this.events.emit('schemaFetched');
   }
 

packages/query-tools/src/neo4jConnection.ts

@@ -167,7 +167,7 @@ export class Neo4jConnection {
           const result = tx.run(query, {});
           return await queryConfig.resultTransformer(result);
         },
-        { metadata: { ...METADATA_BASE, type: queryType } },
+        { timeout: 500, metadata: { ...METADATA_BASE, type: queryType } },
       );
     } finally {
       await session.close();