UITableView clearing the subview's background color on selection

UITableView+Select.swift

// UITableViewCell changes the background color of all sub views when cell is selected or highlighted.
// You can Solve this problem by overriding Tableview cell's setSelected:animated and setHighlighted:animated and resetting view background color.


// In Swift 3.1 :
override func setSelected(_ selected: Bool, animated: Bool) {
    let color = yourView.backgroundColor         
    super.setSelected(selected, animated: animated)

    if selected {
        yourView.backgroundColor = color
    }
}

override func setHighlighted(_ highlighted: Bool, animated: Bool) {
    let color = yourView.backgroundColor
    super.setHighlighted(highlighted, animated: animated)

    if highlighted {
        yourView.backgroundColor = color
    }
}