如何在 D3 中通过函数复用逻辑优化 .attr() 调用

本文介绍如何使用 `selection.each()` 将重复的坐标计算逻辑封装为单次执行的函数，避免在多个 `.attr()` 中冗余运算，提升代码可维护性与运行效率。

在 D3 中，当需要为连线（）元素动态设置 x1、y1、x2、y2 属性，并基于源节点（source）和目标节点（target）位置进行几何偏移（如缩进 40 像素以避开节点圆形边界）时，直接在每个 .attr() 回调中重复计算 diffX、diffY、pathLength 和 offsetX/Y 不仅冗长，还造成不必要的性能开销——同一数据项被反复计算多达四次。

更优解是利用 D3 的 selection.each() 方法：它为每个绑定数据项（d）执行一次自定义逻辑，并允许你在该作用域内直接操作当前 dom 元素（通过 d3.select(n[i])），从而一次性完成全部属性赋值：

link.each(function(d, i, nodes) {   // ✅ 单次计算：所有中间变量只生成一次   const diffX = d.target.x - d.source.x;   const diffY = d.target.y - d.source.y;   const pathLength = Math.sqrt(diffX * diffX + diffY * diffY);   const offsetX = (diffX * 40) / pathLength;   const offsetY = (diffY * 40) / pathLength;    // ✅ 单次选择 + 批量设置属性，语义清晰且高效   d3.select(this) // 或 d3.select(nodes[i])，推荐 this（更简洁）     .attr('x1', d.source.x + offsetX)     .attr('y1', d.source.y + offsetY)     .attr('x2', d.target.x - offsetX)     .attr('y2', d.target.y - offsetY); });

? 关键点说明：this 在 each() 回调中指向当前 DOM 元素（即该 ），无需通过 nodes[i] 索引访问，更简洁可靠；所有数学运算（含平方根）仅执行一次/每条连线，显著优于原始写法中的四次重复计算；逻辑完全内聚：坐标偏移规则集中管理，后续若需调整缩进距离（如从 40 改为 nodeRadius）或增加角度校正，只需修改一处。

⚠️ 注意事项：

• 避免在 .attr() 回调中调用外部方法（如 this.myFunction(d)）导致 this 绑定丢失或上下文错误；each() 是 D3 官方推荐的“批量处理+局部更新”模式；
• 若需复用该逻辑于其他选择集（如不同类型的连线），可将其提取为独立函数：

  function updateLinkPositions(selection, radius = 40) {   selection.each(function(d) {     const dx = d.target.x - d.source.x;     const dy = d.target.y - d.source.y;     const len = Math.hypot(dx, dy); // 更简洁的写法（ES2015+）     const ox = (dx * radius) / len;     const oy = (dy * radius) / len;     d3.select(this)       .attr('x1', d.source.x + ox)       .attr('y1', d.source.y + oy)       .attr('x2', d.target.x - ox)       .attr('y2', d.target.y - oy);   }); } // 使用：updateLinkPositions(link, 35);

这种模式不仅解决了代码重复问题，更体现了 D3 “数据驱动、按需更新”的核心思想——让计算紧贴数据生命周期，让 DOM 操作聚焦于当前元素，最终实现清晰、高效、可扩展的可视化逻辑。