'#1889: Minor code enhancements.

sepinf-inc · Sep 22, 2023 · 5d954a3 · 5d954a3
1 parent 0d3b838
commit 5d954a3
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/iped-parsers/iped-parsers-impl/src/main/java/iped/parsers/whatsapp/ExtractorIOS.java b/iped-parsers/iped-parsers-impl/src/main/java/iped/parsers/whatsapp/ExtractorIOS.java
@@ -157,7 +157,6 @@ protected List<Chat> extractChatList() throws WAExtractorException {
                     }
                 }
 
-                Map<Long, Chat> idToChat = new HashMap<Long, Chat>();                
                 String chatListQuery = hasProfilePictureItemTable ? SELECT_CHAT_LIST : SELECT_CHAT_LIST_NO_PPIC;
 
                 try (ResultSet rs = stmt.executeQuery(chatListQuery)) {
@@ -175,7 +174,6 @@ protected List<Chat> extractChatList() throws WAExtractorException {
                                 activeChats.add(c.getId());
                             }
                             list.add(c);
-                            idToChat.put(c.getId(), c);
                         }
                     }
                 } catch (SQLException ex) {
@@ -198,6 +196,10 @@ protected List<Chat> extractChatList() throws WAExtractorException {
                 }
 
                 // Extract messages of all non-group and group chats at once, not per chat
+                Map<Long, Chat> idToChat = new HashMap<Long, Chat>();
+                for (Chat c : list) {
+                    idToChat.put(c.getId(), c);
+                }
                 extractMessages(conn, idToChat, firstTry, false);
                 extractMessages(conn, idToChat, firstTry, true);