之前写过一篇Swift中的MVC与MVVM
里面说到MVVM目的是为了让我们的VC减负，不要那么臃肿，但是文章当时的实例可以说是一个不完整的MVVM，因为tableView中的代理还是设置为VC，还是需要VC去做获取数据的操作
为VC减负最彻底的方法就是将tableView的代理完全分离出去

• 新建一个类去继承NSObject, UITableViewDataSource, UITableViewDelegate
  （这里我直接放到VM里面了）

public class ExpandAndCheckBoxViewModel: NSObject, UITableViewDataSource, UITableViewDelegate

• 完成必要的方法

    public func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
        let cell: TableViewCell_checkbox = tableView.dequeueReusableCell(for: indexPath)
        let section = models[indexPath.section]
        let row = section.rows[indexPath.row]
        cell.setupCell(model: row)
        return cell
    }

    public func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
        return models[section].expand ? models[section].rows.count : 0
    }

    public func numberOfSections(in tableView: UITableView) -> Int {
        return models.count
    }

• 将tableView中的代理设为新建的这个类

private lazy var viewModel = ExpandAndCheckBoxViewModel(delegate: self)
    
    private lazy var tableView: UITableView = {
        let tableView = UITableView(frame: UIScreen.main.bounds, style: .plain)
        tableView.register(TableViewCell_checkbox.self)
        tableView.register(ExpandHeader.self)
        tableView.tableFooterView = UIView()
        tableView.delegate = viewModel
        tableView.dataSource = viewModel
        return tableView
    }()

• 将VC中原有的相关代码删除
  • VC中除了setup UITableView的方法（UITableView DataSource、Delegate都可以从VC中删除了）

你会发现VC瞬间少了很多行代码，实现了VC的瘦身
同样的方法可以用在UICollectionView里面

如果大家有其他解决方案欢迎留言交流
支持原创，版权所有