swift 面向协议编程

需求：假设你要写一个由一张图片和一个按钮构成的简单应用，产品经理希望按钮被点击的时候图片会抖动

实现1：写一个 UIImageView 的子类，再给它加上一个 shake() 方法

import UIKit
 
class MyImageView: UIImageView {
    // shake() 方法写在这儿
    func shake() {
        let animation = CABasicAnimation(keyPath: "position")
        animation.duration = 0.05
        animation.repeatCount = 5
        animation.autoreverses = true
        animation.fromValue = NSValue(CGPoint: CGPointMake(self.center.x - 4.0, self.center.y))
        animation.toValue = NSValue(CGPoint: CGPointMake(self.center.x + 4.0, self.center.y))
        layer.addAnimation(animation, forKey: "position")
    }
}

当用户点击按钮的时候，只需要调用 ImageView 的 shake 方法就ok了：

import UIKit
 
class ViewController: UIViewController {
    weak var myImageView: MyImageView!
    func onShakeButtonTap(sender: AnyObject) {
        // 在这里调用 shake 方法
        myImageView.shake()
    }
}

任务完成

功能拓展

像实际开发中会发生的那样，当你认为你搞定了任务，可以继续下一项的时候，设计师跳了出来告诉你他们希望按钮能够和 ImageView 一起抖动，当然可以重复上面的做法–写个 UIButton 的子类，再加个 shake 方法，为了快速实现功能可以，但这好像不太符合编程的复用要求。

实现二：用extension来对UIView进行拓展（UIImageView和UIButton都继承自UIView）

import UIKit
 
extension UIView {
    func shake() {
        let animation = CABasicAnimation(keyPath: "position")
        animation.duration = 0.05
        animation.repeatCount = 5
        animation.autoreverses = true
        animation.fromValue = NSValue(CGPoint: CGPointMake(self.center.x - 4.0, self.center.y))
        animation.toValue = NSValue(CGPoint: CGPointMake(self.center.x + 4.0, self.center.y))
        layer.addAnimation(animation, forKey: "position")
    }
}

现在，UIImageView 和 UIButton（以及其他所有视图）都有了可用的 shake() 方法：

class MyImageView: UIImageView {
    // 其他自定义写在这儿
}
 
class ActionButton: UIButton {
    // 其他自定义写在这儿
}
 
class ViewController: UIViewController {
 
    weak var myImageView: MyImageView!
    weak var actionButton: ActionButton!
    
    func onShakeButtonTap(sender: AnyObject) {
        myImageView.shake()
        actionButton.shake()
    }
}

实现三：用协议（Protocol）

Swifty 的解决方案就是用协议！可以利用协议拓展的力量来创建一个带有默认 shake() 方法实现的 Shakeable 协议

import UIKit

protocol Shakeable { }

// 你可以只为 UIView 添加 shake 方法！
extension Shakeable where Self: UIView {
    // shake 方法的默认实现
    func shake() {
        let animation = CABasicAnimation(keyPath: "position")
        animation.duration = 0.05
        animation.repeatCount = 5
        animation.autoreverses = true
        animation.fromValue = NSValue(CGPoint: CGPointMake(self.center.x - 4.0, self.center.y))
        animation.toValue = NSValue(CGPoint: CGPointMake(self.center.x + 4.0, self.center.y))
        layer.addAnimation(animation, forKey: "position")
    }
}

只需要让任何确实需要抖动的视图遵从 Shakeable 协议就好了：

class MyImageView: UIImageView, Shakeable {
    // 其他自定义写在这儿
}
 
class ActionButton: UIButton, Shakeable {
    // 其他自定义写在这儿
}
 
class ViewController: UIViewController {
 
    weak var myImageView: MyImageView!
    weak var actionButton: ActionButton!
    
    func onShakeButtonTap(sender: AnyObject) {
        myImageView.shake()
        actionButton.shake()
    }
}

仅仅通过 MyImageView 和 ActionButton 的类声明，你就能立刻知道它能抖动。

如果设计师跑过来表示希望在抖动的同时 ImageView 能暗淡一点儿，我们也能够利用相同的协议拓展模式添加新的功能，进行超级赞的功能组合。

// 添加暗淡功能
class MyImageView: UIImageView, Shakeable, Dimmable {
    // 其他实现写在这儿
}

当产品经理不再想让 ImageView 抖动的时候，重构起来也超级简单。只要移除对 Shakeable 协议的遵从就好了！

class MyImageView: UIImageView, Dimmable {
    // 其他实现写在这儿
}

结论

综上总结，协议还是很好用的，而且swift提倡协议编程，同时为你的代码库增加了超级棒的可读性，复用性和可维护性。

个人意见：实现一个功能，如果一个地方用，可以直接写实现，如果两个地方用，可以写一个拓展，也可以实现一个协议，看自己习惯和方便。

原文链接