如何在 React 中使用 G6 | 🏆 技术专题第三期征文

3,296 阅读2分钟

概述

AntV G6 是一个简单、易用、完备的图可视化引擎,它在高定制能力的基础上,提供了一系列设计优雅、便于使用的图可视化解决方案。能帮助开发者搭建属于自己的图可视化、图分析、或图编辑器应用。

G6 是一个纯 JS 库,不与任何框架耦合,也就是说可以在任何前端框架中使用,如 React、Vue、Angular 等。由于我们目前绝大多数都是基于 React 技术栈的,所以本篇文章也仅介绍如何在 React 中使用 G6。

在 React 中使用 G6,和在 HTML 中使用基本相同,唯一比较关键的区分就是在实例化 Graph 时,要保证 DOM 容器渲染完成,并能获取到 DOM 元素

在 Demo 中,我们以一个简单的流程图为例,实现如下的效果。

img

功能及实现

Demo 包括以下功能点:

  • 自定义节点;
  • 自定义边;
  • 节点的 tooltip;
  • 边的 tooltip;
  • 节点上面弹出右键菜单;
  • tooltip 及 ContextMenu 如何渲染自定义的 React 组件。

在 React 中,通过  ref.current获取到真实的 DOM 元素,关于 React 的相关知识点,可以参考 React 文档。

import React, { useEffect, useState } from 'react';
import ReactDOM from 'react-dom';
import { data } from './data';
import G6 from '@antv/g6';

export default function () {
  const ref = React.useRef(null);
  let graph = null;

  useEffect(() => {
    if (!graph) {
      graph = new G6.Graph({
        container: ref.current,
        width: 1200,
        height: 800,
        layout: {
          type: 'dagre',
          direction: 'LR',
        },
      });
    }
    graph.data(data);
    graph.render();
  }, []);

  return <div ref={ref}></div>;
}

G6 中渲染 React 组件

节点和边的 tooltip、节点上的右键菜单,G6 中内置的很难满足样式上的需求,这个时候我们就可以通过渲染自定义的 React 组件来实现。Tooltip 和 ContextMenu 都是普通的 React 组件,样式完全由用户控制。交互过程中,在 G6 中需要做的事情就是确定何时渲染组件,以及渲染到何处。在 G6 中获取到是否渲染组件的标识值和渲染位置后,这些值就可以使用 React state 进行管理,后续的所有工作就全部由 React 负责了。

// 边 tooltip 坐标
const [showNodeTooltip, setShowNodeTooltip] = useState(false);
const [nodeTooltipX, setNodeToolTipX] = useState(0);
const [nodeTooltipY, setNodeToolTipY] = useState(0);

// 监听 node 上面 mouse 事件
graph.on('node:mouseenter', (evt) => {
  const { item } = evt;
  const model = item.getModel();
  const { x, y } = model;
  const point = graph.getCanvasByPoint(x, y);

  setNodeToolTipX(point.x - 75);
  setNodeToolTipY(point.y + 15);
  setShowNodeTooltip(true);
});

// 节点上面触发 mouseleave 事件后隐藏 tooltip 和 ContextMenu
graph.on('node:mouseleave', () => {
  setShowNodeTooltip(false);
});

return <div ref={ref}>{showNodeTooltip && <NodeTooltips x={nodeTooltipX} y={nodeTooltipY} />}</div>;

完整的 Demo 源码请戳 「这里」。

AntV G6 是一款开源的图可视化引擎,专注于图可视化及图分析领域。
欢迎关注和 star G6 GitHubgithub.com/antvis/G6
官网:g6.antv.vision/zh/

🏆 技术专题第三期 | 数据可视化的那些事......