finally fix memory issues

gnatok · Jul 18, 2016 · abe8486 · abe8486
1 parent 5802cde
commit abe8486
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 2 deletions.
diff --git a/ViperMCFSwift/Modules/SecondModule/Configurator/SecondModuleConfigurator.swift b/ViperMCFSwift/Modules/SecondModule/Configurator/SecondModuleConfigurator.swift
@@ -20,6 +20,7 @@ class SecondModuleModuleConfigurator {
     private func configure(viewController: SecondModuleViewController) {
 
         let router = SecondModuleRouter()
+        router.transitionHandler = viewController
 
         let presenter = SecondModulePresenter()
         presenter.view = viewController

diff --git a/ViperMCFSwift/Modules/SecondModule/Router/SecondModuleRouter.swift b/ViperMCFSwift/Modules/SecondModule/Router/SecondModuleRouter.swift
@@ -9,5 +9,5 @@
 import ViperMcFlurry
 
 class SecondModuleRouter: NSObject, SecondModuleRouterInput {
-    var transitionHandler: RamblerViperModuleTransitionHandlerProtocol!
+    weak var transitionHandler: RamblerViperModuleTransitionHandlerProtocol!
 }
diff --git a/ViperMCFSwift/Modules/ThirdModule/Configurator/ThirdModuleConfigurator.swift b/ViperMCFSwift/Modules/ThirdModule/Configurator/ThirdModuleConfigurator.swift
@@ -20,6 +20,7 @@ class ThirdModuleModuleConfigurator {
     private func configure(viewController: ThirdModuleViewController) {
 
         let router = ThirdModuleRouter()
+        router.transitionHandler = viewController
 
         let presenter = ThirdModulePresenter()
         presenter.view = viewController

diff --git a/ViperMCFSwift/Modules/ThirdModule/Router/ThirdModuleRouter.swift b/ViperMCFSwift/Modules/ThirdModule/Router/ThirdModuleRouter.swift
@@ -8,4 +8,6 @@
 
 import ViperMcFlurry
 
-class ThirdModuleRouter: ThirdModuleRouterInput { }
+class ThirdModuleRouter: ThirdModuleRouterInput {
+    weak var transitionHandler: RamblerViperModuleTransitionHandlerProtocol!
+}