需求：假设你要写一个由一张图片和一个按钮构成的简单应用，产品经理希望按钮被点击的时候图片会抖动
实现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提倡协议编程，同时为你的代码库增加了超级棒的可读性，复用性和可维护性。
个人意见：实现一个功能，如果一个地方用，可以直接写实现，如果两个地方用，可以写一个拓展，也可以实现一个协议，看自己习惯和方便。
