How can I use switch statement to implement collectionView function in Swift?

Question

I am new to swift and trying to implement collectionView function in a ViewController.Now I have three types of UICollectionViewCell,here is my code:

func co         


        

Answer 1

Hope this help. Not testing at all. If any issues, can you take a snip ?

func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {

  let cell:UICollectionViewCell

  switch collectionView {
    case self.collectionViewCategory:
        let categoryCell = collectionView.dequeueReusableCell(withReuseIdentifier: categoryIdentifier, for: indexPath as IndexPath) as! CategoryCell
        categoryCell.CategoryIcon.image = self.categoryItems[indexPath.item]
        cell = categoryCell 
    case self.collectionViewHour:
        let hourCell = collectionView.dequeueReusableCell(withReuseIdentifier: hourIdentifier, for: indexPath as IndexPath) as! HourCell
        hourCell.hourItem.text = self.hourItems[indexPath.item]
        cell = hourCell
    default:
        //do nothing.
  }
  cell.layer.borderColor = UIColor.lightGray.cgColor
  cell.layer.borderWidth = 1
  cell.layer.cornerRadius = 5
  return cell

}