本文旨在指导开发者如何将 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代码解释：
- MATCH (n:MyNode)-[r:MY_REL]-(): 这部分 Cypher 查询语句用于匹配节点类型为 MyNode，关系类型为 MY_REL 的图数据。你可以根据实际情况修改节点和关系的类型。
- WITH COLLECT(n) as mynodes, COLLECT(r) as myrels: COLLECT 函数用于将匹配到的所有节点收集到 mynodes 列表中，将所有关系收集到 myrels 列表中。
- 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。
- YIELD data: 将导出的 JSON 数据返回。
- 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.' });
}
};代码解释：
- 从 result.records 中提取查询结果。
- 遍历 records，将节点和关系数据分别添加到 nodes 和 links 数组中。注意，你需要根据实际情况调整提取数据的逻辑。
- 创建一个包含 nodes 和 links 数组的 graphData 对象。
- 将 graphData 对象作为 JSON 响应返回给客户端。
总结：
通过结合 APOC 库的 apoc.export.json.data 函数和 Node.js 代码，我们可以方便地将 Neo4j 查询结果转换为 D3.js 等可视化库可以使用的 Graph JSON 格式。这种方法可以简化前端开发流程，提高开发效率。请务必根据实际情况调整代码，以满足你的具体需求。
