Updating the UI

Swift Chat - 11 - Updating the UI.swift

extension MessagingViewController: MessagingDataSourceDelegate {
    
    func dataSourceDidUpdate() {
        tableView.reloadData()
        scrollToLastMessage()
    }
    
    private func scrollToLastMessage() {
        let rowCount = tableView.numberOfRows(inSection: 0)
        if rowCount > 1 {
            let lastIndexPath = IndexPath(row: rowCount - 1, section: 0)
            tableView.scrollToRow(at: lastIndexPath, at: .bottom, animated: true)
        }
    }
}