Spring/spring-data-neo4j
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

closing results for eager iterations

Browse Source
This commit is contained in:
Michael Hunger
2011-04-07 13:57:00 +02:00
parent 478b8bb009
commit 462fe24f3b
1 changed files with 21 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
spring-data-neo4j/src/main/java/org/springframework/data/graph/neo4j/template/PathMappingIterator.java
View File

@@ -17,6 +17,7 @@
package org.springframework.data.graph.neo4j.template;
import org.neo4j.graphdb.Path;
import org.neo4j.helpers.collection.ClosableIterable;
import org.neo4j.helpers.collection.IterableWrapper;
import java.util.ArrayList;
@@ -37,16 +38,24 @@ public class PathMappingIterator {
case EAGER:
case EAGER_STOP_ON_NULL:
List<T> result = new ArrayList<T>();
for (Path path : paths) {
T mapped = pathMapper.mapPath(path);
if (mapped == null && mode == EAGER_STOP_ON_NULL) break;
result.add(mapped);
try {
for (Path path : paths) {
T mapped = pathMapper.mapPath(path);
if (mapped == null && mode == EAGER_STOP_ON_NULL) break;
result.add(mapped);
}
} finally {
close(paths);
}
return result;
case EAGER_IGNORE_RESULTS:
for (Path path : paths) {
pathMapper.mapPath(path);
try {
for (Path path : paths) {
pathMapper.mapPath(path);
}
} finally {
close(paths);
}
return null;
case LAZY:
@@ -61,6 +70,12 @@ public class PathMappingIterator {
}
}
private void close(Iterable<Path> paths) {
if (paths instanceof ClosableIterable) {
((ClosableIterable)paths).close();
}
}
private <T> IterationController.IterationMode getIterationControl(PathMapper<T> pathMapper) {
if (pathMapper instanceof IterationController) {
IterationController.IterationMode result = ((IterationController) pathMapper).getIterationMode();