Swift基础篇 - extension扩展

514 阅读1分钟

前言:

扩展为现有类、结构体、或协议添加新功能。这也包括了为无访问权限的源代码拓展类型的能力(即所谓的逆向建模)

扩展和OC中的category类似。(与OC的分类不同的是,Swift的扩展没有名字)

1.extension 的能力

  • 添加计算实例属性和计算类型属性
  • 定义实例方法和类型方法
  • 提供新初始化器
  • 定义下标
  • 定义和使用新内嵌类型
  • 使现有的类型遵循某协议

扩展可以向一个类型添加新的方法,但是不能重写已有的方法

1.1.为已有的类添加计算实例属性和计算类型属性

extension Double {

  var km : Double {

    return self / 1000.0

  }

}

let speed:Double = 1200.0

print(speed.km)
/*

日志打印:

1.2

*/

1.2.初始化器

  • 扩展可向已有类型添加新的初始化器。这允许扩展其他类型一使初始化器接受自定义类型作为形式参数,或提供该类型的原始实现中未包含的额外初始化选项
  • 扩展能为类添加新的便捷初始化器,但是不能为类添加指定初始化器或反初始化器,指定初始化器和反初始化器必须由原来类的实现提供
struct Size {

  var width = 0.0 ,height = 0.0

}

struct Point {

  var x = 0.0 , y = 0.0

}

struct Rect {

  var origin = Point()

  var size = Size()

}

extension Rect {

  init(center:Point,size : Size){

    let originX = center.x - (size.width / 2)

    let originY = center.y - (size.height / 2)

    self.init(origin: Point(x: originX, y: originY), size: size)

  }

}

let centerRect = Rect(center: Point(x: 3.0, y: 3.0), size: Size(width: 2.0, height: 2.0))

1.3为已有的类型添加新的实例方法和类型方法

extension Int {

  func repetitions(task:() -> Void) {

    for _ in 0..<self {

      task()

    }

  }

}

3.repetitions {

  print("hellow")

}

1.4.mutaing(异变方法)

扩展的实例方法仍可以修改(或异变)实例本身。结构体和枚举类型方法在修改self或本身的属性时必须标记实例方法为mutating,和原本实现的异变方法一样

extension Int {

  mutating func square() {

    self = self * self

  }

}

var someInt = 3

someInt.square()

1.5.下标

扩展能为已有类型添加新的下标

extension Int {

  subscript(digitIndex : Int) -> Int {

    var decimalBase = 1

    for _ in 0..<digitIndex {

      decimalBase *= 10

    }

    return (self / decimalBase) % 10

  }

}

746381295[1]

746381295[2]

746381295[8]

1.6.添加内嵌类型

extension Int {

  enum Kind {

    case nagative,zero,positive

  }

  var kind : Kind {

    switch self {

    case 0 :
      return .zero
    case let x where x > 0:
      return .positive
    default:
      return .nagative
    }
  }

}

print(0.kind)