dag.tsx

import React, { ComponentType, useEffect, useRef } from 'react'
import { register } from '@antv/x6-react-shape'
import {
  Graph,
  Node,
  Cell,
  Path,
  Selection,
  Snapline,
  CellMetadata,
} from '@antv/x6'
import '../index.less'
import './index.less'

interface NodeStatus {
  id: string
  status: string
  label?: string
}

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',
}

export const AlgoNode: React.FC<{ node?: Node }> = React.memo(
  ({ node }) => {
    const data = node?.getData() as NodeStatus
    const { label, status = 'default' } = data

    return (
      <div className={`node ${status}`}>
        <img src={image.logo} alt="logo" />
        <span className="label">{label}</span>
        <span className="status">
          {status === 'success' && <img src={image.success} alt="success" />}
          {status === 'failed' && <img src={image.failed} alt="failed" />}
          {status === 'running' && <img src={image.running} alt="running" />}
        </span>
      </div>
    )
  },
  (_prevProps, nextProps) => {
    const node = nextProps.node
    if (node) {
      // Only re-render if the 'data' property of the node has not changed.
      return !node.hasChanged('data')
    }
    // If no node, no need to re-render.
    return true
  },
)

register({
  shape: 'dag-node',
  width: 180,
  height: 36,
  component: AlgoNode as ComponentType<{ node: Node; graph: Graph }>,
  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.parse(
      `
      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}
    `,
    ).serialize()
  },
  true,
)

const data = [
  {
    id: '1',
    shape: 'dag-node',
    x: 290,
    y: 110,
    data: {
      label: '读数据',
      status: 'success',
    },
    ports: [
      {
        id: '1-1',
        group: 'bottom',
      },
    ],
  },
  {
    id: '2',
    shape: 'dag-node',
    x: 290,
    y: 225,
    data: {
      label: '读数据',
      status: 'success',
    },
    ports: [
      {
        id: '2-1',
        group: 'top',
      },
      {
        id: '2-2',
        group: 'bottom',
      },
      {
        id: '2-3',
        group: 'bottom',
      },
    ],
  },
  {
    id: '3',
    shape: 'dag-node',
    x: 170,
    y: 350,
    data: {
      label: '读数据',
      status: 'success',
    },
    ports: [
      {
        id: '3-1',
        group: 'top',
      },
      {
        id: '3-2',
        group: 'bottom',
      },
    ],
  },
  {
    id: '4',
    shape: 'dag-node',
    x: 450,
    y: 350,
    data: {
      label: '读数据',
      status: 'success',
    },
    ports: [
      {
        id: '4-1',
        group: 'top',
      },
      {
        id: '4-2',
        group: 'bottom',
      },
    ],
  },
  {
    id: '5',
    shape: 'dag-edge',
    source: {
      cell: '1',
      port: '1-1',
    },
    target: {
      cell: '2',
      port: '2-1',
    },
    zIndex: 0,
  },
  {
    id: '6',
    shape: 'dag-edge',
    source: {
      cell: '2',
      port: '2-2',
    },
    target: {
      cell: '3',
      port: '3-1',
    },
    zIndex: 0,
  },
  {
    id: '7',
    shape: 'dag-edge',
    source: {
      cell: '2',
      port: '2-3',
    },
    target: {
      cell: '4',
      port: '4-1',
    },
    zIndex: 0,
  },
]

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',
    },
  ],
]

export const CaseDagExample: React.FC = () => {
  const containerRef = useRef<HTMLDivElement>(null)
  useEffect(() => {
    if (!containerRef.current) return

    const graph: Graph = new Graph({
      container: containerRef.current,
      width: 800,
      height: 600,
      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,
          })
        },
      },
    })

    const selection = new Selection({
      multiple: true,
      rubberEdge: true,
      rubberNode: true,
      modifiers: 'shift',
      rubberband: true,
      showNodeSelectionBox: true,
      resizable: true,
      rotatable: { grid: 15 },
    })
    graph.use(selection)
    graph.use(new Snapline())

    graph.on('edge:connected', ({ edge }) => {
      edge.attr({
        line: {
          strokeDasharray: '',
        },
      })
    })

    graph.on('node:change:data', ({ node }) => {
      const edges = graph.getIncomingEdges(node)
      const { status } = node.getData() as NodeStatus
      edges?.forEach((edge) => {
        if (status === 'running') {
          edge.attr('line/strokeDasharray', 5)
          edge.attr('line/style/animation', 'running-line 30s infinite linear')
        } else {
          edge.attr('line/strokeDasharray', '')
          edge.attr('line/style/animation', '')
        }
      })
    })

    // 初始化节点/边
    const init = (data: CellMetadata[]) => {
      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)
    }

    let timer: number
    // 显示节点状态
    const showNodeStatus = (statusList: NodeStatus[][]) => {
      const list = [...statusList]
      const status = list.shift()
      status?.forEach((item) => {
        const { id, status } = item
        const node = graph.getCellById(id)
        if (node) {
          const data = node.getData() as NodeStatus
          node.setData({
            ...data,
            status: status,
          })
        }
      })
      if (list.length > 0) {
        timer = window.setTimeout(() => {
          showNodeStatus(list)
        }, 3000)
      }
    }

    init(data)
    showNodeStatus(nodeStatusList)

    return () => {
      clearTimeout(timer)
      graph.dispose()
    }
  }, [])

  return (
    <div className="x6-graph-wrap">
      <div ref={containerRef} className="dag" />
    </div>
  )
}