将 Neo4j 查询结果转换为 D3.js 可用的 Graph JSON 格式

将 Neo4j 查询结果转换为 D3.js 可用的 Graph JSON 格式

本文旨在指导开发者如何将 neo4j 数据库查询结果转换为 D3.JS 等可视化库可以使用的 Graph json 格式(包含 nodes 和 links 数组)。通过使用 APOC 库的 apoc.export.json.data 函数,我们可以将节点和关系数据分别导出为 JSON 数组,从而方便地在前端进行图形展示。文章将提供示例代码和相关注意事项,帮助你快速实现数据转换。

使用 APOC 库导出 Graph JSON 数据

Neo4j 数据库本身返回的结果格式可能不直接适用于 D3.js 等图形可视化库。为了方便前端使用,我们需要将查询结果转换为包含 nodes 和 links 数组的 Graph JSON 格式。APOC (Awesome Procedures On Cypher) 库提供了一系列强大的工具,可以帮助我们实现这一目标。

以下示例展示了如何使用 apoc.export.json.data 函数将节点和关系数据导出为 JSON 数组:

MATCH (n:MyNode)-[r:MY_REL]-() WITH COLLECT(n) as mynodes, COLLECT(r) as myrels CALL apoc.export.json.data(mynodes, myrels, NULL, {stream:True, jsonFormat: "JSON", writeNodeProperties:False}) YIELD data RETURN data

代码解释:

  1. MATCH (n:MyNode)-[r:MY_REL]-(): 这部分 Cypher 查询语句用于匹配节点类型为 MyNode,关系类型为 MY_REL 的图数据。你可以根据实际情况修改节点和关系的类型。
  2. WITH COLLECT(n) as mynodes, COLLECT(r) as myrels: COLLECT 函数用于将匹配到的所有节点收集到 mynodes 列表中,将所有关系收集到 myrels 列表中。
  3. CALL apoc.export.json.data(mynodes, myrels, null, {stream:True, jsonFormat: “JSON”, writeNodeProperties:False}): 这行代码调用 APOC 库的 apoc.export.json.data 函数,将节点和关系数据导出为 JSON 格式。
    • mynodes: 节点列表。
    • myrels: 关系列表。
    • null: 表示不导出任何属性。
    • {stream:True, jsonFormat: “JSON”, writeNodeProperties:False}: 配置选项。
      • stream:True: 启用流式输出,适用于大型数据集。
      • jsonFormat: “JSON”: 指定 JSON 格式。
      • writeNodeProperties:False: 不导出节点属性,如果需要导出节点属性,可以设置为 true。
  4. YIELD data: 将导出的 JSON 数据返回。
  5. RETURN data: 返回数据。

注意事项:

  • 确保你的 Neo4j 数据库已经安装了 APOC 库。你可以通过 Neo4j Browser 或 Neo4j Desktop 安装 APOC 库。
  • apoc.export.json.data 函数有多种配置选项,可以根据你的需求进行调整。例如,你可以指定导出的文件名、是否导出节点属性等。详细的配置选项可以参考 APOC 官方文档。
  • 上述代码将节点和关系数据分别导出为两个 JSON 数组。如果你需要将它们合并为一个包含 nodes 和 links 数组的 Graph JSON 对象,需要在 Node.js 代码中进行处理。

在 Node.js 中处理 Neo4j 查询结果

在你的 Node.js express 应用中,你可以使用 neo4j-driver 连接 Neo4j 数据库,并执行上述 Cypher 查询。然后,你需要将查询结果转换为 D3.js 可用的 Graph JSON 格式。

以下是一个示例代码:

import neo4j from 'neo4j-driver'; import {   NEO4J_PASSWORD,   NEO4J_URL,   NEO4J_USERNAME, } from '../../constants/index.js';  export const runMatchQuery = async (req, res) => {   const query = req?.body?.query;   if (query?.toLowerCase()?.includes('match')) {     const driver = neo4j.driver(       NEO4J_URL,       neo4j.auth.basic(NEO4J_USERNAME, NEO4J_PASSWORD)     );     const session = driver.session();      try {       const result = await session.run(query);       const records = result.records;        // 提取节点和关系数据       let nodes = [];       let links = [];        if (records && records.length > 0) {           const data = records[0].get("data"); // Assuming 'data' is the key in the RETURN statement           if(data && data.nodes){               nodes = data.nodes;           }           if(data && data.relationships){               links = data.relationships;           }       }         const graphData = {         nodes: nodes,         links: links,       };        return res.status(200).json(graphData);     } catch (error) {       console.log(error);       return res.status(500).json({ message: error.message });     } finally {       await session.close();       await driver.close();     }   } else {     return res.status(405).json({ message: 'Query not permitted.' });   } };

代码解释:

  1. 从 result.records 中提取查询结果。
  2. 遍历 records,将节点和关系数据分别添加到 nodes 和 links 数组中。注意,你需要根据实际情况调整提取数据的逻辑。
  3. 创建一个包含 nodes 和 links 数组的 graphData 对象。
  4. 将 graphData 对象作为 JSON 响应返回给客户端。

总结:

通过结合 APOC 库的 apoc.export.json.data 函数和 Node.js 代码,我们可以方便地将 Neo4j 查询结果转换为 D3.js 等可视化库可以使用的 Graph JSON 格式。这种方法可以简化前端开发流程,提高开发效率。请务必根据实际情况调整代码,以满足你的具体需求。

© 版权声明
THE END
喜欢就支持一下吧
点赞13 分享