Merge pull request #137 from zhujunsan/kerwin_main

feat: 增加用户使用量统计页面

Author: Kerwin1202

package.json

@@ -25,6 +25,8 @@
   "dependencies": {
     "@traptitech/markdown-it-katex": "^3.6.0",
     "@vueuse/core": "^9.13.0",
+    "chart.js": "^4.3.0",
+    "dayjs": "^1.11.7",
     "highlight.js": "^11.7.0",
     "html2canvas": "^1.4.1",
     "jwt-decode": "^3.1.2",
@@ -33,6 +35,7 @@
     "naive-ui": "^2.34.3",
     "pinia": "^2.0.33",
     "vue": "^3.2.47",
+    "vue-chartjs": "^5.2.0",
     "vue-i18n": "^9.2.2",
     "vue-router": "^4.1.6"
   },
@@ -41,6 +44,7 @@
     "@commitlint/cli": "^17.4.4",
     "@commitlint/config-conventional": "^17.4.4",
     "@iconify/vue": "^4.1.0",
+    "@types/chart.js": "^2.9.37",
     "@types/crypto-js": "^4.1.1",
     "@types/katex": "^0.16.0",
     "@types/markdown-it": "^12.2.3",

pnpm-lock.yaml

@@ -26,6 +26,7 @@
   "dependencies": {
     "axios": "^1.3.4",
     "chatgpt": "^5.2.4",
+    "dayjs": "^1.11.7",
     "dotenv": "^16.0.3",
     "esno": "^0.16.3",
     "express": "^4.18.2",

service/package.json

@@ -23,6 +23,7 @@ import {
   getChats,
   getUser,
   getUserById,
+  getUserStatisticsByDay,
   insertChat,
   insertChatUsage,
   renameChatRoom,
@@ -716,6 +717,19 @@ router.post('/audit-test', rootAuth, async (req, res) => {
   }
 })
 
+router.post('/statistics/by-day', auth, async (req, res) => {
+  try {
+    const userId = req.headers.userId
+    const { start, end } = req.body as { start: number; end: number }
+
+    const data = await getUserStatisticsByDay(new ObjectId(userId as string), start, end)
+    res.send({ status: 'Success', message: '', data })
+  }
+  catch (error) {
+    res.send(error)
+  }
+})
+
 app.use('', router)
 app.use('/api', router)
 app.set('trust proxy', 1)

service/pnpm-lock.yaml

@@ -1,5 +1,6 @@
 import { MongoClient, ObjectId } from 'mongodb'
 import * as dotenv from 'dotenv'
+import dayjs from 'dayjs'
 import { ChatInfo, ChatRoom, ChatUsage, Status, UserInfo } from './model'
 import type { ChatOptions, Config, UsageResponse } from './model'
 
@@ -211,3 +212,72 @@ export async function updateConfig(config: Config): Promise<Config> {
     return config
   return null
 }
+
+export async function getUserStatisticsByDay(userId: ObjectId, start: number, end: number): Promise<any> {
+  const pipeline = [
+    { // filter by dateTime
+      $match: {
+        dateTime: {
+          $gte: start,
+          $lte: end,
+        },
+        userId,
+      },
+    },
+    { // convert dateTime to date
+      $addFields: {
+        date: {
+          $dateToString: {
+            format: '%Y-%m-%d',
+            date: {
+              $toDate: '$dateTime',
+            },
+          },
+        },
+      },
+    },
+    { // group by date
+      $group: {
+        _id: '$date',
+        promptTokens: {
+          $sum: '$promptTokens',
+        },
+        completionTokens: {
+          $sum: '$completionTokens',
+        },
+        totalTokens: {
+          $sum: '$totalTokens',
+        },
+      },
+    },
+    { // sort by date
+      $sort: {
+        _id: 1,
+      },
+    },
+  ]
+
+  const aggStatics = await usageCol.aggregate(pipeline).toArray()
+
+  const step = 86400000 // 1 day in milliseconds
+  const result = {
+    promptTokens: null,
+    completionTokens: null,
+    totalTokens: null,
+    chartData: [],
+  }
+  for (let i = start; i <= end; i += step) {
+    // Convert the timestamp to a Date object
+    const date = dayjs(i, 'x').format('YYYY-MM-DD')
+
+    const dateData = aggStatics.find(x => x._id === date)
+      || { _id: date, promptTokens: 0, completionTokens: 0, totalTokens: 0 }
+
+    result.promptTokens += dateData.promptTokens
+    result.completionTokens += dateData.completionTokens
+    result.totalTokens += dateData.totalTokens
+    result.chartData.push(dateData)
+  }
+
+  return result
+}

service/src/index.ts

@@ -216,3 +216,10 @@ export function fetchUpdateBaseSetting<T = any>(config: ConfigState) {
     data: config,
   })
 }
+
+export function fetchUserStatistics<T = any>(start: number, end: number) {
+  return post<T>({
+    url: '/statistics/by-day',
+    data: { start, end },
+  })
+}