[Refactor] #183 - DefaultImageTransformRepository actor로 변경

Author: OneTen19

Neki-iOS/Core/Sources/ImageTransform/Data/DefaultImageTransformRepository.swift

@@ -10,33 +10,32 @@ import Vision
 import ComposableArchitecture
 import CoreImage
 
-public final class DefaultImageTransformRepository: ImageTransformRepository {
+public actor DefaultImageTransformRepository: ImageTransformRepository {
 
-    private let modelTask: Task<VNCoreMLModel, Error>
+    private var cachedModel: VNCoreMLModel?
     private let ciContext = CIContext(options: [.cacheIntermediates: false])
 
-    public init() {
-        let config = MLModelConfiguration()
-        config.computeUnits = .all
+    public func transform(image inputImage: CGImage) async throws -> CGImage {
 
-        self.modelTask = Task {
+        if self.cachedModel == nil {
+            let config = MLModelConfiguration()
+            config.computeUnits = .all
+            
             let coreML = try whiteboxcartoonization(configuration: config).model
-            return try VNCoreMLModel(for: coreML)
+            self.cachedModel = try VNCoreMLModel(for: coreML)
         }
-    }
-    
-    public func transform(image inputImage: CGImage) async throws -> CGImage {
-        let visionModel = try await modelTask.value
 
-        return try await Task(priority: .userInitiated) {
-            let request = VNCoreMLRequest(model: visionModel)
-            request.imageCropAndScaleOption = .scaleFit
-            
-            let handler = VNImageRequestHandler(cgImage: inputImage)
-            try handler.perform([request])
-            
-            return try self.extractCGImage(from: request.results)
-        }.value
+        guard let visionModel = self.cachedModel else {
+            throw ImageTransformRepositoryError.renderingFailed
+        }
+        
+        let request = VNCoreMLRequest(model: visionModel)
+        request.imageCropAndScaleOption = .scaleFit
+        
+        let handler = VNImageRequestHandler(cgImage: inputImage)
+        try handler.perform([request])
+        
+        return try self.extractCGImage(from: request.results)
     }
 }