test loading

app/api/admin/charts/route.ts

@@ -0,0 +1,130 @@
+import { NextRequest, NextResponse } from "next/server";
+import prisma from "@/lib/prisma";
+import { addHours, subMinutes } from "date-fns";
+import { EChartsOption } from "echarts";
+
+interface StringKeyedObject {
+  [key: string]: { [key: string]: number };
+}
+type StringSet = Set<string>;
+type StringArray = string[];
+
+function HandleLogData(
+  todayLog: [{ userName: string; logToken: number; model: string }],
+) {
+  // 先遍历一遍，获取所有的模型和名字。
+  let all_models: StringSet = new Set();
+  let all_names: StringSet = new Set();
+  // 拼接数据结构
+  let data_by_name: StringKeyedObject = {};
+  todayLog.map((log) => {
+    all_models.add(log.model);
+    all_names.add(log.userName);
+    data_by_name[log.userName] = data_by_name[log.userName] ?? {};
+    data_by_name[log.userName][log.model] =
+      (data_by_name[log.userName][log.model] || 0) + log.logToken;
+    // 这么顺利，顺便加个总数吧。
+    data_by_name[log.userName]["all_token"] =
+      (data_by_name[log.userName]["all_token"] || 0) + log.logToken;
+  });
+  //
+  // 然后遍历并以all_token，排序。
+  const userNameList: StringArray = Array.from(all_names).sort((a, b) => {
+    return data_by_name[a]["all_token"] - data_by_name[b]["all_token"];
+  });
+  // 将值按模型分为两个序列
+  const modelNameList = Array.from(all_models).map((model) => {
+    return {
+      name: model,
+      data: userNameList.map((userName) => {
+        return data_by_name[userName][model] ?? null;
+      }),
+    };
+  });
+  return {
+    modelNameList,
+    userNameList,
+    data_by_name,
+  };
+}
+
+async function handle(req: NextRequest) {
+  const { searchParams } = new URL(req.url);
+  const date = searchParams.get("date") ?? "";
+  // 当天日期的开始和结束
+  let currentTime = new Date(date);
+  // today = subMinutes(today, today.getTimezoneOffset())
+  const startOfTheDayInTimeZone = new Date(
+    currentTime.getFullYear(),
+    currentTime.getMonth(),
+    currentTime.getDate(),
+    0,
+    0,
+    0,
+  );
+  const endOfTheDayInTimeZone = addHours(startOfTheDayInTimeZone, +24); // 当天的结束时间
+  const todayLog = await prisma.logEntry.findMany({
+    where: {
+      createdAt: {
+        gte: startOfTheDayInTimeZone.toISOString(), // gte 表示 '大于等于'
+        lte: endOfTheDayInTimeZone.toISOString(), // lte 表示 '小于等于'
+      },
+    },
+    include: {
+      user: true,
+    },
+  });
+  // @ts-ignore
+  const log_data = HandleLogData(todayLog);
+  // console.log('log_data', log_data)
+
+  const usageByModelOption: EChartsOption = {
+    tooltip: {
+      trigger: "axis",
+      axisPointer: {
+        type: "shadow",
+      },
+    },
+    title: {
+      show: true,
+      text: "token使用分析",
+    },
+    legend: {},
+    grid: {
+      left: "3%",
+      right: "4%",
+      bottom: "3%",
+      containLabel: true,
+    },
+    xAxis: [
+      {
+        type: "value",
+      },
+    ],
+    yAxis: [
+      {
+        type: "category",
+        data: log_data.userNameList,
+      },
+    ],
+    series: log_data.modelNameList.map((item) => {
+      return {
+        ...item,
+        type: "bar",
+        emphasis: {
+          focus: "series",
+        },
+        label: {
+          show: true,
+          position: "right",
+        },
+        colorBy: "series",
+        stack: "model",
+      };
+    }),
+  };
+  return NextResponse.json(usageByModelOption);
+}
+
+export const GET = handle;
+// export const POST = handle;