Sora/Data/ImageCacheManager.swift


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

import Combine
import SwiftUI

final class ImageCacheManager {
  // MARK: - Singleton
  static let shared = ImageCacheManager()

  // MARK: - Private Properties
  private let cache = URLCache(
    memoryCapacity: 100 * 1_024 * 1_024,  // 100 MB
    diskCapacity: 500 * 1_024 * 1_024,  // 500 MB
    directory: FileManager.default.urls(for: .cachesDirectory, in: .userDomainMask).first?
      .appendingPathComponent("SoraImageCache", isDirectory: true)
  )
  private var cancellables = Set<AnyCancellable>()
  private let cancellablesQueue = DispatchQueue(
    label: "\(Bundle.main.bundleIdentifier!).ImageCacheManager.cancellablesQueue"
  )
  private let downloadQueue = OperationQueue()

  // MARK: - Initialisation
  private init() {
    downloadQueue.maxConcurrentOperationCount = 5
  }

  // MARK: - Public Methods
  func preloadImages(_ urls: [URL]) {
    urls.forEach { url in
      downloadQueue.addOperation {
        self.cancellablesQueue.sync {
          URLSession.shared.dataTaskPublisher(for: url)
            .map { CachedURLResponse(response: $0.response, data: $0.data) }
            .sink(
              receiveCompletion: { _ in () },
              receiveValue: { [self] in cache.storeCachedResponse($0, for: URLRequest(url: url)) }
            )
            .store(in: &self.cancellables)
        }
      }
    }
  }
}