重构D3.js代码时出现Cannot read property 'push' of undefined错误求助

首先，你遇到的Cannot read property 'push' of undefined错误，根源是重构后的代码没有正确更新link.source和link.target的引用，和原代码的核心逻辑完全不一致。

原代码的核心逻辑 #

原代码里的这两行是关键：

link.source = nodes[link.source] || (nodes[link.source] = { name: link.source });
link.target = nodes[link.target] || (nodes[link.target] = { name: link.target });

它同时完成了两个必须的动作：

1. 检查nodes对象中是否存在对应名称的节点对象；
2. 如果不存在，就创建一个新节点对象并添加到nodes中，然后把link.source/link.target指向这个新对象；如果已存在，直接把link.source/link.target指向已有的节点对象。

这样处理后，links数组里的每个link的source和target都是节点对象的引用——这是D3 force布局正常工作的必要条件，因为force布局需要通过对象引用来关联节点和连线。

你重构代码的问题 #

你重构后的代码只在节点不存在时往nodes里添加对象，但完全漏掉了更新link.source和link.target的取值，它们还是原来的字符串（比如"Baratheon"）。当D3 force布局尝试处理这些字符串时，自然会因为无法找到对应的节点对象属性而抛出错误。

正确的重构方式 #

如果你想把逻辑拆成更清晰的if结构，需要保证最终link.source和link.target都指向nodes里的对象，比如：

links.forEach(function(link) {
  // 处理source节点
  if (!nodes[link.source]) {
    nodes[link.source] = { name: link.source };
  }
  link.source = nodes[link.source]; // 必须给link.source赋值节点对象

  // 处理target节点
  if (!nodes[link.target]) {
    nodes[link.target] = { name: link.target };
  }
  link.target = nodes[link.target]; // 必须给link.target赋值节点对象

  console.log(nodes);
});

如果习惯更简洁的现代JS写法（ES6+），也可以这样写：

links.forEach(link => {
  nodes[link.source] ??= { name: link.source };
  nodes[link.target] ??= { name: link.target };
  link.source = nodes[link.source];
  link.target = nodes[link.target];
  console.log(nodes);
});

修复后的完整代码 #

把上面的正确逻辑替换到你的完整代码中，就能正常运行了：

<!DOCTYPE html>
<meta charset="utf-8">
<style>
.node { fill: #4D00DD; stroke: #fff; stroke-width: 2px; }
.link { stroke: #777; stroke-width: 8px; }
</style>
<body>
<script src="https://d3js.org/d3.v3.min.js"></script>
<script>
var width = 1080, height = 960;
var links = [
{ source : "Baratheon", target : "Lannister"},
{ source : "Baratheon", target : "Stark"},
{ source : "Lannister", target : "Stark"},
];
var nodes = {};
// 修复后的节点解析逻辑
links.forEach(function(link) {
  if (!nodes[link.source]) {
    nodes[link.source] = { name: link.source };
  }
  link.source = nodes[link.source];

  if (!nodes[link.target]) {
    nodes[link.target] = { name: link.target };
  }
  link.target = nodes[link.target];

  console.log(nodes);
});
// add svg to our body
var svg = d3.select("body").append("svg")
.attr("width", width)
.attr("height", height);
var force = d3.layout.force()
.size([width, height])
.nodes(d3.values(nodes))
.links(links)
.on("tick", tick)
.linkDistance(600)
.start();
var link = svg.selectAll(".link")
.data(links)
.enter().append("line")
.attr("class", "link")
var node = svg.selectAll(".node")
.data(force.nodes())
.enter().append("circle")
.attr("class", "node")
.attr("r", width * 0.03);
function tick(e) {
node.attr("cx", function(d) { return d.x; })
.attr("cy", function(d) { return d.y; })
.call(force.drag);
link.attr("x1", function(d) { return d.source.x; })
.attr('y1', function(d) { return d.source.y; })
.attr('x2', function(d) { return d.target.x; })
.attr("y2", function(d) { return d.target.y; })
}
</script>
</body>
</html>

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