From 2884bf102f2159f9b011f0eca9c1917c6d41200f Mon Sep 17 00:00:00 2001
From: zhaowei <cp3hnu@gmail.com>
Date: Tue, 13 May 2025 14:22:15 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E7=82=B9=E8=B5=9E=E5=88=9D=E7=A8=BF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../Dataset/components/ResourceInfo/index.tsx  | 18 ++++++++++++++++++
 .../Dataset/components/ResourceItem/index.tsx  |  1 +
 react-ui/src/pages/Dataset/config.tsx          |  2 ++
 react-ui/src/services/dataset/index.js         | 15 +++++++++++++++
 4 files changed, 36 insertions(+)

diff --git a/react-ui/src/pages/Dataset/components/ResourceInfo/index.tsx b/react-ui/src/pages/Dataset/components/ResourceInfo/index.tsx
index 2eeb09b8..8566bcf2 100644
--- a/react-ui/src/pages/Dataset/components/ResourceInfo/index.tsx
+++ b/react-ui/src/pages/Dataset/components/ResourceInfo/index.tsx
@@ -13,6 +13,7 @@ import {
 } from '@/pages/Dataset/config';
 import GraphLegend from '@/pages/Model/components/GraphLegend';
 import ModelEvolution from '@/pages/Model/components/ModelEvolution';
+import { praiseResourceReq, unpraiseResourceReq } from '@/services/dataset';
 import { VersionChangedMessage } from '@/utils/constant';
 import { openAntdModal } from '@/utils/modal';
 import { to } from '@/utils/promise';
@@ -189,6 +190,20 @@ const ResourceInfo = ({ resourceType }: ResourceInfoProps) => {
     });
   };
 
+  // 处理点赞
+  const handlePraise = async () => {
+    const request = info.praised === true ? unpraiseResourceReq : praiseResourceReq;
+    const [res] = await to(request(info.id));
+    if (res) {
+      message.success('操作成功');
+      setInfo({
+        ...info,
+        praised: !info.praised,
+        praises_count: info.praised ? info.praises_count - 1 : info.praises_count + 1,
+      });
+    }
+  };
+
   const items = [
     {
       key: ResourceInfoTabKeys.Introduction,
@@ -248,6 +263,9 @@ const ResourceInfo = ({ resourceType }: ResourceInfoProps) => {
               {(info[tagPropertyName] as string) || '--'}
             </div>
           )}
+          <Button type="primary" onClick={handlePraise}>
+            {info.praised ? '取消点赞' : '点赞'} {info.praises_count}
+          </Button>
         </Flex>
         <Flex align="center">
           <span style={{ marginRight: '10px' }}>版本号：</span>
diff --git a/react-ui/src/pages/Dataset/components/ResourceItem/index.tsx b/react-ui/src/pages/Dataset/components/ResourceItem/index.tsx
index 22078902..b23a9166 100644
--- a/react-ui/src/pages/Dataset/components/ResourceItem/index.tsx
+++ b/react-ui/src/pages/Dataset/components/ResourceItem/index.tsx
@@ -41,6 +41,7 @@ function ResourceItem({ item, isPublic, onClick, onRemove }: ResourceItemProps)
         )}
       </Flex>
       <div className={styles['resource-item__description']}>{item.description}</div>
+      <div style={{ marginBottom: 20 }}>点赞数量：{item.praises_count}</div>
       <Flex justify="space-between" gap={'0 8px'}>
         <div className={styles['resource-item__time']}>
           <img
diff --git a/react-ui/src/pages/Dataset/config.tsx b/react-ui/src/pages/Dataset/config.tsx
index 61c63531..af1993b9 100644
--- a/react-ui/src/pages/Dataset/config.tsx
+++ b/react-ui/src/pages/Dataset/config.tsx
@@ -162,6 +162,8 @@ export interface ResourceData {
   usage?: string;
   relative_paths?: string;
   train_task?: TrainTask; // 训练任务
+  praises_count: number; // 点赞数
+  praised: boolean; // 是否点赞
 }
 
 // 数据集数据
diff --git a/react-ui/src/services/dataset/index.js b/react-ui/src/services/dataset/index.js
index dcacf124..f1adfb17 100644
--- a/react-ui/src/services/dataset/index.js
+++ b/react-ui/src/services/dataset/index.js
@@ -187,3 +187,18 @@ export function deleteUploadFileReq(params) {
     params,
   });
 }
+
+
+// 点赞
+export function praiseResourceReq(id) {
+  return request(`/api/mmp/newmodel/praise/${id}`, {
+    method: 'POST',
+  });
+}
+
+// 取消点赞
+export function unpraiseResourceReq(id) {
+  return request(`/api/mmp/newmodel/unpraise/${id}`, {
+    method: 'DELETE',
+  });
+}
\ No newline at end of file