perf: use count() for song.info

Signed-off-by: ZTL-UwU <zhangtianli2006@163.com>
SMS-COSMO · Oct 1, 2024 · 4ba5b9b · 4ba5b9b
1 parent 846a425
commit 4ba5b9b
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 6 deletions.
diff --git a/pages/index.vue b/pages/index.vue
@@ -18,7 +18,7 @@ const { data: arrangementList } = await $api.arrangement.listSafe.useQuery();
 const songListInfo = ref(0);
 const timeCanSubmit = ref(true);
 try {
-  songListInfo.value = await $api.song.info.query();
+  songListInfo.value = await $api.song.info.query({ getAll: false });
   timeCanSubmit.value = await $api.time.currently.query();
 } catch (err) {
   useErrorHandler(err);
@@ -132,7 +132,7 @@ async function refreshData() {
   if (timeCanSubmit.value) {
     try {
       const res = await Promise.all([
-        $api.song.info.query(),
+        $api.song.info.query({ getAll: false }),
         $api.time.currently.query(),
         $api.song.listSafe.query(),
       ]);

diff --git a/server/trpc/controllers/song.ts b/server/trpc/controllers/song.ts
@@ -1,5 +1,5 @@
 import { LibsqlError } from '@libsql/client';
-import { and, desc, eq, gt, inArray, or } from 'drizzle-orm';
+import { and, count, desc, eq, gt, inArray, or } from 'drizzle-orm';
 import { type TNewSong, db } from '../../db/db';
 import { songs } from '~/server/db/schema';
 import type { TStatus } from '~/types';
@@ -90,4 +90,17 @@ export class SongController {
       return { success: false, message: '服务器内部错误' };
     }
   }
+
+  async count(getAll: boolean) {
+    try {
+      let res = 0;
+      if (getAll)
+        res = (await db.select({ count: count() }).from(songs))[0]?.count;
+      else
+        res = (await db.select({ count: count() }).from(songs).where(gt(songs.createdAt, new Date(Date.now() - 4 * 24 * 60 * 60 * 1000))))[0]?.count;
+      return { success: true, res, message: '获取成功' };
+    } catch {
+      return { success: false, message: '服务器内部错误' };
+    }
+  }
 }
diff --git a/server/trpc/routers/song.ts b/server/trpc/routers/song.ts
@@ -101,10 +101,13 @@ export const songRouter = router({
     }),
 
   info: publicProcedure
-    .query(async ({ ctx }) => {
-      const res = await ctx.songController.getList();
+    .input(z.object({
+      getAll: z.boolean(),
+    }))
+    .query(async ({ ctx, input }) => {
+      const res = await ctx.songController.count(input.getAll);
       if (!res.success || !res.res)
         throw new TRPCError({ code: 'BAD_REQUEST', message: res.message });
-      else return res.res.length;
+      else return res.res;
     }),
 });