使用Neo4j Java Driver查询Memgraph时速度极慢的问题排查

环境背景 #

• 本地Docker部署Memgraph 2.11.0，已插入50万节点
• 技术栈：Neo4j Java Driver 5.13.0、Java 21、Quarkus应用
• 问题现象：执行MATCH (n:SomeNode) WHERE n.compoundKey IN $ids RETURN n查询，传入180个ID时，Java端耗时22秒，Memgraph Lab仅需5ms

优化方案 #

1. 统一参数传递类型 #

Memgraph对参数类型的识别可能与Neo4j存在差异，将集合转为字符串数组传递，确保Memgraph能正确匹配索引：

public List<Noad> getByKeys(Collection<String> keys) {
    try (Session session = driver.session()) {
        Map<String, Object> params = new HashMap<>();
        // 将Collection转为字符串数组
        params.put("ids", keys.toArray(new String[0]));
        final String query = """
                MATCH (n:SomeNode)
                WHERE n.compoundKey IN $ids
                RETURN n
                """;
        Result result = session.run(query, params);
        // 结果处理逻辑
        ...
    }
}

2. 改用显式事务执行 #

默认的session.run会使用自动事务，显式调用读写事务能减少额外的事务管理开销：

public List<Noad> getByKeys(Collection<String> keys) {
    try (Session session = driver.session()) {
        return session.executeWrite(tx -> {
            Map<String, Object> params = new HashMap<>();
            params.put("ids", keys.toArray(new String[0]));
            final String query = """
                    MATCH (n:SomeNode)
                    WHERE n.compoundKey IN $ids
                    RETURN n
                    """;
            Result result = tx.run(query, params);
            // 高效映射结果
            return result.stream()
                    .map(r -> r.get("n").asNode())
                    .map(node -> {
                        Noad noad = new Noad();
                        noad.setCompoundKey(node.get("compoundKey").asString());
                        // 其他字段映射
                        return noad;
                    })
                    .collect(Collectors.toList());
        });
    }
}

3. 调整Driver连接池与协议配置 #

在Quarkus的application.yml中添加连接优化配置，启用Bolt压缩并调整超时参数：

neo4j:
    pool:
      max-connection-pool-size: 50
      log-leaked-sessions: true
      connection-timeout: 30s
      idle-timeout: 10m
    uri: bolt://localhost:7687
    config:
      connection.liveness_check_timeout: 5s
      bolt.compression: true

4. 验证执行计划一致性 #

在Java代码中添加EXPLAIN前缀执行查询，对比Memgraph Lab的执行计划，确认是否都命中了n.compoundKey上的索引：

EXPLAIN MATCH (n:SomeNode) WHERE n.compoundKey IN $ids RETURN n

如果Java端执行计划未走索引，大概率是参数类型不匹配导致的，回到第一步调整。

关键说明 #

Memgraph Lab会自动优化参数格式和事务执行模式，而Neo4j Java Driver需要显式配置才能对齐这种优化效果，优先排查参数类型和事务模式这两个核心差异点。

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