[Swift]正确的使用 extension 的方式

官网文档:Swift - Extensions

extension可以做什么:

• Add computed instance properties and computed type properties (添加计算型实例属性和计算型类型属性)
• Define instance methods and type methods (定义实例方法和类型方法)
• Provide new initializers (提供新的构造器)
• Define subscripts (定义下标)
• Define and use new nested types (定义和使用新的嵌套类型)
• Make an existing type conform to a protocol (使已存在的类型遵守某个协议)

几个比较方便的 extension 使用技巧

• 1 .划分区域,区分代码功能

  • 比如:

    class AppDelegate: UIResponder {
    
    var window: UIWindow?
    }
    ///实现 App 的声明周期
    extension AppDelegate: UIApplicationDelegate {
        func application(_ application: UIApplication, didFinishLaunchingWithOptions launchOptions: [UIApplication.LaunchOptionsKey: Any]?) -> Bool {
            return true
        }
        func applicationWillResignActive(_ application: UIApplication) {
        }
    }
    extension AppDelegate {
        func thirdSetting() {
            ///配置第三方信息
        }
    }
    复制代码

    class ViewController: UIViewController {
        fileprivate (set) var tableView: UITableView!
    }
    typealias DataSource = ViewController
    extension DataSource: UITableViewDataSource, UITableViewDelegate {
        public func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
            ///code
        }
    
        public func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
            ///code
        }
    }
    复制代码

• 2.像 OC 一样添加分类

  OC的 Category 添加类方法和对象方法时一般会在方法前添加前缀,比如 SDWebImage 中的 UIImageView+WebCache 方法都是以 sd_开头,ReactiveObjC都是以 rac_开头,当然 swift 的 extension 也可以这样使用前缀来区分系统方法和自定义方法,不过很多大神已经不这么做了,比如 RxCocoa 里的 rx,Kinfisher 的 kf.使用如以下代码:

      ///Rx
      let textFiled = UITextField()
      ///textFiled的 text 有变化的时候会调用 block
      let dispose = textFiled.rx.text.subscribe(onNext: { (text) in
          print(text)
      }, onError: { (error) in
          
      }, onCompleted: {
          
      }, onDisposed: nil)
      dispose.dispose()
  复制代码

      ///Kingfisher
      let imgView = UIImageView()
      ///异步设置图片
      let url = URL(string: "xxx")
      imgView.kf.setImage(with: url)
  复制代码

  查看 Kinfisher 和 RxCocoa 的源码,这个 rx 和 kf 大概是以下这种形式的代码:

  ///这个就是 kf 和 rx 的定义
  ///
  public struct AriSwift {
      public let base: Base
      public init(_ base: Base) {self.base = base}
  }
  public protocol AriSwiftCompatible {
      associatedtype CompatibleType
      static var `as`: AriSwift.Type { get set }
      var `as`: AriSwift { get set }
  }
  
  public extension AriSwiftCompatible {
      static var `as`: AriSwift.Type {
          get {return AriSwift.self}
          set {}
      }
      var `as`: AriSwift {
          get {return AriSwift(self)}
          set {}
      }
  }
  ///为所有继承 NSObject的类添加 as 的实例对象和类对象
  extension NSObject: AriSwiftCompatible {}
  复制代码

  ///给AriSwift 添加Base是 UIScreen 的拓展
  public extension AriSwift where Base: UIScreen {
      public static var width: CGFloat {return UIScreen.main.bounds.width}
      public static var height: CGFloat {return UIScreen.main.bounds.height}
  }
  
  ///调用 
  UIScreen.as.width /// 414.0
  复制代码

  在 Swift 中如果经常会用到结构体,为结构体Base 为结构体的对象添加分类时就不能使用 Base: XX 这种形式了,以 String 为例,代码如下

  ///记得 String 要遵守下这个协议
  extension String: AriSwiftCompatible {}
  extension AriSwift where Base == String {
      ///传入时间(秒)返回字符串格式
      ///```
      ///e.g.  180 -> "03:00"
      ///```
      static func formatTime(with seconds: Int) -> String {
          let hour = seconds / 3600
          let minut = (seconds % 3600)/60
          let second = seconds % 60
          if hour == 0 {
              return String(format: "%02zd:%02zd", minut,second)
          }else{
              return String(format: "%zd:%02zd:%02zd",hour,minut,second)
          }
      }
  }
  复制代码

  使用这种方式添加extension 能很好的区分自己写的代码和系统的代码,也可以代替前缀.

  还可以使用这个方法在我们的项目中封装 Objc 的第三方框架,比如 MJRefresh,MBProgressHUD

  import AriSwift
  import MJRefresh
  extension AriSwift where Base: UIScrollView {
      func addHeader(with target: Any, action: Selector){
          base.mj_header = MJRefreshNormalHeader(refreshingTarget: target, refreshingAction: action)
      }
      func addHeader(with action: @escaping (()->()) ) {
          base.mj_header = MJRefreshNormalHeader(refreshingBlock: {
              action()
          })
      }
      func addFooter(with target: Any, action: Selector){
          base.mj_footer = MJRefreshBackStateFooter(refreshingTarget: target, refreshingAction: action)
      }
      func addFooter(with action: @escaping (()->()) ){
          base.mj_footer = MJRefreshBackStateFooter(refreshingBlock: {
              action()
          })
      }
      func headerEndRefreshing(){
          base.mj_header.endRefreshing()
      }
      func footerEndRefreshing(){
          base.mj_footer.endRefreshing()
      }
      func footerEndRefreshingWithNoMoreData(){
          base.mj_footer.endRefreshingWithNoMoreData()
      }
  }
  复制代码

  import AriSwift
  import MBProgressHUD
  extension AriSwift where Base: UIViewController {
      func showHud(message: String, to view: UIView? = nil) -> MBProgressHUD{
          let targetView = view ?? UIApplication.shared.keyWindow!
          let hud = MBProgressHUD.showAdded(to: targetView, animated: true)
          hud.label.text = message
          hud.removeFromSuperViewOnHide = true
          return hud
      }
  }
  复制代码

  这样也可以减少项目中使用这些第三方的地方频繁的 import

  p.s. 我自己制作常用的 extension 工具库 AriSwift

  git: github.com/AnnieAri/Ar…

  也可以使用 pod 'AriSwift' 安装到项目中

转载于:https://juejin.im/post/5ca56e076fb9a05e6835d5a0