antvx6-DAG

前言

DAG 是 Antv X6 官网示例，博主将他转化为 Vue 版本。
官网地址：https://x6.antv.antgroup.com/examples/showcase/practices/#dag

代码

1、contain

<!-- container.vue -->
<template>
  <div id="container"></div>
</template>
<script setup lang="ts">
import { Graph, Path, Cell } from "@antv/x6";
import { Selection } from "@antv/x6-plugin-selection";
import { register } from "@antv/x6-vue-shape";
import { onMounted } from "vue-demi";
import AlgoNode from "./algoNode.vue";

interface NodeStatus {
  id: string;
  status: "default" | "success" | "failed" | "running";
  label?: string;
}

register({
  shape: "dag-node",
  width: 180,
  height: 36,
  component: AlgoNode,
  ports: {
    groups: {
      top: {
        position: "top",
        attrs: {
          circle: {
            r: 4,
            magnet: true,
            stroke: "#C2C8D5",
            strokeWidth: 1,
            fill: "#fff",
          },
        },
      },
      bottom: {
        position: "bottom",
        attrs: {
          circle: {
            r: 4,
            magnet: true,
            stroke: "#C2C8D5",
            strokeWidth: 1,
            fill: "#fff",
          },
        },
      },
    },
  },
});

Graph.registerEdge(
  "dag-edge",
  {
    inherit: "edge",
    attrs: {
      line: {
        stroke: "#C2C8D5",
        strokeWidth: 1,
        targetMarker: null,
      },
    },
  },
  true
);

Graph.registerConnector(
  "algo-connector",
  (s, e) => {
    const offset = 4;
    const deltaY = Math.abs(e.y - s.y);
    const control = Math.floor((deltaY / 3) * 2);

    const v1 = { x: s.x, y: s.y + offset + control };
    const v2 = { x: e.x, y: e.y - offset - control };

    return Path.normalize(
      `M ${s.x} ${s.y}
       L ${s.x} ${s.y + offset}
       C ${v1.x} ${v1.y} ${v2.x} ${v2.y} ${e.x} ${e.y - offset}
       L ${e.x} ${e.y}
      `
    );
  },
  true
);

const nodeStatusList = [
  [
    {
      id: "1",
      status: "running",
    },
    {
      id: "2",
      status: "default",
    },
    {
      id: "3",
      status: "default",
    },
    {
      id: "4",
      status: "default",
    },
  ],
  [
    {
      id: "1",
      status: "success",
    },
    {
      id: "2",
      status: "running",
    },
    {
      id: "3",
      status: "default",
    },
    {
      id: "4",
      status: "default",
    },
  ],
  [
    {
      id: "1",
      status: "success",
    },
    {
      id: "2",
      status: "success",
    },
    {
      id: "3",
      status: "running",
    },
    {
      id: "4",
      status: "running",
    },
  ],
  [
    {
      id: "1",
      status: "success",
    },
    {
      id: "2",
      status: "success",
    },
    {
      id: "3",
      status: "success",
    },
    {
      id: "4",
      status: "failed",
    },
  ],
];

onMounted(() => {
  const graph: Graph = new Graph({
    container: document.getElementById("container")!,
    width: "800",
    height: "800",
    panning: {
      enabled: true,
      eventTypes: ["leftMouseDown", "mouseWheel"],
    },
    mousewheel: {
      enabled: true,
      modifiers: "ctrl",
      factor: 1.1,
      maxScale: 1.5,
      minScale: 0.5,
    },
    highlighting: {
      magnetAdsorbed: {
        name: "stroke",
        args: {
          attrs: {
            fill: "#fff",
            stroke: "#31d0c6",
            strokeWidth: 4,
          },
        },
      },
    },
    connecting: {
      snap: true,
      allowBlank: false,
      allowLoop: false,
      highlight: true,
      connector: "algo-connector",
      connectionPoint: "anchor",
      anchor: "center",
      validateMagnet({ magnet }) {
        return magnet.getAttribute("port-group") !== "top";
      },
      createEdge() {
        return graph.createEdge({
          shape: "dag-edge",
          attrs: {
            line: {
              strokeDasharray: "5 5",
            },
          },
          zIndex: -1,
        });
      },
    },
  });
  graph.use(
    new Selection({
      multiple: true,
      rubberEdge: true,
      rubberNode: true,
      modifiers: "shift",
      rubberband: true,
    })
  );
  graph.on("edge:connected", ({ edge }) => {
    debugger;
    edge.attr({
      line: {
        strokeDasharray: "",
      },
    });
  });

  graph.on("node:click", ({ e, x, y, node, view }) => {
    debugger;
  });

  graph.on("node:change:data", ({ node }) => {
    //获取连接到节点/边的输入边，即边的终点为指定节点/边的边。
    const edges = graph.getIncomingEdges(node);
    const { status } = node.getData() as NodeStatus;
    edges?.forEach((edge) => {
      if (status === "running") {
        edge.setAttrs({
          line: {
            strokeDasharray: 5,
            style: { animation: "running-line 30s infinite linear" }, // 边的动画效果显示，需要在css添加对应的动画效果
          },
        });
        // 给边添加样式，跟上面的效果一样
        // edge.attr("line/strokeDasharray", 5);
        // edge.attr("line/style/animation", 'ant-line 30s infinite linear');
      } else {
        edge.attr("line/strokeDasharray", "");
        edge.attr("line/style/animation", "");
      }
    });
  });

  // 初始化节点/边
  const init = (data: Cell.Metadata[]) => {
    const cells: Cell[] = [];
    data.forEach((item) => {
      if (item.shape === "dag-node") {
        cells.push(graph.createNode(item));
      } else {
        cells.push(graph.createEdge(item));
      }
    });
    graph.resetCells(cells);
  };

  // 显示节点状态
  const showNodeStatus = async (statusList: NodeStatus[][]) => {
    const status = statusList.shift();
    status?.forEach((item) => {
      const id = item.id;
      const status = item.status;
      const node = graph.getCellById(id);
      const data = node.getData();
      node.setData({
        ...data,
        status,
      });
    });
    setTimeout(() => {
      showNodeStatus(statusList);
    }, 3000);
  };

  fetch("../../src/data/dag.json")
    .then((response) => response.json())
    .then((data) => {
      init(data);
      showNodeStatus(nodeStatusList);
      graph.centerContent();
    });
});
</script>
<style>
@keyframes running-line {
  to {
    stroke-dashoffset: -1000;
  }
}
</style>

2、child

<!-- child -->
<template>
  <div :className="`node ${nodeInfo.status}`">
    <img :src="`${image.logo}`" alt="logo" />
    <span className="label">{{ nodeInfo.label }}</span>
    <span className="status">
      <img
        v-if="`${nodeInfo.status}` == 'success'"
        :src="`${image.success}`"
        alt="success"
      />
      <img
        v-if="`${nodeInfo.status}` == 'failed'"
        :src="`${image.failed}`"
        alt="failed"
      />
      <img
        v-if="`${nodeInfo.status}` == 'running'"
        :src="`${image.running}`"
        alt="running"
      />
    </span>
  </div>
</template>
<script setup lang="ts">
import { inject, onMounted, reactive } from "vue-demi";
import { Graph } from "@antv/x6";

const getNode = inject("getNode");
const node = getNode();
const data = node.getData();
const nodeInfo = reactive({});
Object.keys(data).forEach((key) => {
  nodeInfo[key] = data[key];
});
node.on("change:data", ({ current }) => {
  Object.keys(current).forEach((key) => {
    nodeInfo[key] = current[key];
  });
});

const image = {
  logo: "https://gw.alipayobjects.com/mdn/rms_43231b/afts/img/A*evDjT5vjkX0AAAAAAAAAAAAAARQnAQ",
  success:
    "https://gw.alipayobjects.com/mdn/rms_43231b/afts/img/A*6l60T6h8TTQAAAAAAAAAAAAAARQnAQ",
  failed:
    "https://gw.alipayobjects.com/mdn/rms_43231b/afts/img/A*SEISQ6My-HoAAAAAAAAAAAAAARQnAQ",
  running:
    "https://gw.alipayobjects.com/mdn/rms_43231b/afts/img/A*t8fURKfgSOgAAAAAAAAAAAAAARQnAQ",
};
</script>
<style scoped>
.node {
  display: flex;
  align-items: center;
  width: 100%;
  height: 100%;
  background-color: #fff;
  border: 1px solid #c2c8d5;
  border-left: 4px solid #5f95ff;
  border-radius: 4px;
  box-shadow: 0 2px 5px 1px rgba(0, 0, 0, 0.06);
}
.node img {
  width: 20px;
  height: 20px;
  flex-shrink: 0;
  margin-left: 8px;
}
.node .label {
  display: inline-block;
  flex-shrink: 0;
  width: 104px;
  margin-left: 8px;
  color: #666;
  font-size: 12px;
}
.node .status {
  flex-shrink: 0;
}
.node.success {
  border-left: 4px solid #52c41a;
}
.node.failed {
  border-left: 4px solid #ff4d4f;
}
.node.running .status img {
  animation: spin 1s linear infinite;
}
.x6-node-selected .node {
  border-color: #1890ff;
  border-radius: 2px;
  box-shadow: 0 0 0 4px #d4e8fe;
}
.x6-node-selected .node.success {
  border-color: #52c41a;
  border-radius: 2px;
  box-shadow: 0 0 0 4px #ccecc0;
}
.x6-node-selected .node.failed {
  border-color: #ff4d4f;
  border-radius: 2px;
  box-shadow: 0 0 0 4px #fedcdc;
}
.x6-edge:hover path:nth-child(2) {
  stroke: #1890ff;
  stroke-width: 1px;
}

.x6-edge-selected path:nth-child(2) {
  stroke: #1890ff;
  stroke-width: 1.5px !important;
}

@keyframes spin {
  from {
    transform: rotate(0deg);
  }
  to {
    transform: rotate(360deg);
  }
}
</style>