Add File

2025-09-08 16:37:45 +08:00
parent a5a3a2e142
commit 3c0799ef51
1 changed files with 108 additions and 0 deletions
--- a/frontend/src/views/chat/component/charts/Line.ts
+++ b/frontend/src/views/chat/component/charts/Line.ts
@@ -0,0 +1,108 @@
+import { BaseG2Chart } from '@/views/chat/component/BaseG2Chart.ts'
+import type { ChartAxis, ChartData } from '@/views/chat/component/BaseChart.ts'
+import type { G2Spec } from '@antv/g2'
+import { checkIsPercent } from '@/views/chat/component/charts/utils.ts'
+
+export class Line extends BaseG2Chart {
+  constructor(id: string) {
+    super(id, 'line')
+  }
+
+  init(axis: Array<ChartAxis>, data: Array<ChartData>) {
+    super.init(axis, data)
+
+    const x = this.axis.filter((item) => item.type === 'x')
+    const y = this.axis.filter((item) => item.type === 'y')
+    const series = this.axis.filter((item) => item.type === 'series')
+
+    if (x.length == 0 || y.length == 0) {
+      return
+    }
+
+    const _data = checkIsPercent(y[0], data)
+
+    const options: G2Spec = {
+      ...this.chart.options(),
+      type: 'view',
+      data: _data.data,
+      encode: {
+        x: x[0].value,
+        y: y[0].value,
+        color: series.length > 0 ? series[0].value : undefined,
+      },
+      axis: {
+        x: {
+          title: x[0].name,
+          labelFontSize: 12,
+          labelAutoHide: {
+            type: 'hide',
+            keepHeader: true,
+            keepTail: true,
+          },
+          labelAutoRotate: false,
+          labelAutoWrap: true,
+          labelAutoEllipsis: true,
+        },
+        y: { title: y[0].name },
+      },
+      scale: {
+        x: {
+          nice: true,
+        },
+        y: {
+          nice: true,
+        },
+      },
+      children: [
+        {
+          type: 'line',
+          encode: {
+            shape: 'smooth',
+          },
+          // labels: [
+          //   {
+          //     text: (data: any) => {
+          //       const value = data[y[0].value]
+          //       if (value === undefined || value === null) {
+          //         return ''
+          //       }
+          //       return `${value}${_data.isPercent ? '%' : ''}`
+          //     },
+          //     style: {
+          //       dx: -10,
+          //       dy: -12,
+          //     },
+          //     transform: [
+          //       { type: 'contrastReverse' },
+          //       { type: 'exceedAdjust' },
+          //       { type: 'overlapHide' },
+          //     ],
+          //   },
+          // ],
+          tooltip: (data) => {
+            if (series.length > 0) {
+              return {
+                name: data[series[0].value],
+                value: `${data[y[0].value]}${_data.isPercent ? '%' : ''}`,
+              }
+            } else {
+              return { name: y[0].name, value: `${data[y[0].value]}${_data.isPercent ? '%' : ''}` }
+            }
+          },
+        },
+        {
+          type: 'point',
+          style: {
+            fill: 'white',
+          },
+          encode: {
+            size: 1.5,
+          },
+          tooltip: false,
+        },
+      ],
+    } as G2Spec
+
+    this.chart.options(options)
+  }
+}