Skip to content

Commit 51e23eb

Browse files
fix
1 parent 94f4e09 commit 51e23eb

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

progress-tracking/src/main/java/org/neo4j/gds/mem/GraphStoreMemoryContainer.java

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -44,10 +44,13 @@ long addGraph(GraphStoreAddedEvent graphStoreAddedEvent){
4444
}
4545

4646
long removeGraph(GraphStoreRemovedEvent graphStoreRemovedEvent){
47-
var graphMemoryToRemove = graphStoreRemovedEvent.memoryInBytes();
48-
var graphsMemoryAfterRemoval = graphStoreReservedMemory.addAndGet(-graphMemoryToRemove);
49-
graphStoresMemory.get(graphStoreRemovedEvent.user()).remove(graphStoreRemovedEvent.graphName());
50-
return graphsMemoryAfterRemoval;
47+
//TODO: for more effective progress tracking
48+
//var graphMemoryToRemove = graphStoreRemovedEvent.memoryInBytes();
49+
var graphMemoryToRemove= graphStoresMemory.get(graphStoreRemovedEvent.user()).remove(graphStoreRemovedEvent.graphName());
50+
if (graphMemoryToRemove==null) {
51+
return graphStoreReservedMemory.get();
52+
}
53+
return graphStoreReservedMemory.addAndGet(-graphMemoryToRemove);
5154
}
5255

5356
long graphStoreReservedMemory(){

0 commit comments

Comments
 (0)