apoc.export.util.ExportUtils Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of apoc-common Show documentation
Show all versions of apoc-common Show documentation
Data types package for Neo4j Procedures
package apoc.export.util;
import apoc.export.cypher.ExportFileManager;
import apoc.result.ProgressInfo;
import apoc.util.QueueBasedSpliterator;
import apoc.util.QueueUtil;
import apoc.util.Util;
import org.neo4j.graphdb.GraphDatabaseService;
import org.neo4j.procedure.TerminationGuard;
import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.ExecutorService;
import java.util.function.Consumer;
import java.util.stream.Stream;
import java.util.stream.StreamSupport;
public class ExportUtils {
private ExportUtils() {}
public static Stream getProgressInfoStream(GraphDatabaseService db,
ExecutorService executorService,
TerminationGuard terminationGuard,
String format,
ExportConfig exportConfig,
ProgressReporter reporter,
ExportFileManager cypherFileManager,
Consumer dump) {
long timeout = exportConfig.getTimeoutSeconds();
final ArrayBlockingQueue queue = new ArrayBlockingQueue<>(1000);
ProgressReporter reporterWithConsumer = reporter.withConsumer(
(pi) -> QueueUtil.put(queue, pi == ProgressInfo.EMPTY ? ProgressInfo.EMPTY : new ProgressInfo(pi).drain(cypherFileManager.getStringWriter(format), exportConfig), timeout)
);
Util.inTxFuture(null, executorService, db, tx -> {
dump.accept(reporterWithConsumer);
return true;
}, 0, _ignored -> {}, _ignored -> QueueUtil.put(queue, ProgressInfo.EMPTY, timeout));
QueueBasedSpliterator spliterator = new QueueBasedSpliterator<>(queue, ProgressInfo.EMPTY, terminationGuard, (int) timeout);
return StreamSupport.stream(spliterator, false);
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy