File tree Expand file tree Collapse file tree 1 file changed +7
-4
lines changed
progress-tracking/src/main/java/org/neo4j/gds/mem Expand file tree Collapse file tree 1 file changed +7
-4
lines changed Original file line number Diff line number Diff 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 (){
You can’t perform that action at this time.
0 commit comments