add pg

lhpqaq · Sep 15, 2024 · f014c56 · f014c56
1 parent e581a3c
commit f014c56
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 10 deletions.
diff --git a/bigtop-manager-dao/src/main/resources/mapper/postgresql/ChatMessageMapper.xml b/bigtop-manager-dao/src/main/resources/mapper/postgresql/ChatMessageMapper.xml
@@ -26,11 +26,13 @@
         SELECT *
         FROM llm_chat_message
         WHERE thread_id = #{threadId}
+          AND is_deleted = false
     </select>
 
-    <delete id="deleteByThreadId">
-        DELETE FROM llm_chat_message
+    <update id="deleteByThreadId" parameterType="java.lang.Long">
+        UPDATE llm_chat_message
+        SET is_deleted = true
         WHERE thread_id = #{threadId}
-    </delete>
+    </update>
 
 </mapper>
diff --git a/bigtop-manager-dao/src/main/resources/mapper/postgresql/ChatThreadMapper.xml b/bigtop-manager-dao/src/main/resources/mapper/postgresql/ChatThreadMapper.xml
@@ -33,26 +33,39 @@
         SELECT *
         FROM llm_chat_thread
         WHERE user_id = #{userId}
+          AND is_deleted = false
     </select>
 
     <select id="findAllByPlatformAuthorizedIdAndUserId" resultType="org.apache.bigtop.manager.dao.po.ChatThreadPO">
         SELECT *
         FROM llm_chat_thread
-        WHERE platform_id = #{platformId} AND user_id = #{userId}
+        WHERE platform_id = #{platformId} AND user_id = #{userId} AND is_deleted = false
     </select>
 
     <select id="findByThreadId" resultMap="ChatThreadResultMap">
-        SELECT * FROM llm_chat_thread WHERE id = #{id}
+        SELECT * FROM llm_chat_thread WHERE id = #{id} AND is_deleted = false
     </select>
 
     <insert id="saveWithThreadInfo" parameterType="org.apache.bigtop.manager.dao.po.ChatThreadPO" useGeneratedKeys="true" keyProperty="id">
         INSERT INTO llm_chat_thread (platform_id, user_id, model, thread_info)
         VALUES (#{platformId}, #{userId}, #{model}, #{threadInfo, typeHandler=org.apache.bigtop.manager.dao.handler.JsonTypeHandler})
             ON DUPLICATE KEY UPDATE
-            platform_id = VALUES(platform_id),
-            user_id = VALUES(user_id),
-            model = VALUES(model),
-            thread_info = VALUES(thread_info)
+                                 platform_id = VALUES(platform_id),
+                                 user_id = VALUES(user_id),
+                                 model = VALUES(model),
+                                 thread_info = VALUES(thread_info)
     </insert>
 
+    <update id="deleteByPlatformId" parameterType="java.lang.Long">
+        UPDATE llm_chat_thread
+        SET is_deleted = true
+        WHERE platform_id = #{platformId}
+    </update>
+
+    <update id="deleteByThreadId" parameterType="java.lang.Long">
+        UPDATE llm_chat_thread
+        SET is_deleted = true
+        WHERE id = #{id}
+    </update>
+
 </mapper>
diff --git a/bigtop-manager-dao/src/main/resources/mapper/postgresql/PlatformAuthorizedMapper.xml b/bigtop-manager-dao/src/main/resources/mapper/postgresql/PlatformAuthorizedMapper.xml
@@ -31,7 +31,11 @@
     </resultMap>
 
     <select id="findByPlatformId" resultMap="PlatformAuthorizedResultMap">
-        SELECT * FROM llm_platform_authorized WHERE id = #{id}
+        SELECT * FROM llm_platform_authorized WHERE id = #{id} AND is_deleted = false
+    </select>
+
+    <select id="findAllPlatform" resultMap="PlatformAuthorizedResultMap">
+        SELECT * FROM llm_platform_authorized WHERE is_deleted = false
     </select>
 
     <insert id="saveWithCredentials" parameterType="org.apache.bigtop.manager.dao.po.PlatformAuthorizedPO" useGeneratedKeys="true" keyProperty="id">
@@ -42,4 +46,10 @@
                                  credentials = VALUES(credentials)
     </insert>
 
+    <update id="deleteByPlatformId" parameterType="java.lang.Long">
+        UPDATE llm_platform_authorized
+        SET is_deleted =true
+        WHERE id = #{id}
+    </update>
+
 </mapper>