feat(analytics): sunburst for platform bots

koishijs · Dec 15, 2023 · 1fd4481 · 1fd4481
1 parent 1233cd7
commit 1fd4481
Show file tree

Hide file tree

Showing 6 changed files with 48 additions and 37 deletions.
diff --git a/plugins/analytics/client/charts/bot.ts b/plugins/analytics/client/charts/bot.ts
@@ -1,43 +1,49 @@
 import { Context } from '@koishijs/client'
-import { createChart } from './utils'
+import { createChart, Tooltip } from './utils'
 
 export default (ctx: Context) => {
   ctx.slot({
     type: 'chart',
     component: createChart({
-      title: '各平台消息数量',
+      title: '各平台占比',
       fields: ['analytics'],
       options({ analytics }) {
         if (!Object.keys(analytics.messageByBot).length) return
-        const platformData = Object
+        const data = Object
           .entries(analytics.messageByBot)
           .map(([key, value]) => ({
             name: key,
-            value: Object.values(value).reduce((a, b) => a + b.send, 0),
+            children: Object
+              .entries(value)
+              .map(([key, value]) => ({
+                name: value.name || key,
+                value: value.send,
+              })),
           }))
-          .sort((a, b) => b.value - a.value)
-        const botData = platformData.flatMap(({ name }) => {
-          return Object
-            .entries(analytics.messageByBot[name])
-            .map(([key, value]) => ({
-              name: key,
-              platform: name,
-              value: value.send,
-            }))
-            .sort((a, b) => b.value - a.value)
-        })
 
         return {
+          tooltip: Tooltip.item(({ data }) => {
+            return `${data.children ? '平台' : '昵称'}：${data.name}<br>消息数量：${data.value.toFixed(2)}`
+          }),
           series: [{
-            type: 'pie',
-            data: platformData,
-            radius: ['0%', '35%'],
-            minShowLabelAngle: 3,
-          }, {
-            type: 'pie',
-            data: botData,
-            radius: ['35%', '65%'],
-            minShowLabelAngle: 3,
+            type: 'sunburst',
+            data,
+            radius: ['0', '65%'],
+            nodeClick: false,
+            emphasis: {
+              focus: 'ancestor',
+            },
+            levels: [{}, {
+              label: {
+                rotate: 'tangential',
+              },
+            }, {
+              label: {
+                position: 'outside',
+                padding: 3,
+                silent: false,
+              },
+            }],
           }],
         }
       },

diff --git a/plugins/analytics/client/charts/command.ts b/plugins/analytics/client/charts/command.ts
@@ -16,7 +16,7 @@ export default (ctx: Context) => {
         return {
           tooltip: Tooltip.item(({ data }) => {
             const output = [data.name]
-            output.push(`日均调用：${data.value}`)
+            output.push(`日均调用：${data.value.toFixed(2)}`)
             return output.join('<br>')
           }),
           series: [{

diff --git a/plugins/analytics/client/charts/echarts.ts b/plugins/analytics/client/charts/echarts.ts
@@ -1,9 +1,9 @@
 import { use } from 'echarts/core'
 import { CanvasRenderer } from 'echarts/renderers'
 import { GridComponent, TooltipComponent } from 'echarts/components'
-import { BarChart, LineChart, PieChart } from 'echarts/charts'
+import { BarChart, LineChart, PieChart, SunburstChart } from 'echarts/charts'
 import VChart from 'vue-echarts'
 
-use([BarChart, CanvasRenderer, GridComponent, LineChart, TooltipComponent, PieChart])
+use([BarChart, CanvasRenderer, GridComponent, LineChart, TooltipComponent, PieChart, SunburstChart])
 
 export default VChart
diff --git a/plugins/analytics/client/charts/history.ts b/plugins/analytics/client/charts/history.ts
@@ -19,15 +19,15 @@ export default (ctx: Context) => {
           }),
           xAxis: {
             type: 'category',
-            data: Object.keys(analytics.messageByDate).reverse(),
+            data: Object.keys(analytics.messageByDate),
           },
           yAxis: {
             type: 'value',
           },
           series: {
             type: 'line',
             smooth: true,
-            data: Object.values(analytics.messageByDate).map(stats => stats.send).reverse(),
+            data: Object.values(analytics.messageByDate).map(stats => stats.send),
           },
         }
       },

diff --git a/plugins/analytics/client/charts/utils.ts b/plugins/analytics/client/charts/utils.ts
@@ -24,6 +24,7 @@ export function createChart({ title, fields, options }: ChartOptions) {
 interface CommonData {
   name: string
   value: number
+  children?: CommonData
 }
 
 export namespace Tooltip {

diff --git a/plugins/analytics/src/index.ts b/plugins/analytics/src/index.ts
@@ -1,4 +1,4 @@
-import { $, Context, deepEqual, Dict, Logger, Row, Schema, Session, Time } from 'koishi'
+import { $, Context, deepEqual, Dict, Logger, pick, Row, Schema, Session, Time, Universal } from 'koishi'
 import { DataService } from '@koishijs/console'
 import { resolve } from 'path'
 
@@ -54,7 +54,7 @@ class Analytics extends DataService<Analytics.Payload> {
       hour: 'integer',
       name: 'string',
       selfId: 'string',
-      userId: 'string',
+      userId: 'integer',
       channelId: 'string',
       platform: 'string',
       count: 'integer',
@@ -111,8 +111,8 @@ class Analytics extends DataService<Analytics.Payload> {
     }
   }
 
-  private addAudit<T extends Analytics.Message | Analytics.Command>(buffer: T[], index: Omit<T, 'count'>) {
-    const audit = buffer.find(data => deepEqual(data, index))
+  private addAudit<T extends Analytics.Audit>(buffer: T[], index: Omit<T, 'count'>) {
+    const audit = buffer.find(data => deepEqual(pick(data, Object.keys(index) as (keyof T)[]), index))
     if (audit) {
       audit.count += 1
     } else {
@@ -182,9 +182,13 @@ class Analytics extends DataService<Analytics.Payload> {
         count: row => $.sum(row.count),
       })
       .execute()
-    const result = {} as Dict<Dict<MessageStats>>
+    const result = {} as Dict<Dict<MessageStats & Universal.User>>
     data.forEach((stat) => {
-      const entry = (result[stat.platform] ||= {})[stat.selfId] ||= { send: 0, receive: 0 }
+      const entry = (result[stat.platform] ||= {})[stat.selfId] ||= {
+        ...this.ctx.bots[`${stat.platform}:${stat.selfId}`]?.user,
+        send: 0,
+        receive: 0,
+      }
       entry[stat.type] = stat.count / 7
     })
     return result
@@ -296,7 +300,7 @@ namespace Analytics {
 
   export interface Command extends Index {
     name: string
-    userId: string
+    userId: number
     channelId: string
     count: number
   }
@@ -308,7 +312,7 @@ namespace Analytics {
     guildIncrement: number
     dauHistory: Dict<number>
     commandRate: Dict<number>
-    messageByBot: Dict<Dict<MessageStats>>
+    messageByBot: Dict<Dict<MessageStats & Universal.User>>
     messageByDate: Dict<MessageStats>
     messageByHour: MessageStats[]
   }