feat: limit export kb

src/pages/api/plugins/kb/data/exportModelData.ts

@@ -1,6 +1,6 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
-import { connectToDatabase } from '@/service/mongo';
+import { connectToDatabase, User } from '@/service/mongo';
 import { authUser } from '@/service/utils/auth';
 import { PgClient } from '@/service/pg';
 
@@ -14,10 +14,28 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
       throw new Error('缺少参数');
     }
 
+    await connectToDatabase();
+
     // 凭证校验
     const { userId } = await authUser({ req, authToken: true });
 
-    await connectToDatabase();
+    const thirtyMinutesAgo = new Date(Date.now() - 30 * 60 * 1000);
+
+    // auth export times
+    const authTimes = await User.findOne(
+      {
+        _id: userId,
+        $or: [
+          { 'limit.exportKbTime': { $exists: false } },
+          { 'limit.exportKbTime': { $lte: thirtyMinutesAgo } }
+        ]
+      },
+      '_id limit'
+    );
+
+    if (!authTimes) {
+      throw new Error('上次导出未到半小时，每半小时仅可导出一次。');
+    }
 
     // 统计数据
     const count = await PgClient.count('modelData', {
@@ -36,6 +54,11 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
       item.a.replace(/\n/g, '\\n')
     ]);
 
+    // update export time
+    await User.findByIdAndUpdate(userId, {
+      'limit.exportKbTime': new Date()
+    });
+
     jsonRes(res, {
       data
     });

src/pages/kb/components/DataCard.tsx

@@ -131,6 +131,10 @@ const DataCard = ({ kbId }: { kbId: string }) => {
           type: 'text/csv',
           filename: 'data.csv'
         });
+        toast({
+          title: '导出成功，下次导出需要半小时后',
+          status: 'success'
+        });
       } catch (error) {
         error;
       }
@@ -164,10 +168,10 @@ const DataCard = ({ kbId }: { kbId: string }) => {
           mr={2}
           size={'sm'}
           isLoading={isLoadingExport}
-          title={'换行数据导出时，会进行格式转换'}
+          title={'半小时仅能导出1次'}
           onClick={() => onclickExport()}
         >
-          导出
+          导出csv
         </Button>
         <Menu autoSelect={false}>
           <MenuButton as={Button} size={'sm'}>

src/service/models/user.ts

@@ -16,6 +16,10 @@ const UserSchema = new Schema({
     get: (val: string) => hashPassword(val),
     select: false
   },
+  createTime: {
+    type: Date,
+    default: () => new Date()
+  },
   avatar: {
     type: String,
     default: '/icon/human.png'
@@ -41,9 +45,11 @@ const UserSchema = new Schema({
     type: String,
     default: ''
   },
-  createTime: {
-    type: Date,
-    default: () => new Date()
+  limit: {
+    exportKbTime: {
+      // Every half hour
+      type: Date
+    }
   }
 });
 

src/types/mongoSchema.d.ts

@@ -22,6 +22,9 @@ export interface UserModelSchema {
   promotion: {
     rate: number;
   };
+  limit: {
+    exportKbTime?: Date;
+  };
 }
 
 export interface AuthCodeSchema {