背景
日常开发中一定会遇到这种场景，在某个类中提供了这样一个方法：
@interface TTDoSomething:NSObject
+ (void)doSomethingWithName:(NSString *)name configA:(ConfigA)configA configB:(ConfigB)configB configC:(ConfigC)configC;
@end
这个方法通过四个参数name, configA, configB, configB提供了 doSomeThing的功能，可是这个方法的参数实在是有点多，有的时候我只需要一个默认 config 就好了，并不是每个 config 都需要传入一个值，这时候可以把这个多参数的方法作为全能方法，然后以它为基础实现几个便利方法：
@interface TTDoSomething:NSObject
+ (void)doSomethingWithName:(NSString *)name configA:(ConfigA)configA;
+ (void)doSomethingWithName:(NSString *)name configB:(ConfigB)configB;
+ (void)doSomethingWithName:(NSString *)name configC:(ConfigC)configC;
+ (void)doSomethingWithName:(NSString *)name configA:(ConfigA)configA configB:(ConfigB)configB configC:(ConfigC)configC;
@end
@implementation TTDoSomething
+ (void)doSomethingWithName:(NSString *)name configA:(ConfigA)configA {
[self doSomethingWithName:name configA:configA configB:nil configC:nil];
}
+ (void)doSomethingWithName:(NSString *)name configB:(ConfigB)configB {
[self doSomethingWithName:name configA:nil configB:configB configC:nil];
}
...
@end
但是这么做的缺点也显而易见：
- 组合不灵活，如果需要增加 2 个种 config 组合的便利方法，需要增加新方法；
- 扩展不便，如果全能方法增加了新的配置 configD 需要在现有便利方法的基础上增加更多的便利方法；
链式表达式
今天突然灵光一闪，想到可以用链式表达式解决这个问题。在 iOS 开发领域因为 Masonry 这个库的影响，链式表达式知名度已经很高，因此这篇文章不再赘述其原理。直接进入主题：
首先声明一个链式表达式的链扣类：
.h
@interface TTDoSomethingMaker:NSObject //这里命名也参考 Masonry, 定义为 Maker
// 声明需要的 block 属性，包括 Name 和 Config
@property (nonatomic, copy, readonly) TTDoSomethingMaker *(^name)(NSString *name);
@property (nonatomic, copy, readonly) TTDoSomethingMaker *(^configA)(ConfigA configA);
@property (nonatomic, copy, readonly) TTDoSomethingMaker *(^configB)(ConfigB configB);
@property (nonatomic, copy, readonly) TTDoSomethingMaker *(^configC)(ConfigC configC);
@end
.m
@interface TTDoSomethingMaker ()
// 在.m 中声明保存 name 和 config 的属性，这里不需要暴露给外部，所以不需要声明在 .h
@property(nonatomic, copy) NSString *nameValue;
@property(nonatomic, assign) ConfigA configAValue;
@property(nonatomic, assign) ConfigB configBValue;
@property(nonatomic, assign) ConfigC configCValue;
@end
@implementation TTDoSomethingMaker
// 实现所有的 block getter
- (NSString *)name {
return ^TTDoSomethingMaker *(NSString *name) {
self.nameValue = name;
return self;
};
}
- (ConfigA)configA {
return ^TTDoSomethingMaker *(ConfigA configA) {
self.configAValue = configA;
return self;
};
}
- (ConfigB)configB {
return ^TTDoSomethingMaker *(ConfigB configB) {
self.configBValue = configB;
return self;
};
}
- (ConfigC)configC {
return ^TTDoSomethingMaker *(ConfigC configC) {
self.configCValue = configC;
return self;
};
}
@end
然后在提供功能的 TTDoSomething 类中增加一个方法：
+ (void)doSomething:(void(^)(TTDoSomethingMaker *maker))block {
TTDoSomethingMaker *maker = TTDoSomethingMaker.new;
block ?: block(maker);
// todo
[self doSomethingWithName:maker.nameValue configA:maker.configAValue configB:configBValue configC:configCValue]
}
于是当我们这样调用：
[TTDoSomething doSomething:^(TTDoSomethingMaker *maker) {
maker.name(@"Demo").configA(ConfigAOne);
}];
就相当于调用了
[TTDoSomething doSomethingWithName:@"Demo" configA:ConfigAOne configB:0 configC:0];
如果此时想要将 ConfigB 配置为 ConfigBOne, 只需要在原有的链式表达式后面加一个 .configB(ConfigBOne) 即可。
而如果想给万能方法增加一个 ConfigD, 也只需要修改一下 doSomething: 和 TTDoSomethingMaker 的实现，完美地遵循了开闭原则。
小优化一发
上面 doSomething: 的实现里我加了一个 // todo 的注释，在这个位置可以做一些默认实现，比如说合法性判断和默认值：
+ (void)doSomething:(void(^)(TTDoSomethingMaker *maker))block {
TTDoSomethingMaker *maker = TTDoSomethingMaker.new;
!block ?: block(maker);
NSAssert(maker.nameValue, @"Parameter 'name' should not be nil.");
// 将 ConfigB 配置为 ConfigBTwo
if (maker.configBValue == 0) {
maker.configBValue = ConfigBTwo;
}
[self doSomethingWithName:maker.nameValue configA:maker.configAValue configB:configBValue configC:configCValue]
}
另外，上面的 block getter 可以封装一个宏，减少重复性的代码：
#define TTDoSomethingMakerProperty(TYPE, NAME) - (TTDoSomethingMaker *(^)(TYPE))NAME {\
return ^TTDoSomethingMaker *(TYPE NAME) {\
self.NAME##Value = NAME;\
return self;\
};\
}\
@implementation TTDoSomethingMaker
TTDoSomethingMakerProperty(NSString *, name)
TTDoSomethingMakerProperty(ConfigA, configA)
TTDoSomethingMakerProperty(ConfigB, configB)
TTDoSomethingMakerProperty(ConfigC, configC)
@end
总结
Masonry 利用链式表达式将繁琐的 AutoLayout API 优化成了更表意、灵活、简洁的语法。而链式表达式可以随意自由组合的特性，恰好也是 Objective-C 不支持方法默认参数的优雅解决方式，虽然和 Masonry 的使用场景不一样，但是仔细想想，这两个场景都是对开闭原则的实现。
