Updating TableView Height Constraint

gistfile1.swift

@IBOutlet weak var tableView: UITableView!
@IBOutlet weak var tableViewHeightConstraint: NSLayoutConstraint!

var tableViewCell: UITableViewCell = tableView.cellForRowAtIndexPath(NSIndexPath(forRow: 0, inSection: 0))
var cellHeight: CGFloat = tableViewCell?.frame.height

var dataSource: [AnyObject]?

tableViewHeightConstraint.constant = calculateTableViewHeight(cellHeight, dataSource.count)

tableView.setNeedsLayout()
tableView.layoutIfNeeded()