如何实现Neo4j自定义过程的迭代式流式结果返回？

问题核心 #

编写Neo4j自定义过程时，不想先把所有结果塞进List再转成流，希望直接迭代返回结果以减少内存占用，但尝试的代码触发了NotInTransactionException错误。

错误原因 #

你写的bar()方法用try-with-resources包裹事务，会导致事务在方法返回流时就被自动关闭——try-with-resources会在代码块执行完毕（即return语句执行后）立即关闭资源（这里是事务），但流是延迟执行的，后续遍历流时自然找不到有效事务，因此报错。

正确解法 #

Neo4j自定义过程不需要手动管理事务：当你给@Procedure标记mode为READ或WRITE时，Neo4j会自动为过程调用创建并维护事务。只需通过上下文注入的GraphDatabaseService直接操作即可，它会自动绑定到当前事务。

正确代码示例 #

import org.neo4j.graphdb.*;
import org.neo4j.procedure.*;

import java.util.Spliterators;
import java.util.stream.Stream;
import java.util.stream.StreamSupport;

public class MyProcedures {

    @Context
    public GraphDatabaseService db;

    @Procedure(value = "bar", mode = Mode.READ)
    public Stream<MyLong> bar() {
        // 直接使用上下文注入的db，自动关联Neo4j管理的事务
        ResourceIterator<Node> iterator = db.findNodes(Label.label("A"));
        
        // 转换迭代器为流，并绑定迭代器关闭操作防止资源泄漏
        return StreamSupport.stream(Spliterators.spliteratorUnknownSize(iterator, Spliterator.ORDERED), false)
                .map(node -> new MyLong((Long) node.getProperty("a")))
                .onClose(iterator::close);
    }

    // 输出结果封装类
    public static class MyLong {
        public final long value;

        public MyLong(long value) {
            this.value = value;
        }
    }
}

关键注意点 #

• 用@Context注入GraphDatabaseService，无需手动开启/提交事务，Neo4j会根据过程执行结果自动处理事务（成功则提交，异常则回滚）。
• 转换ResourceIterator为流时，必须通过onClose()绑定迭代器的关闭操作，避免数据库资源泄漏。
• 流会按需遍历节点，不会一次性把所有结果加载到内存，完美实现迭代返回的需求。

验证效果 #

调用call bar() yield value with value return value limit 3时，Neo4j会：

1. 自动开启读事务
2. 按需遍历Label("A")的节点，返回前3个结果
3. 流遍历完成后关闭迭代器，自动提交事务

内容的提问来源于stack exchange，提问作者Antoine