前言

链式编程就是将多个方法用点语法链接起来,让代码更加简洁, 可读性更强.相信大家对Masonry都不陌生. 看下面这段代码:

make.width.height.equalTo(0);

相当于一句话调用了下面三个方法

- (MASConstraint *)width {
    return [self addConstraintWithLayoutAttribute:NSLayoutAttributeWidth];
}

- (MASConstraint *)height {
    return [self addConstraintWithLayoutAttribute:NSLayoutAttributeHeight];
}

- (MASConstraint * (^)(id))equalTo {
    return ^id(id attribute) {
        return self.equalToWithRelation(attribute, NSLayoutRelationEqual);
    };
}

原理

原理就是调用的属性或者方法的返回值是调用者本身, Masonry中的链式编程的特点是方法或者属性的返回值是block, 而block的返回值是调用者本身, block的参数就是需要处理的数据.

初探SnapKit

SnapKit和Masonry都是<a href=https://github.com/SnapKit>同一个作者 </a>, 在SnapKit中, 他也同样运用了链式编程思想. 我们看一个简单的例子, 比如我们设置一个控件的top和left距离父控件为1:

contentView.snp_makeConstraints { (make) in
           make.top.left.equalTo(1)
        }

进入left或者top所在的文件, 如图

 /// left edge
    public var left: ConstraintDescriptionExtendable { return self.makeConstraintDescription(ConstraintAttributes.Left) }
    
    /// top edge
    public var top: ConstraintDescriptionExtendable { return self.makeConstraintDescription(ConstraintAttributes.Top) }

我们可以再次查看到ConstraintDescriptionExtendable是一个protocol

public protocol ConstraintDescriptionExtendable: ConstraintDescriptionRelatable

继续进入makeConstraintDescription方法中

internal func makeConstraintDescription(attributes: ConstraintAttributes) -> ConstraintDescription {
        let item = ConstraintItem(object: self.view, attributes: attributes)
        let constraintDescription = ConstraintDescription(fromItem: item)
        self.constraintDescriptions.append(constraintDescription)
        return constraintDescription
    }

很明显方法的返回值是ConstraintDescription的实例对象. ConstraintDescription是什么呢, 我们进去看看:

internal class ConstraintDescription: ConstraintDescriptionExtendable, ConstraintDescriptionEditable, ConstraintDescriptionFinalizable

ok, 我们找到了我们想要的东西了.ConstraintDescription是一个遵守ConstraintDescriptionExtendable协议的类,恰好left和top等也是遵守该协议. 实际上和我们上面所述的链式编程的原理已经符合了, 每次我们调用left, top...等属性时, 都会返回其方法调用者本身.

  - 建议大家以Masonry为例,可能更容易理解一些

swift版链式编程计算器

假设我们需要给任意对象添加计算的方法

①.新建Calculater类用来处理计算和保存计算结果:

import UIKit

enum MyError: ErrorType {
    case DivZero // 除数为0
}

class Calculater: NSObject {
    // 用来保存结果的
    var result : Float = 0.0
    
    func add(num: Float) -> Calculater {
        result += num
        return self
    }
    
    func sub(num: Float) -> Calculater {
        result -= num
        return self
    }
    
    func mul(num: Float) -> Calculater {
        result *= num
        return self
    }
    
    func div(num: Float) throws -> Calculater {
        guard num != 0 else{
            throw MyError.DivZero
        }
        result /= num
        return self
    }
}

注意三点:
    1.每个方法的返回值是调用者本身
    2.除法需要判断除数为0的情况
    3.swift2之后抛出异常需要ErrorType类型

②. 给NSObject创建一个分类, 让所有的对象都有计算方法
NSObject+Calculate.swift文件如下:

import UIKit

extension NSObject
{
    // @noescape表示在函数return前closure生命周期结束，可以优化编译, swift2.1后引入的关键字
    class func sl_calculate(@noescape closure: (make: Calculater) -> Void) -> Float{
        let maker = Calculater()
        closure(make: maker)
      return maker.result
    }
}

关于@noescape这玩意, 我也是学着SnapKit照葫芦画瓢, 上次看完官方文档过段时间就直接忘记怎么用了, 全靠了SnapKit...

③. 验证

import UIKit
class ViewController: UIViewController {

    override func viewDidLoad() {
        super.viewDidLoad()
        
        let result = NSObject.sl_calculate { (make) in
            make.add(5).add(6).mul(4)
            
            do{
              try make.div(0)
            }catch{
                print("除数不能为0")
            }
        }
        print(result)
    }
}