swift map学习

let array = [1, 2, 3, 4, 5]

现在有需求将array里的元素全部变成字符串

func convertString(_ array: [Int]) -> [String] {
  var result = [String]()
  for item in array {
    result.append(String(item))
  }
  return result
}

现在有需求将数组里的全部元素*2

func convertString(_ array: [Int]) -> [Int] {
  var result = [Int]()
  for item in array {
    result.append(item * 2)
  }
  return result
}

现在有需求将数组里的全部元素 + 10
....复制上面代码,改一点点.

可以通过系统map函数做到.

但是下面就自己写一个map函数当做学习
也是给Array做个扩展
从上面的各种需求来看,其实就是拿到数组里的每一个元素做处理
就从这开始着手:
定义一个泛型函数,函数需要返回什么类型,有调用者自己决定.

• 函数有一个参数,是一个block类型,这个block里会将数组的元素当做参数带出去,拿* 到这个参数需要做什么处理有调用者自己决定(就像上面的需求一样,或是乘以2 或是变成字符串,或者干嘛,随意)
  • block还有一个返回值,返回值是泛型类型,就是你拿到数组元素处理完以后的类型.
• 函数还有一个返回值,返回的是一个数组,就是处理完每一个元素以后得到的新值,数组类型也是有处理时的类型决定,也是一个泛型

extension Array {
  public typealias element = Array.SubSequence.Element
  func customMap<T>(_ transform: (element) -> T) -> [T] {
    var result = [T]()
    for item in self {
      result.append(transform(item))
    }
    return result
  }
}

使用:

[1,2,3,4,5].customMap({$0 * 2})
[1,2,3,4,5].customMap { (item) -> String in
  return String(item)
}

学习笔记.