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

Question

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

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

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扩展中的两件事：

1. 调用MyProtocol和UIViewController API
2. 覆盖UIViewController方法initialize()以便能够swizzle viewWillAppear:

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

1. 我们不能扩展有条件的类 （即extension UIViewController where Self: MyProtocol ）
2. 如果我们改为扩展协议 ，我们可以extension MyProtocol where Self: UIViewController添加条件extension MyProtocol where Self: UIViewController但是我们不能覆盖协议扩展中类的方法 ，这意味着我们不能public override class func initialize() swizzling所需的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
        }

    }

}

缺点：

• 不是“迅捷的做事方式”
• 调用swizzling方法并对所有UIViewControllers生效，即使我们只验证那些符合MyProtocol协议的运行敏感代码行的方法。

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