Question

你好=）我刚刚面临一个设计问题，我需要（基本上）做以下事情：

我想在符合协议viewWillAppear: 的任何UIViewController子类MyProtocol上注入一些代码。在代码中解释：

protocol MyProtocol
{
    func protocolFunction() {
        //do cool stuff...
    }
}

extension UIViewController where Self: MyProtocol //<-----compilation error
{
    public override class func initialize()
    {
        //swizzling stuff switching viewWillAppear(_: Bool) with xxx_viewWillAppear(animated: Bool)
    }

    //  MARK: - Swizzling

    func xxx_viewWillAppear(animated: Bool)
    {
        self.xxx_viewWillAppear(animated)

        //invoke APIs from  
        self.protocolFunction() // MyProtocol APIs
        let viewLoaded = self.isViewLoaded // UIViewController APIs

    }
}

这里的主要问题是我需要UIVIewController扩展名中的两件事：

同时调用MyProtocol和UIViewController API
覆盖UIViewController方法initialize()，以便能够调动viewWillAppear:

这两个功能似乎不兼容（从Swift 3开始），因为：

我们无法扩展条件（即extension UIViewController where Self: MyProtocol）
如果我们改为扩展协议，我们可以添加条件extension MyProtocol where Self: UIViewController但我们不能覆盖协议扩展中的类的方法，这意味着我们不能进行调整所需的public override class func initialize()。

所以我想知道是否有人可以为我面临的这个问题提供Swifty解决方案？ =）

提前致谢!!

Answer 1

你接近解决方案。只需要另辟蹊径。仅在UIViewController的一部分时扩展协议。

protocol MyProtocol
{
  func protocolFunction() {
    //do cool stuff...
  }
}

extension MyProtocol where Self: UIViewController {
  public override class func initialize()
  {
    //swizzling stuff switching viewWillAppear(_: Bool) with xxx_viewWillAppear(animated: Bool)
  }

  //  MARK: - Swizzling

  func xxx_viewWillAppear(animated: Bool)
  {
    self.xxx_viewWillAppear(animated)

    //invoke APIs from  
    self.protocolFunction() // MyProtocol APIs
    let viewLoaded = self.isViewLoaded // UIViewController APIs
  }
}

Answer 2

嗯，到目前为止，我还没有找到真正令人满意的方法，但我决定发布我最后为这个特定问题做的事情。简而言之，解决方案就是这样（使用原始示例代码）：

protocol MyProtocol
{
    func protocolFunction() {
        //do cool stuff...
    }
}

extension UIViewController //------->simple extension on UIViewController directly
{
    public override class func initialize()
    {
        //swizzling stuff switching viewWillAppear(_: Bool) with xxx_viewWillAppear(animated: Bool)
    }

    //  MARK: - Swizzling

    func xxx_viewWillAppear(animated: Bool)
    {
        self.xxx_viewWillAppear(animated)

        //------->only run when self conforms to MyProtocol
        if let protocolConformingSelf = self as? MyProtocol { 
            //invoke APIs from  
            protocolConformingSelf.protocolFunction() // MyProtocol APIs
            let viewLoaded = protocolConformingSelf.isViewLoaded // UIViewController APIs
        }

    }

}

缺点：

不是＆＃34; Swift方式＆＃34;做事
调用swizzling方法并对所有UIViewControllers生效，即使我们只验证那些符合MyProtocol协议的方法来运行敏感的代码行。

我非常希望它可以帮助那些面临类似情况的其他人=）

仅当符合特定协议时，才对类进行Swift扩展

2 个答案: