对话流实现 文件上传

schoolNewsServ/ai/src/main/resources/mapper/AiUploadFileMapper.xml

@@ -8,6 +8,8 @@
         <result column="user_id" property="userID" jdbcType="VARCHAR"/>
         <result column="knowledge_id" property="knowledgeId" jdbcType="VARCHAR"/>
         <result column="conversation_id" property="conversationID" jdbcType="VARCHAR"/>
+        <result column="message_id" property="messageID" jdbcType="VARCHAR"/>
+        <result column="sys_file_id" property="sysFileId" jdbcType="VARCHAR"/>
         <result column="file_name" property="fileName" jdbcType="VARCHAR"/>
         <result column="file_path" property="filePath" jdbcType="VARCHAR"/>
         <result column="file_size" property="fileSize" jdbcType="BIGINT"/>
@@ -16,12 +18,10 @@
         <result column="extracted_text" property="extractedText" jdbcType="LONGVARCHAR"/>
         <result column="dify_document_id" property="difyDocumentId" jdbcType="VARCHAR"/>
         <result column="dify_batch_id" property="difyBatchId" jdbcType="VARCHAR"/>
-        <result column="vector_status" property="vectorStatus" jdbcType="INTEGER"/>
+        <result column="dify_upload_file_id" property="difyUploadFileId" jdbcType="VARCHAR"/>
         <result column="chunk_count" property="chunkCount" jdbcType="INTEGER"/>
         <result column="status" property="status" jdbcType="INTEGER"/>
         <result column="error_message" property="errorMessage" jdbcType="VARCHAR"/>
-        <result column="creator" property="creator" jdbcType="VARCHAR"/>
-        <result column="updater" property="updater" jdbcType="VARCHAR"/>
         <result column="create_time" property="createTime" jdbcType="TIMESTAMP"/>
         <result column="update_time" property="updateTime" jdbcType="TIMESTAMP"/>
         <result column="delete_time" property="deleteTime" jdbcType="TIMESTAMP"/>
@@ -30,9 +30,9 @@
 
     <!-- 基础字段 -->
     <sql id="Base_Column_List">
-        id, user_id, knowledge_id, conversation_id, file_name, file_path, file_size, 
-        file_type, mime_type, extracted_text, dify_document_id, dify_batch_id, 
-        vector_status, chunk_count, status, error_message, creator, updater, 
+        id, user_id, knowledge_id, conversation_id, message_id, sys_file_id, file_name, file_path, file_size, 
+        file_type, mime_type, extracted_text, dify_document_id, dify_batch_id, dify_upload_file_id,
+        chunk_count, status, error_message, 
         create_time, update_time, delete_time, deleted
     </sql>
 
@@ -56,9 +56,6 @@
                 <if test="filter.fileType != null and filter.fileType != ''">
                     AND file_type = #{filter.fileType}
                 </if>
-                <if test="filter.vectorStatus != null">
-                    AND vector_status = #{filter.vectorStatus}
-                </if>
                 <if test="filter.status != null">
                     AND status = #{filter.status}
                 </if>
@@ -69,14 +66,14 @@
     <!-- insertUploadFile（插入文件记录） -->
     <insert id="insertUploadFile" parameterType="org.xyzh.common.dto.ai.TbAiUploadFile">
         INSERT INTO tb_ai_upload_file (
-            id, user_id, knowledge_id, conversation_id, file_name, file_path, file_size,
-            file_type, mime_type, extracted_text, dify_document_id, dify_batch_id,
-            vector_status, chunk_count, status, error_message, creator, updater,
+            id, user_id, knowledge_id, conversation_id, message_id, sys_file_id, file_name, file_path, file_size,
+            file_type, mime_type, extracted_text, dify_document_id, dify_batch_id, dify_upload_file_id,
+            chunk_count, status, error_message,
             create_time, update_time, deleted
         ) VALUES (
-            #{ID}, #{userID}, #{knowledgeId}, #{conversationID}, #{fileName}, #{filePath}, #{fileSize},
-            #{fileType}, #{mimeType}, #{extractedText}, #{difyDocumentId}, #{difyBatchId},
-            #{vectorStatus}, #{chunkCount}, #{status}, #{errorMessage}, #{creator}, #{updater},
+            #{ID}, #{userID}, #{knowledgeId}, #{conversationID}, #{messageID}, #{sysFileId}, #{fileName}, #{filePath}, #{fileSize},
+            #{fileType}, #{mimeType}, #{extractedText}, #{difyDocumentId}, #{difyBatchId}, #{difyUploadFileId},
+            #{chunkCount}, #{status}, #{errorMessage},
             #{createTime}, #{updateTime}, #{deleted}
         )
     </insert>
@@ -88,6 +85,8 @@
             <if test="userID != null">user_id = #{userID},</if>
             <if test="knowledgeId != null">knowledge_id = #{knowledgeId},</if>
             <if test="conversationID != null">conversation_id = #{conversationID},</if>
+            <if test="messageID != null">message_id = #{messageID},</if>
+            <if test="sysFileId != null">sys_file_id = #{sysFileId},</if>
             <if test="fileName != null">file_name = #{fileName},</if>
             <if test="filePath != null">file_path = #{filePath},</if>
             <if test="fileSize != null">file_size = #{fileSize},</if>
@@ -96,11 +95,10 @@
             <if test="extractedText != null">extracted_text = #{extractedText},</if>
             <if test="difyDocumentId != null">dify_document_id = #{difyDocumentId},</if>
             <if test="difyBatchId != null">dify_batch_id = #{difyBatchId},</if>
-            <if test="vectorStatus != null">vector_status = #{vectorStatus},</if>
+            <if test="difyUploadFileId != null">dify_upload_file_id = #{difyUploadFileId},</if>
             <if test="chunkCount != null">chunk_count = #{chunkCount},</if>
             <if test="status != null">status = #{status},</if>
             <if test="errorMessage != null">error_message = #{errorMessage},</if>
-            <if test="updater != null">updater = #{updater},</if>
             <if test="updateTime != null">update_time = #{updateTime},</if>
         </set>
         WHERE id = #{ID} AND deleted = 0
@@ -110,8 +108,7 @@
     <update id="deleteUploadFile" parameterType="org.xyzh.common.dto.ai.TbAiUploadFile">
         UPDATE tb_ai_upload_file
         SET deleted = 1,
-            delete_time = NOW(),
-            updater = #{updater}
+            delete_time = NOW()
         WHERE id = #{ID} AND deleted = 0
     </update>
 
@@ -186,4 +183,31 @@
         ORDER BY create_time DESC
     </select>
 
+    <!-- selectFilesByMessageId（根据消息ID查询关联的文件列表） -->
+    <select id="selectFilesByMessageId" resultMap="BaseResultMap">
+        SELECT
+        <include refid="Base_Column_List"/>
+        FROM tb_ai_upload_file
+        WHERE message_id = #{messageId}
+          AND deleted = 0
+        ORDER BY create_time ASC
+    </select>
+
+    <!-- batchInsertUploadFiles（批量插入文件记录） -->
+    <insert id="batchInsertUploadFiles" parameterType="java.util.List">
+        INSERT INTO tb_ai_upload_file (
+            id, user_id, knowledge_id, conversation_id, message_id, sys_file_id, file_name, file_path, file_size,
+            file_type, mime_type, dify_document_id, dify_batch_id, dify_upload_file_id,
+            chunk_count, status, create_time, update_time, deleted
+        ) VALUES
+        <foreach collection="files" item="file" separator=",">
+            (
+                #{file.ID}, #{file.userID}, #{file.knowledgeId}, #{file.conversationID}, #{file.messageID},
+                #{file.sysFileId}, #{file.fileName}, #{file.filePath}, #{file.fileSize},
+                #{file.fileType}, #{file.mimeType}, #{file.difyDocumentId}, #{file.difyBatchId}, #{file.difyUploadFileId},
+                #{file.chunkCount}, #{file.status}, #{file.createTime}, #{file.updateTime}, #{file.deleted}
+            )
+        </foreach>
+    </insert>
+
 </mapper>