在通用Swift函数中进行类型转换

Question

假设我有一个具有相同属性的UICollectionViewCell和UITableViewCell。 除了可以使用两个函数填充这些单元格之外，我还可以使用一个泛型来获取某些东西，确定是什么，然后将其转换为正确的东西，以在返回之前对其执行操作吗？

我的想法是：

func setUpCell<T>(event: Event, cell:T) -> T {

    // figure out what T is and cast it
    cell.event.bar = event.bar
    return cell

}

这是避免大量代码重复的好方法吗？

Answer 1

根据您的模型类型

struct Event {
    let title: String
    let desc: String
}

定义这个协议

protocol EventCell: class {
    var id: String? { get set }
    var desc: String? { get set }
}

现在使您的UITabelViewCell和UICollectionViewCell符合它

class TableCell: UITableViewController, EventCell {
    var id: String?
    var desc: String?
}

class CollectionCell: UICollectionViewCell, EventCell {
    var id: String?
    var desc: String?
}

最后定义这个扩展

extension EventCell {
    func populate(event:Event) {
        self.id = event.id
        self.desc = event.desc
    }
}

而已。 现在，您的两个单元格（ UITabelViewCell和UICollectionViewCell ）都具有populate方法！

Answer 2

这符合您的想法吗？

import UIKit

struct Event {
  var bar:Int = 0
}

// Protocol to group common additions
protocol ViewCellAdditions {
  init()
  var separatorInset:Int { get set }
  var event:Event { get set}
}

// Generic function to work on any class that adopts ViewCellAdditions
func setUpCell<T: ViewCellAdditions>(event: Event, cell:T, foo:Int) -> T {
  var newCell = T()
  newCell.separatorInset = foo
  newCell.event.bar = event.bar
  return newCell
}

// Class that adopts ViewCellAdditions
class NewCellClass: ViewCellAdditions {
  required init() {}
  var separatorInset:Int = 10
  var event:Event = Event()
}

// How to use it
let aCell = NewCellClass()
let aEvent = Event()
let newCell = setUpCell(aEvent, cell: aCell, foo: 5)