目的
模仿 Masonry 连续运用点语法的操作
[self.view mas_makeConstraints:^(MASConstraintMaker *make) {
make.top.equalTo(@10).offset(1);
}];
写出一个连加的操作
make.add(10).add(10);
想看结果的请直接跳到“最终结果”
分析
一. 定义 SQMath 类
SQMath.h
#import <Foundation/Foundation.h>
@interface SQMath : NSObject
– (NSInteger)result;
– (void)add:(int)number;
@end
SQMath.m
#import “SQMath.h”
@interface SQMath ()
@property (nonatomic, assign) NSInteger number;
@end
@implementation SQMath
– (NSInteger)result {
return self.number;
}
– (void)add:(int)number {
self.number += number;
}
@end
使用这个 SQMath 的 add 方法
SQMath *math = [[SQMath alloc] init];
[math add:10];
[math add:20];
NSLog(@”%ld”, [math result]);
二. 将函数调用改为点语法
如果要用点语法，需要让 -add 从一个方法变成一个 add 的属性。但是这样就没有办法传参了
– (NSInteger)add;
但是如果返回值是一个 NSInteger (^)(NSInteger) 类型的 block 就可以了。math.add 返回的是这个 block，这个 block 是需要一个 NSInteger 为参数（加数），返回值是 NSInteger（结果）。
SQMath.m
– (NSInteger (^)(NSInteger count))add {
__weak typeof(self) weakSelf=self;
NSInteger (^addBlock)(NSInteger) = ^(NSInteger addCount){
__strong typeof(weakSelf) strongSelf = weakSelf;
strongSelf.number += addCount;
return strongSelf.number;
};
return addBlock;
}
或者
– (NSInteger (^)(NSInteger count))add {
__weak typeof(self) weakSelf=self;
return ^(NSInteger addCount) {
__strong typeof(weakSelf) strongSelf = weakSelf;
strongSelf.number += addCount;
return strongSelf.number;
};
}
使用这个 SQMath 的 add 方法
SQMath *math = [[SQMath alloc] init];
NSLog(@”%ld”, math.add(10));
NSLog(@”%ld”, math.add(20));
NSLog(@”%ld”, math.add(30));
三. 连续使用点语法
只要将 Block 的返回值更改为 self。这样每次 add 返回的则变成了 SQMath 的实例对象，这样就可以实现连续点语法的效果了。
– (SQMath* (^)(NSInteger count))add {
__weak typeof(self) weakSelf=self;
return ^(NSInteger addCount) {
__strong typeof(weakSelf) strongSelf = weakSelf;
strongSelf.number += addCount;
return self;
};
}
使用这个 SQMath 的 add 方法
SQMath *math = [[SQMath alloc] init];
NSLog(@”%ld”, math.add(10).add(20).add(30).result) ;
四. 将这个改为 NSNumber 的 Category
NSNumber+SQMath.h
#import <Foundation/Foundation.h>
#import “SQMath.h”
@interface NSNumber (Math)
– (NSInteger)sq_add:(void(^)(SQMath *make))block;
@end
NSNumber+SQMath.m
#import “NSNumber+SQMath.h”
@implementation NSNumber (SQMath)
– (NSInteger)sq_add:(void(^)(SQMath *))block {
SQMath *math = [[SQMath alloc] init];
block(math);
return math.result;
}
@end
NSNumber+SQMath 使用
NSInteger result = [@10 sq_add:^(SQMath * make) {
make.add(10).add(20);
}];
NSLog(@”%ld”, result);
最终结果
SQChainProgramming
ps: 链式编程什么时候用我还真不太清楚，但我知道面试的时候肯定有用 哈哈。