iOS中如何修改UINavigationController的根视图控制器并避免内存浪费？

Hey there! Let's break down how to properly switch your root view controller while keeping memory usage in check.

1. Fix the Runtime Root Controller Switch #

When you need to switch to MusicPage after your app has launched, simply replacing the window's root view controller works, but we can add cleanup steps to ensure the old TabBarView and its navigation controller get properly released:

// Trigger this code when your condition is met
func switchToMusicPage() {
    // Create your new destination controller
    let musicPage = MusicPage()
    let newNavController = UINavigationController(rootViewController: musicPage)
    
    // Clean up the old root controller stack first
    if let oldNav = window?.rootViewController as? UINavigationController {
        // Clear out all view controllers in the old nav stack
        oldNav.viewControllers.removeAll()
        
        // If your TabBarView has ongoing tasks (timers, network calls, observers), add a cleanup method
        if let tabBarView = oldNav.viewControllers.first as? TabBarView {
            tabBarView.stopOngoingTasks() // Replace with your actual cleanup method name
        }
    }
    
    // Finally, switch the root view controller
    window?.rootViewController = newNavController
}

2. Optimize Initial Root Controller Setup #

If you can check the condition when the app first launches, it's better to set the correct root controller upfront instead of switching later. This avoids creating unnecessary controllers entirely:

func application(_ application: UIApplication, didFinishLaunchingWithOptions launchOptions: [UIApplication.LaunchOptionsKey: Any]?) -> Bool {
    window = UIWindow(frame: UIScreen.main.bounds)
    window?.makeKeyAndVisible()
    
    // Check your condition first
    let shouldLaunchToMusicPage = checkYourCondition() // Replace with your actual condition logic
    
    if shouldLaunchToMusicPage {
        let musicPage = MusicPage()
        navigationController = UINavigationController(rootViewController: musicPage)
    } else {
        let tabBarView = TabBarView()
        navigationController = UINavigationController(rootViewController: tabBarView)
    }
    
    window?.rootViewController = navigationController
    return true
}

3. Memory Management Notes #

• Automatic Reference Counting (ARC) will handle releasing the old TabBarView and its navigation controller as long as there are no strong references keeping them alive. Make sure you don't store these controllers in global variables or capture them in retaining closures.
• Manual Cleanup: Always stop timers, cancel network requests, and remove any NotificationCenter observers in your TabBarView (and its child controllers) when they're no longer needed. This prevents memory leaks even if ARC tries to release the controller.

内容的提问来源于stack exchange，提问作者Pratik