From 32fe0d245fb4570cdd74e1bfe834def9cd48923f Mon Sep 17 00:00:00 2001
From: Milho001 <l.milhomemfrancochristino@uu.nl>
Date: Thu, 20 Mar 2025 12:26:50 +0000
Subject: [PATCH] chore: always allow updating query at the end of the call

---
 src/readers/queryService.ts | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/readers/queryService.ts b/src/readers/queryService.ts
index 4609e93..0973c02 100644
--- a/src/readers/queryService.ts
+++ b/src/readers/queryService.ts
@@ -7,7 +7,11 @@ import { query2Cypher } from '../utils/cypher/converter';
 import { parseCountCypherQuery, parseCypherQuery } from '../utils/cypher/queryParser';
 import { formatTimeDifference } from 'ts-common/src/logger/logger';
 import { Query2BackendQuery } from '../utils/reactflow/query2backend';
-import type { GraphQueryResultFromBackend, GraphQueryResultMetaFromBackend } from 'ts-common/src/model/webSocket/graphResult';
+import type {
+  CountQueryResultFromBackend,
+  GraphQueryResultFromBackend,
+  GraphQueryResultMetaFromBackend,
+} from 'ts-common/src/model/webSocket/graphResult';
 import { RabbitMqBroker } from 'ts-common/rabbitMq';
 import { Neo4jConnection } from 'ts-common/neo4j';
 import type { QueryCypher } from '../utils/cypher/converter/queryConverter';
@@ -190,7 +194,7 @@ export const queryServiceReader = async (frontendPublisher: RabbitMqBroker, mlPu
           const result = await queryService(ss.dbConnections[i], cypher, message.useCached);
 
           // Cache nodeCounts such that we can display differentiation for each query
-          await ums.updateQuery(headers.message.sessionData.userID, message.saveStateID, {
+          await ums.updateQuery(ss.userId, message.saveStateID, {
             ...activeQueryInfo,
             graph: {
               ...activeQueryInfo.graph,
-- 
GitLab