Swift - 扩展

以前在OC开发中,系统提供了Category,可以对类进行扩充方法; 而Swift中没有了这个概念,取而代之的是扩展(Extension),功能也是对类进行方法的扩展
下面我们进行Swift的扩展(以扩展UIButton为例)

1.选择下图红色选种按钮 并点击 Next

屏幕快照 2017-03-21 下午9.35.47.png

2.暂且起一个 UIButton-Extension 的类名把

屏幕快照 2017-03-21 下午9.37.15.png

3.在UIButton-Extension类中,我们敲 extension UIButton 发现没有提示,这是因为该类继承自 Foundation , 故把Foundation 改为UIKit就可以了

屏幕快照 2017-03-21 下午9.37.51.png

下图就有提示

屏幕快照 2017-03-21 下午9.38.22.png

4.创建便利函数
convenience : 便利 使用convenience修饰的构造函数,叫做便利构造函数
便利构造函数,通常用在对系统的类进行构造函数的扩充时使用

 便利构造函数的特点:
 1.便利构造函数通常是写在extension里面
 2.便利构造函数init前面需要加convenience
 3.在便利构造函数中需要明确的调用self.init()

extension UIButton {
    convenience init(imageName:String,bgImageName:String) {
        self.init()

        setImage(UIImage(named:imageName), for: .normal)
        setBackgroundImage(UIImage(named:bgImageName), for: .normal)
        sizeToFit() 
    }
}

5.在ViewController中调用

class ViewController: UIViewController {
    lazy var btn : UIButton = UIButton(imageName: "compose_icon_add", bgImageName: "compose_button")

    override func viewDidLoad() {
        super.viewDidLoad()

        btn.center = view.center
        view.addSubview(btn)  
    }

}

感觉比OC类别好用