feat: add chart type source in container (#2108)

* feat: add chart type source in container * fix: cr
antvis · Dec 17, 2020 · c26eb37 · c26eb37
1 parent 48547f8
commit c26eb37
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/__tests__/unit/core/index-spec.ts b/__tests__/unit/core/index-spec.ts
@@ -203,4 +203,20 @@ describe('core', () => {
     expect(line.chart.limitInPlot).toBe(true);
     line.destroy();
   });
+
+  it('plot dataset sourceType', () => {
+    const container = createDiv('');
+    const line = new Line(container, {
+      data: partySupport.filter((o) => o.type === 'FF'),
+      xField: 'date',
+      yField: 'value',
+    });
+
+    line.render();
+
+    expect(container.dataset.chartSourceType).toBe('G2Plot');
+    line.destroy();
+
+    expect(container.dataset.chartSourceType).toBe(undefined);
+  });
 });
diff --git a/src/core/plot.ts b/src/core/plot.ts
@@ -22,6 +22,8 @@ export type PickOptions = Pick<
   | 'limitInPlot'
 >;
 
+const SOURCE_ATTRIBUTE_NAME = 'data-chart-source-type';
+
 /**
  * 所有 plot 的基类
  */
@@ -77,6 +79,9 @@ export abstract class Plot<O extends PickOptions> extends EE {
       supportCSSTransform,
       limitInPlot,
     });
+
+    // 给容器增加标识，知道图表的来源区别于 G2
+    this.container.setAttribute(SOURCE_ATTRIBUTE_NAME, 'G2Plot');
   }
 
   /**
@@ -229,6 +234,8 @@ export abstract class Plot<O extends PickOptions> extends EE {
     this.chart.destroy();
     // 清空已经绑定的事件
     this.off();
+
+    this.container.removeAttribute(SOURCE_ATTRIBUTE_NAME);
   }
 
   /**