From 6413bc8e4c400e2d6a1b767b2d45dbccdfc8fa88 Mon Sep 17 00:00:00 2001
From: Matthew Chen <charlesmchen@gmail.com>
Date: Wed, 15 Nov 2017 11:12:30 -0500
Subject: [PATCH] Evacuate the view item cache.

---
 .../ConversationView/ConversationViewController.m        | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/Signal/src/ViewControllers/ConversationView/ConversationViewController.m b/Signal/src/ViewControllers/ConversationView/ConversationViewController.m
index d735a6a04..263d92ee1 100644
--- a/Signal/src/ViewControllers/ConversationView/ConversationViewController.m
+++ b/Signal/src/ViewControllers/ConversationView/ConversationViewController.m
@@ -2866,6 +2866,15 @@ typedef NS_ENUM(NSInteger, MessagesRangeSizeMode) {
                 }
                 break;
             }
+            case YapDatabaseViewChangeDelete: {
+                // Discard cached view items after deletes.
+                YapCollectionKey *collectionKey = rowChange.collectionKey;
+                OWSAssert(collectionKey.key.length > 0);
+                if (collectionKey.key) {
+                    [self.viewItemMap removeObjectForKey:collectionKey.key];
+                }
+                break;
+            }
             default:
                 break;
         }