You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
111 lines
3.3 KiB
111 lines
3.3 KiB
from typing import Dict, List, Any, Optional
|
|
|
|
|
|
def convert_node_to_g6_v5(neo4j_node: Dict[str, Any]) -> Dict[str, Any]:
|
|
"""
|
|
将 Neo4j 节点（含 id 和 props）转换为 G6 v5 节点格式。
|
|
|
|
:param neo4j_node: 来自 Neo4jUtil 的节点字典，必须包含 'id' 字段
|
|
:return: G6 v5 节点对象
|
|
"""
|
|
node_id = neo4j_node.get("id")
|
|
if node_id is None:
|
|
raise ValueError("节点必须包含 'id' 字段")
|
|
|
|
# 构建 data：排除 'id'，保留其他属性
|
|
data = {k: v for k, v in neo4j_node.items() if k != "id"}
|
|
|
|
# 确保有显示用的 name 或 label（G6 默认使用 data.name）
|
|
if "name" not in data and "label" not in data:
|
|
data["name"] = str(node_id)
|
|
|
|
g6_node = {
|
|
"id": node_id,
|
|
"data": data,
|
|
# 可选字段（前端可覆盖）：
|
|
# "type": "circle",
|
|
# "style": {"size": 32, "fill": "violet"},
|
|
"states": [],
|
|
"combo": None
|
|
}
|
|
return g6_node
|
|
|
|
|
|
def build_g6_graph_data(
|
|
neo4j_util: 'Neo4jUtil',
|
|
node_label: str,
|
|
rel_type: Optional[str] = None
|
|
) -> Dict[str, List[Dict[str, Any]]]:
|
|
"""
|
|
构建 G6 v5 兼容的图数据结构（nodes + edges）。
|
|
|
|
:param neo4j_util: 已连接的 Neo4jUtil 实例
|
|
:param node_label: 中心节点标签（用于获取初始节点集）
|
|
:param rel_type: 关系类型（可选，若为 None 则获取所有关系）
|
|
:return: {'nodes': [...], 'edges': [...]}
|
|
"""
|
|
# 1. 获取中心节点（指定标签的所有节点）
|
|
center_nodes = neo4j_util.find_all_nodes(node_label)
|
|
|
|
# 2. 获取所有相关关系（可按类型过滤）
|
|
relationships = neo4j_util.find_all_relationships(rel_type=rel_type)
|
|
|
|
# 3. 使用字典去重节点（key: elementId）
|
|
g6_node_map: Dict[str, Dict[str, Any]] = {}
|
|
|
|
# 添加中心节点
|
|
for node in center_nodes:
|
|
node_id = node.get("id")
|
|
if node_id:
|
|
g6_node_map[node_id] = convert_node_to_g6_v5(node)
|
|
|
|
# 4. 处理关系，提取 source/target 节点并构建边
|
|
g6_edges: List[Dict[str, Any]] = []
|
|
|
|
for rel in relationships:
|
|
source_node = rel.get("source")
|
|
target_node = rel.get("target")
|
|
|
|
if not source_node or not target_node:
|
|
continue
|
|
|
|
source_id = source_node.get("id")
|
|
target_id = target_node.get("id")
|
|
|
|
if not source_id or not target_id:
|
|
continue
|
|
|
|
# 自动加入 source 和 target 节点（去重）
|
|
g6_node_map[source_id] = convert_node_to_g6_v5(source_node)
|
|
g6_node_map[target_id] = convert_node_to_g6_v5(target_node)
|
|
|
|
# 构建 G6 边对象
|
|
edge_data = {}
|
|
|
|
# 关系类型
|
|
rel_type_str = rel.get("type")
|
|
if rel_type_str:
|
|
edge_data["relationship"] = rel_type_str
|
|
|
|
# 合并关系属性
|
|
rel_props = rel.get("relProps") or {}
|
|
edge_data.update(rel_props)
|
|
|
|
g6_edge = {
|
|
"source": source_id,
|
|
"target": target_id,
|
|
"type": "line", # G6 默认边类型
|
|
"data": edge_data,
|
|
# "style": {}, # 可由前端统一配置
|
|
"states": []
|
|
}
|
|
g6_edges.append(g6_edge)
|
|
|
|
# 5. 组装结果
|
|
result = {
|
|
"nodes": list(g6_node_map.values()),
|
|
"edges": g6_edges
|
|
}
|
|
return result
|
|
|
|
|
|
|