improve(desktop): improve clearing history messages

emqx · Dec 21, 2022 · b6fa0f8 · b6fa0f8
1 parent f458453
commit b6fa0f8
Showing 1 changed file with 5 additions and 15 deletions.
diff --git a/src/database/services/MessageService.ts b/src/database/services/MessageService.ts
@@ -99,20 +99,10 @@ export default class MessageService {
   }
 
   public async cleanInConnection(connectionId: string) {
-    const query: MessageEntity[] | undefined = await this.messageRepository
-      .createQueryBuilder('ms')
-      .where('ms.connectionId = :connectionId', { connectionId })
-      .getMany()
-    if (!query || !query.length) {
-      return
-    }
-    const deleteID: string[] = []
-    query.forEach((entity) => {
-      entity.id && deleteID.push(entity.id)
-    })
-    if (!deleteID.length) {
-      return
-    }
-    await this.messageRepository.delete(deleteID)
+    await this.messageRepository
+      .createQueryBuilder()
+      .delete()
+      .where('connectionId = :connectionId', { connectionId })
+      .execute()
   }
 }