假设我有一个从公共(public)父类(super class)继承的对象集合(在这种情况下这比协议(protocol)更可取):
class ObjectSuperClass {
type: ObjectType
}
class ObjectClass1: ObjectSuperClass {
type = .Type1
}
class ObjectClass2: ObjectSuperClass {
type = .Type2
}
我希望创建一个通用的搜索功能,如下所示:
func objectsOfType<T: ObjectSuperClass>(T.class, otherFilter: Any?) -> [T]
可用于搜索给定的子类型,返回更具体的结果数组:
let result = objectsOfType(ObjectClass2.class, otherFilter: nil) -> [ObjectClass2]
(pseudo-swift)
我觉得这是泛型可以提供帮助的地方,但看不出应该在哪里放置约束。可能吗?
最佳答案
这很有效...
func filterType<T>(list: [AnyObject]) -> [T]
{
return list.filter{ $0 is T }.map{ $0 as! T }
}
...假设您将结果分配给已明确键入的内容,如以下示例所示:
class ObjectSuperClass: CustomStringConvertible
{
let myType: String
init(aString: String)
{
myType = aString
}
var description: String { return myType }
}
class ObjectClass1: ObjectSuperClass
{
init()
{
super.init(aString: "<t 1>")
}
}
class ObjectClass2: ObjectSuperClass
{
init()
{
super.init(aString: "<t 2>")
}
}
let unfilteredList: [AnyObject] = [ ObjectClass1(), ObjectClass2(), ObjectSuperClass(aString: "<Who knows>")]
let filteredList1: [ObjectClass1] = filterType(list: unfilteredList)
print("\(filteredList1)") // <t 1>
let filteredList2: [ObjectClass2] = filterType(list: unfilteredList)
print("\(filteredList2)") // <t 2>
let filteredList3: [ObjectSuperClass] = filterType(list: unfilteredList)
print("\(filteredList3)") // [<t 1>, <t 2>, <Who knows>]
T 在每种情况下都是从请求的返回类型中推断出来的。该函数本身会根据元素是否属于所需类型来过滤原始数组,然后将过滤后的结果强制转换为正确的类型。
如果你想要一个“额外的过滤器”,你不需要显式输入结果,只要 T 可以从你的额外过滤器函数中推断出来。
func extraFilterType<T>(list: [AnyObject], extraFilter: T -> Bool) -> [T]
{
return list.filter{ $0 is T }.map{ $0 as! T }.filter(extraFilter)
}
let filteredList = extraFilterType(unfilteredList){
(element : ObjectClass2) -> Bool in
!element.description.isEmpty
}
print("\(filteredList)") // <t 2>
编辑
filterType
函数的精巧版本将使用 flatMap()
func filterType<T>(list: [Any]) -> [T]
{
return list.flatMap{ $0 as? T }
}
编辑 2
自 Swift 4.something 以来,Flatmap 不推荐用于可选值,请使用 compactMap
func filterType<T>(list: [Any]) -> [T]
{
return list.compactMap{ $0 as? T }
}
关于Swift 泛型 : return type based on parameter type,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/34591554/