缘由

iOS开发，我们常用的组件UITableView,UICollectionView，通过代理的方式实现数据的加载，如果我们在UIViewController委托代理，会导致代码的冗余，同时Controller过于庞大，不利于代码的阅读
Objective-C时代我们通过MVVM框架，实现Controller的瘦身，但是仍然会存在很多冗余的代码
Swift时代我们可以在MVVM的基础上更好的实现Controller的瘦身，减少冗余的代码

具体的实现方式可以参考MGTV-Swift，觉得不错赏个🌟

实现

ViewModelProtocol

• 定义通用的协议，主要用于CollectionViewMode和TableViewModel

protocol ViewModelProtocol {
    associatedtype DataType //定义用语TableView，CollectionView的数据模型
    
    var datas: [[DataType]] { get set } //用于存储数据的数组，二维数组，满足多个section的要求
    
    associatedtype ViewType //定义View的类型，主要有UICollectionView,UITableView
    associatedtype CellType //定义Cell的类型，主要有UITableViewCell，UICollectionViewCell
    
    func cellConfig(_ view: ViewType, datas: [[DataType]], indexPath: IndexPath) -> CellType //定义方法，用于返回cell
}

• 协议中并未定义Identifier来获取需要展示的cell，cell的获取部分交由外部进行提供，这样可以更灵活的适用于多样式的cell展示

CollectionViewModel

• 通过typealias重定义UICollectionView的协议
  typealias CollectionViewProtocol = UICollectionViewDelegate & UICollectionViewDataSource & UICollectionViewDelegateFlowLayout
• 因为extension不能的一些限制，所以通用的CollectionViewModel采用OOP的方式实现

class CollectionViewModel<T>: NSObject, ViewModelProtocol, CollectionViewProtocol {

    typealias DataType = T
    typealias ViewType = UICollectionView
    typealias CellType = UICollectionViewCell
    
    var datas: [[T]] = []
    //子类需要重写这个方法返回已经UICollectionViewCell
    func cellConfig(_ view: UICollectionView, datas: [[T]], indexPath: IndexPath) -> UICollectionViewCell {
        return UICollectionViewCell()
    }
    //可根据子类的样式重写需要的属性，主要用于Cell的样式
    var minimumLineSpacing: CGFloat { return 0 }
    var minimumInteritemSpacing: CGFloat { return 0 }
    var insetForSection: UIEdgeInsets { return UIEdgeInsets(top: 0, left: 0, bottom: 0, right: 0) }
    var itemSize: CGSize { return CGSize(width: 0, height: 0) }
    
    init(_ collectionDatas: [[T]]) {
        datas = collectionDatas
        super.init()
    }
    
    //MARK: - UICollectionView DataSource
    func numberOfSections(in collectionView: UICollectionView) -> Int {
        return datas.count
    }
    func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
        return datas[section].count
    }
    
    func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
        return cellConfig(collectionView, datas: datas, indexPath: indexPath)
    }
    
    //MARK: UICollectionView DelegateFlowlayout
    func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> CGSize {
        //默认是page类型
        let itemSizeWith = itemSize.width == 0 ? collectionView.bounds.size.width : itemSize.width
        let itemSizeHeight = itemSize.height == 0 ? collectionView.bounds.size.height : itemSize.height
        return CGSize(width: itemSizeWith, height: itemSizeHeight)
    }
    
    func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, insetForSectionAt section: Int) -> UIEdgeInsets {
        return insetForSection
    }
    
    func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, minimumLineSpacingForSectionAt section: Int) -> CGFloat {
        return minimumLineSpacing
    }
    
    func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, minimumInteritemSpacingForSectionAt section: Int) -> CGFloat {
        return minimumInteritemSpacing
    }
}

TableViewModel

• 通过typealias重定义UITableView的协议
  typealias TableViewProtocol = UITableViewDelegate & UITableViewDataSource

class TableViewModel<T>: NSObject, ViewModelProtocol, TableViewProtocol {
    typealias DataType = T
    
    typealias ViewType = UITableView
    typealias CellType = UITableViewCell
    
    var datas: [[T]] = []
    
    func cellConfig(_ view: UITableView, datas: [[T]], indexPath: IndexPath) -> UITableViewCell {
        return UITableViewCell()
    }
    
    func caculateHeight(_ indexPath: IndexPath) -> CGFloat {
        return 0
    }
    
    var cellHeight: CGFloat = -1
    
    init(_ tableDatas: [[T]]) {
        datas = tableDatas
        super.init()
    }
    
    //MARK: - UITableViewDelegate
    func tableView(_ tableView: UITableView, heightForRowAt indexPath: IndexPath) -> CGFloat {
        guard cellHeight > 0 else {
            return caculateHeight(indexPath)
        }
        return cellHeight
    }
    
    //MARK: - UITableViewDataSource
    func numberOfSections(in tableView: UITableView) -> Int {
        return datas.count
    }
    func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
        return datas[section].count
    }
    
    func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
        return cellConfig(tableView, datas: datas, indexPath: indexPath)
    }
}