From 2785cf9c19ced5ba19e85eaaa6f788ef627d7c08 Mon Sep 17 00:00:00 2001
From: nixzhu <zhuhongxu@gmail.com>
Date: Tue, 8 Dec 2015 11:05:02 +0800
Subject: [PATCH] pre cache AttachmentType

---
 .../MediaPreview/MediaPreviewViewController.swift    | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/Yep/ViewControllers/MediaPreview/MediaPreviewViewController.swift b/Yep/ViewControllers/MediaPreview/MediaPreviewViewController.swift
index d4ba277e..8926fa71 100644
--- a/Yep/ViewControllers/MediaPreview/MediaPreviewViewController.swift
+++ b/Yep/ViewControllers/MediaPreview/MediaPreviewViewController.swift
@@ -193,6 +193,18 @@ class MediaPreviewViewController: UIViewController {
         view.addGestureRecognizer(swipeDown)
 
         currentIndex = startIndex
+
+        for previewMedia in previewMedias {
+
+            if case let .AttachmentType(attachment) = previewMedia {
+
+                ImageCache.sharedInstance.imageOfAttachment(attachment, withSize: nil, completion: { [weak self] (url, image, _) in
+                    if let image = image {
+                        self?.attachmentImagePool.addImage(image, forKey: attachment.URLString)
+                    }
+                })
+            }
+        }
     }
 
     var isFirstLayoutSubviews = true