Swift 面向协议开发笔记
1.控件动画的封装
比如UIButton，UILabel，UITableViewCell当用户触碰这些控件的时候，我们的需求是要给用户展示弹跳动画。如果每个控件都去实现这个动画，那么就会出现代码重复，并且用继承去实现这个功能会有明显的缺点，接下来我们用面向协议来封装一下：
import UIKit
//添加一个动画协议
protocol AnimationBeat {
}
//添加扩展
extension AnimationBeat where Self: UIView {
//随便添加一个弹跳动画
func animationBeat() {
self.transform = CGAffineTransform()
UIView.animateKeyframes(withDuration: 0.3, delay: 0, options: UIViewKeyframeAnimationOptions(rawValue: 0), animations: {
UIView.addKeyframe(withRelativeStartTime: 0, relativeDuration: 1 / 3.0, animations: {
self.transform = CGAffineTransform(scaleX: 1.1, y: 1.1)
})
UIView.addKeyframe(withRelativeStartTime: 1 / 3.0, relativeDuration: 1 / 3.0, animations: {
self.transform = CGAffineTransform(scaleX: 0.9, y: 0.9)
})
UIView.addKeyframe(withRelativeStartTime: 2 / 3.0, relativeDuration: 1 / 3.0, animations: {
self.transform = CGAffineTransform(scaleX: 1.0, y: 1.0)
})
}, completion: nil)
}
}
只要某个控件需要这个动画，那么它就可以遵守上面我们添加的协议，当前控件就会有这个弹跳动画的能力，如下：
//MyLabel遵守AnimationBeat协议
import UIKit
class MyLabel: UILabel, AnimationBeat {
}
//接下来我们就可以在控制器中调用动画的方法了
let myLable = MyLabel()
//调用协议中的动画
myLable.animationBeat()
2.加载 Xib 的封装
对于一个项目来说肯定避免不了会用到Xib，如果项目用到了50个Xib，就会写出类似50个重复的代码，如下：
extension ShopView {
class func loadXib() -> ShopView {
return Bundle.main.loadNibNamed("\(self)", owner: nil, options: nil)?.first as! ShopView
}
}
因为加载Xib的方法写的太麻烦，所以封装一个类方法，方便外界调用，但是上面这个方法会在很多地方重复出现。那我们就可以用协议来解决这个重复代码的问题，如下：
protocol LoadXibProtocol {
}
//为上面定义的协议添加一个扩展
extension LoadXibProtocol where Self: UIView {
///提供加载XIB方法
static func loadXib(xibStr: String? = nil) -> Self {
return Bundle.main.loadNibNamed(xibStr ?? "\(self)", owner: nil, options: nil)?.last as! Self
}
}
我们创建一个Swift文件专门添加协议，在这个文件中添加上面这段代码，之后每个View只要遵守这个LoadXibProtocol协议就能调用loadXib方法，如下：
import UIKit
///遵守LoadXibProtocol协议
class ShopView: UIView, LoadXibProtocol {
}
那接下来我们使用就非常简单了，并且提供了2种加载Xib的方式，看实际情况来使用：
import UIKit
class ViewController: UIViewController {
override func viewDidLoad() {
super.viewDidLoad()
//第一种方法
let shopView = ShopView.loadXib()
//第二种方法
let shopView = ShopView.loadXib(xibStr: "ShopView")
view.addSubview(shopView)
}
}
3.Cell 的注册标识封装
在实际开发中，很多地方都会用到UITableView、UICollectionView，并且每次使用都要自己注册下Cell，有时候还想不起来怎么写注册标识，写多了的话还是比较烦的。接下来我们就用面向协议来封装Cell注册标识，以后就不用再手写注册标识了，如下：
import UIKit
//定义一个协议
protocol RegisterRepeat {
}
//为协议扩展identifier属性和xib属性
extension RegisterRepeat {
static var identifier: String {
return "\(self)"
}
static var xib: UINib? {
return nil
}
}
//为UITableView扩展一个方法
extension UITableView {
func lr_registerCell<T: UITableViewCell>(cell: T.Type) where T: RegisterRepeat {
if let xib = T.xib {
// T遵守了RegisterRepeat协议，所以通过T就能取出identifier这个属性
register(xib, forCellReuseIdentifier: T.identifier)
}else {
register(cell, forCellReuseIdentifier: T.identifier)
}
}
func lr_dequeueReusableCell<T: UITableViewCell>(indexPath: IndexPath) -> T where T: RegisterRepeat {
return dequeueReusableCell(withIdentifier: T.identifier, for: indexPath) as! T
}
}
首先我们定义了一个RegisterRepeat协议，为这个协议扩展了2个属性，identifier就是自动返回遵守当前协议的Cell标识，xib就是如果从xib加载Cell时需要用到的一个属性。接下来我们会为UITableView扩展2个方法，lr_registerCell是注册Cell的方法，在这个方法中我们添加T这个泛型（这个T必须传入的是UITableViewCell类型），并且这个泛型T需要遵守RegisterRepeat这个协议。最后我们通过判断是否是从Xib中加载，然后在进行Cell注册。同理lr_dequeueReusableCell就是取出这个泛型T，并且需要遵守RegisterRepeat协议。接下来我们的自定义Cell只要遵守这个RegisterRepeat协议就可以了，如下：
//第一个自定义的Cell，代码加载（非Xib加载）
import UIKit
class ShopViewCell: UITableViewCell, RegisterRepeat {
}
//第二个自定义的Cell，Xib加载
import UIKit
class HomeViewCell: UITableViewCell, RegisterRepeat {
//因为这个cell是从Xib中加载，需要重写RegisterRepeat协议中的xib属性，给它赋值。
static var xib: UINib? {
return UINib(nibName: "HomeViewCell", bundle: nil)
}
}
接下来我们就可以在控制器实现下我们扩展的方法：
import UIKit
class ViewController: UIViewController {
override func viewDidLoad() {
super.viewDidLoad()
let myTableView = UITableView(frame: view.bounds, style: .plain)
myTableView.delegate = self
myTableView.dataSource = self
// myTableView.register(UITableViewCell.self, forCellReuseIdentifier: "myTableViewID")
//通过协议代码加载Cell
// myTableView.lr_registerCell(cell: ShopViewCell.self)
//通过协议Xib加载Cell
myTableView.lr_registerCell(cell: HomeViewCell.self)
view.addSubview(myTableView)
}
}
extension ViewController: UITableViewDelegate, UITableViewDataSource {
func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
return 10
}
func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
// let cell = tableView.dequeueReusableCell(withIdentifier: "myTableViewID", for: indexPath)
// 通过协议取出Cell
let cell = tableView.lr_dequeueReusableCell(indexPath: indexPath) as HomeViewCell
cell.textLabel?.text = "\(indexPath.row)"
return cell
}
func tableView(_ tableView: UITableView, heightForRowAt indexPath: IndexPath) -> CGFloat {
return 40
}
}
从上面的代码来看，我们注册的Cell方法再也不用去关心如何去写标识符，你只要把遵守RegisterRepeat协议的Cell类名给我，我就会自动帮你添加标识符。同理取Cell的时候也不用关心Cell的标识符了。
