iOS 在view中获取当前控制器

原创

mob64ca12e4594b 2024-09-26 06:51:53 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12e4594b的原创作品，请联系作者获取转载授权，否则将追究法律责任

如何在 iOS 的 View 中获取当前控制器

在 iOS 开发中，控制器（ViewController）是最主要的组件之一，负责管理视图和其交互。由于 iOS 的视图层次结构较为复杂，尤其是嵌套视图和多个视图控制器的嵌套，获取当前控制器成为一个常见需求。本文将介绍如何在一个视图中有效地获取当前控制器，并提供代码示例。

控制器的概念

在 iOS 中，控制器是 UIViewController 的实例，它用于管理和呈现视图（UIView）。每当我们创建一个新界面或修改现有界面时，我们通常会新建一个控制器实例并将其展示出来。在复杂的应用中，可能有多个控制器相互嵌套，这增加了获取当前控制器的复杂性。

获取当前控制器的方法

获取当前控制器的一种常用方法是递归地查找视图的下层控制器。以下是实现这一功能的具体方法。

代码示例

我们可以创建一个UIView的扩展，增加一个方法来获取当前的视图控制器：

import UIKit

extension UIView {
    func getCurrentViewController() -> UIViewController? {
        var responder: UIResponder? = self
        
        while responder != nil {
            if let viewController = responder as? UIViewController {
                return viewController
            }
            responder = responder?.next
        }
        return nil
    }
}

在上面的代码中，我们定义了一个 getCurrentViewController 方法，它利用了 UIResponder 机制，依次访问当前视图的响应者链，直到找到一个 UIViewController 实例。

使用示例

一旦我们有了这个方法，就可以在任何一个视图中使用它来获取当前控制器。例如：

let currentVC = someView.getCurrentViewController()
print("当前控制器是:\(String(describing: currentVC))")

在用户交互中，可能需要通过按钮点击来触发导航操作，我们可以像下面这样处理：

@IBAction func buttonClicked(_ sender: UIButton) {
    if let currentVC = self.getCurrentViewController() {
        let alert = UIAlertController(title: "提示", message: "您当前在控制器：\(String(describing: currentVC))", preferredStyle: .alert)
        alert.addAction(UIAlertAction(title: "确定", style: .default))
        currentVC.present(alert, animated: true, completion: nil)
    }
}

处理复杂的视图层次结构

在某些情况下，当前控制器可能生成于其它 UI 组件中（如 NavagationController）。因此，有时我们需要调整方法，考虑到这些情况。例如：

func getVisibleViewController(from vc: UIViewController?) -> UIViewController? {
    if let navController = vc as? UINavigationController {
        return getVisibleViewController(from: navController.visibleViewController)
    }
    
    if let tabController = vc as? UITabBarController {
        return getVisibleViewController(from: tabController.selectedViewController)
    }
    
    if let presentedVC = vc?.presentedViewController {
        return getVisibleViewController(from: presentedVC)
    }
    
    return vc
}

结合上述方法，我们可以在应用的任意层次中获取到当前控制器。

序列图

为了便于理解这个流程，下面是一个简单的序列图，表示获取当前控制器的过程：

sequenceDiagram
    participant View as View
    participant Responder as UIResponder
    participant Controller as UIViewController

    View->>Responder: 访问响应者链
    Responder->>Controller: 检查是否为控制器
    Controller-->>View: 返回当前控制器