1.控件動畫的封裝

比如UIButton，UILabel，UITableViewCell當(dāng)用戶觸碰這些控件的時候，我們的需求是要給用戶展示彈跳動畫。如果每個控件都去實現(xiàn)這個動畫，那么就會出現(xiàn)代碼重復(fù)，并且用繼承去實現(xiàn)這個功能會有明顯的缺點，接下來我們用面向協(xié)議來封裝一下：

import UIKit
//添加一個動畫協(xié)議
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)
    }
}

只要某個控件需要這個動畫，那么它就可以遵守上面我們添加的協(xié)議，當(dāng)前控件就會有這個彈跳動畫的能力，如下：

//MyLabel遵守AnimationBeat協(xié)議
  import UIKit
  class MyLabel: UILabel, AnimationBeat {
  }

//接下來我們就可以在控制器中調(diào)用動畫的方法了
  let myLable = MyLabel()
  //調(diào)用協(xié)議中的動畫
  myLable.animationBeat()

2.加載 Xib 的封裝

對于一個項目來說肯定避免不了會用到Xib，如果項目用到了50個Xib，就會寫出類似50個重復(fù)的代碼，如下：

extension ShopView {
    class func loadXib() -> ShopView {
        return Bundle.main.loadNibNamed("\(self)", owner: nil, options: nil)?.first as! ShopView
    }
}

因為加載Xib的方法寫的太麻煩，所以封裝一個類方法，方便外界調(diào)用，但是上面這個方法會在很多地方重復(fù)出現(xiàn)。那我們就可以用協(xié)議來解決這個重復(fù)代碼的問題，如下：

protocol LoadXibProtocol {
}
//為上面定義的協(xié)議添加一個擴展
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
    }
}

我們創(chuàng)建一個Swift文件專門添加協(xié)議，在這個文件中添加上面這段代碼，之后每個View只要遵守這個LoadXibProtocol協(xié)議就能調(diào)用loadXib方法，如下：

import UIKit
///遵守LoadXibProtocol協(xié)議
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 的注冊標(biāo)識封裝

在實際開發(fā)中，很多地方都會用到UITableView、UICollectionView，并且每次使用都要自己注冊下Cell，有時候還想不起來怎么寫注冊標(biāo)識，寫多了的話還是比較煩的。接下來我們就用面向協(xié)議來封裝Cell注冊標(biāo)識，以后就不用再手寫注冊標(biāo)識了，如下：

import UIKit
//定義一個協(xié)議
protocol RegisterRepeat {
}

//為協(xié)議擴展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協(xié)議，所以通過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協(xié)議，為這個協(xié)議擴展了2個屬性，identifier就是自動返回遵守當(dāng)前協(xié)議的Cell標(biāo)識，xib就是如果從xib加載Cell時需要用到的一個屬性。接下來我們會為UITableView擴展2個方法，lr_registerCell是注冊Cell的方法，在這個方法中我們添加T這個泛型（這個T必須傳入的是UITableViewCell類型），并且這個泛型T需要遵守RegisterRepeat這個協(xié)議。最后我們通過判斷是否是從Xib中加載，然后在進行Cell注冊。同理lr_dequeueReusableCell就是取出這個泛型T，并且需要遵守RegisterRepeat協(xié)議。接下來我們的自定義Cell只要遵守這個RegisterRepeat協(xié)議就可以了，如下：

//第一個自定義的Cell，代碼加載（非Xib加載）
import UIKit
class ShopViewCell: UITableViewCell, RegisterRepeat {
}

//第二個自定義的Cell，Xib加載
import UIKit
class HomeViewCell: UITableViewCell, RegisterRepeat {
    //因為這個cell是從Xib中加載，需要重寫RegisterRepeat協(xié)議中的xib屬性，給它賦值。
    static var xib: UINib? {
        return UINib(nibName: "HomeViewCell", bundle: nil)
    }
}

接下來我們就可以在控制器實現(xiàn)下我們擴展的方法：

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")
        //通過協(xié)議代碼加載Cell
        // myTableView.lr_registerCell(cell: ShopViewCell.self)
        //通過協(xié)議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)
        // 通過協(xié)議取出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方法再也不用去關(guān)心如何去寫標(biāo)識符，你只要把遵守RegisterRepeat協(xié)議的Cell類名給我，我就會自動幫你添加標(biāo)識符。同理取Cell的時候也不用關(guān)心Cell的標(biāo)識符了。

參考： Practical Protocol-Oriented-Programming