From b562b457a4c6ca7b0bc26d69013df21fba40256e Mon Sep 17 00:00:00 2001
From: jiang <jiang@bonus>
Date: Tue, 26 Nov 2024 10:20:05 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/com/bonus/ai/utils/MinioUtil.java    | 29 +++++++++++++++++++
 .../resources/mapper/DatasetFileMapper.xml    |  7 +++--
 2 files changed, 33 insertions(+), 3 deletions(-)

diff --git a/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/utils/MinioUtil.java b/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/utils/MinioUtil.java
index fcff693..3e429da 100644
--- a/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/utils/MinioUtil.java
+++ b/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/utils/MinioUtil.java
@@ -209,4 +209,33 @@ public class MinioUtil {
     public String joinPath(String folderPath, String filename) {
         return folderPath.replaceAll("/+$", "") + "/" + filename;
     }
+
+    /**
+     * 获取文件的临时访问 URL，指定过期时间
+     * @param bucketName 存储桶名称
+     * @param objectName 存储对象名称（文件名）
+     * @param expiryTimeInSeconds URL 的有效时长（秒）
+     * @return 文件的临时访问 URL
+     * @throws Exception 若生成 URL 过程中发生异常
+     */
+    public String getFileUrl(String bucketName, String objectName, int expiryTimeInSeconds) throws Exception {
+        return minioClient.getPresignedObjectUrl(GetPresignedObjectUrlArgs.builder()
+                .bucket(bucketName)
+                .object(objectName)
+                .expiry(expiryTimeInSeconds)
+                .method(Method.GET)
+                .build());
+    }
+
+    /**
+     * 获取文件的临时访问 URL，默认过期时间为 7 天
+     * @param bucketName 存储桶名称
+     * @param objectName 存储对象名称（文件名）
+     * @return 文件的临时访问 URL
+     * @throws Exception 若生成 URL 过程中发生异常
+     */
+    public String getFileUrl(String bucketName, String objectName) throws Exception {
+        // 604800 秒 = 7 天
+        return getFileUrl(bucketName, objectName, 604800);
+    }
 }
diff --git a/bonus-modules/bonus-ai/src/main/resources/mapper/DatasetFileMapper.xml b/bonus-modules/bonus-ai/src/main/resources/mapper/DatasetFileMapper.xml
index d46bffb..991b99d 100644
--- a/bonus-modules/bonus-ai/src/main/resources/mapper/DatasetFileMapper.xml
+++ b/bonus-modules/bonus-ai/src/main/resources/mapper/DatasetFileMapper.xml
@@ -27,12 +27,13 @@
     </delete>
 
     <!-- 删除数据集文件映射关系 -->
-    <update id="deleteDatasetFiles">
-        update FROM ai_dataset_file_map
+    <delete id="deleteDatasetFiles" parameterType="String">
+        delete from ai_dataset_file_map
+            where  dataset_id in
         <foreach collection="array" item="datasetId" open="(" separator="," close=")">
             #{datasetId}
         </foreach>
-    </update>
+    </delete>
 
 <!--    <delete id="deleteDatasetFiles">-->
 <!--        DELETE FROM ai_dataset_file_map-->