如何利用Objective-C写一个精美的DSL-objective-c入门

推荐序：本文是来自美团的 iOS 技术专家臧成威的投稿。臧老师在 StuQ 开完 RactiveCocoa 的两次系列课程后，最近新开了一门《iOS 实战黑魔法》的新课程，课程内容涉及很多 Objective-C Runtime, Swift 等底层的知识和应用技巧，如果你感兴趣，可以看文末的介绍。

感谢臧成威的授权，以下是文章正文。

背景

在程序开发中，我们总是希望能够更加简洁、更加语义化地去表达自己的逻辑，链式调用是一种常见的处理方式。我们常用的 Masonry、 Expecta 等第三方库就采用了这种处理方式。

// Masonry 
[view1 mas_makeConstraints:^(MASConstraintMaker *make) { 
    make.top.equalTo(superview.mas_top).with.offset(padding.top); 
    make.left.equalTo(superview.mas_left).with.offset(padding.left); 
    make.bottom.equalTo(superview.mas_bottom).with.offset(-padding.bottom); 
    make.right.equalTo(superview.mas_right).with.offset(-padding.right); 
}];

// Expecta 
expect(@"foo").to.equal(@"foo"); // `to` is a syntactic sugar and can be safely omitted. 
expect(foo).notTo.equal(1); 
expect([bar isBar]).to.equal(YES); 
expect(baz).to.equal(3.14159);

像这种用于特定领域的表达方式，我们叫做 DSL (Domain Specific Language)，本文就介绍一下如何实现一个链式调用的 DSL.

链式调用的实现

我们举一个具体的例子，比如我们用链式表达式来创建一个 UIView，设置其 frame、backgroundColor，并添加至某个父 View。

对于最基本的 Objective-C (在 iOS4 block 出现之前)，如果要实现链式调用，只能是这个样子的：

UIView *aView = [[[[UIView alloc] initWithFrame:aFrame] bgColor:aColor] intoView:aSuperView];

有了 block，我们可以把中括号的这种写法改为点语法的形式

UIView *aView = AllocA(UIView).with.position(x, y).size(width, height).bgColor(aColor).intoView(aSuperView); 
 
// 当x和y为默认值0和0或者width和height为默认值0的时候，还可以省略 
UIView *bView = AllocA(UIView).with.size(width, height).bgColor(aColor).intoView(aSuperView);

可以看出，链式语法的语义性很明确，后者的语法更加紧凑，下面我们从两个角度看一下后者的实现。

1. 从语法层面来看

链式调用可以用两种方式来实现：

1).在返回值中使用属性来保存方法中的信息

比如，Masonry 中的 .left .right .top .bottom 等方法，调用时会返回一个 MASConstraintMaker 类的实例，里面有 left/right/top/bottom 等属性来保存每次调用时的信息;

make.left.equalTo(superview.mas_left).with.offset(15);

再比如，Expecta 中的方法 .notTo 方法会返回一个 EXPExpect 类的实例，里面有个 BOOL 属性 self.negative 来记录是否调用了 .notTo;

expect(foo).notTo.equal(1);

再比如，上例中的 .with 方法，我们可以直接 return self;

2).使用 block 类型的属性来接受参数

比如 Masonry 中的 .offset(15) 方法，接收一个 CGFloat 作为参数，可以在 MASConstraintMaker 类中添加一个 block 类型的属性：

@property (nonatomic, copy) MASConstraintMaker* (^offset)(CGFloat);

比如例子中的 .position(x, y)，可以给的某类中添加一个属性：

@property (nonatomic, copy) ViewMaker* (^position)(CGFloat x, CGFloat y);

在调用 .position(x, y) 方法时，执行这个block，返回 ViewMaker 的实例保证链式调用得以进行。

2. 从语义层面来看

从语义层面上，需要界定哪些是助词，哪些是需要接受参数的。为了保证链式调用能够完成，需要考虑传入什么，返回什么。

还是以上面的例子来讲：

UIView *aView = AllocA(UIView).with.position(x, y).size(width, height).bgColor(aColor).intoView(aSuperView);

分步来看一下，这个 DSL 表达式需要描述的是一个祈使句，以 Alloc 开始，以 intoView 截止。在 intoView 终结语之前，我们对 UIView 进行一定的修饰，利用 position size bgColor 这些。

下面我们分别从四段来看，如何实现这样一个表达式：

(1) 宾语

在 AllocA(UIView) 的语义中，我们确定了宾语是 a UIVIew。由于确定 UIView 是在 intoView 截止那时，所以我们需要创建一个中间类来保存所有的中间条件，这里我们用 ViewMaker 类。

@interface ViewMaker : NSObject 
@property (nonatomic, strong) Class viewClass; 
@property (nonatomic, assign) CGPoint position; 
@property (nonatomic, assign) CGPoint size; 
@property (nonatomic, strong) UIColor *color; 
@end

另外我们可以注意到AllocA是一个函数，而UIView无法直接传递到这个函数中，语法就要变成 AllocA([UIView class]) 而失去了简洁性。所以我们需要先定义一个宏来“吞”掉中括号和 class 这个方法：

#define AllocA(aClass)  alloc_a([aClass class]) 
 
ViewMaker* alloc_a(Class aClass){ 
    ViewMaker *maker = ViewMaker.new; 
    maker.viewClass = aClass; 
    return maker; 
}

(2) 助词

很多时候，为了让 DSL 的语法看起来更加连贯，我们需要一些助词来帮助，例如 Masonry 里面的 make.top.equalTo(superview.mas_top).with.offset(padding.top) 这句中的 with 就是这样一个助词。

而这个助词和我们学过的语法一样，通常没有什么实际效果，简单返回self就可以。

@interface ViewMaker : NSObject 
@property (nonatomic, strong) Class viewClass; 
@property (nonatomic, assign) CGPoint position; 
@property (nonatomic, assign) CGPoint size; 
@property (nonatomic, strong) UIColor *color; 
@property (nonatomic, readonly) ViewMaker *with; 
@end 
 
@implementation ViewMaker 
 
- (ViewMaker *)with 
{ 
    return self; 
} 
@end

需要注意的是，返回自己，就没有办法阻止用户不断调用自己 .with.with.with ，为了避免这种情况，可以新生成一个类，每个类都拥有自己所在层次的方法，避免跃层调用。

@interface ViewMaker : NSObject 
@property (nonatomic, strong) Class viewClass; 
@property (nonatomic, assign) CGPoint position; 
@property (nonatomic, assign) CGPoint size; 
@property (nonatomic, strong) UIColor *color; 
@end 
 
@interface ViewClassHelper : NSObject 
@property (nonatomic, strong) Class viewClass; 
@property (nonatomic, readonly) ViewMaker *with; 
@end 
 
#define AllocA(aClass)  alloc_a([aClass class]) 
 
ViewClassHelper* alloc_a(Class aClass){ 
    ViewClassHelper *helper = ViewClassHelper.new; 
    helper.viewClass = aClass; 
    return helper; 
} 
@implementation ViewClassHelper 
 
- (ViewMaker *)with 
{ 
    ViewMaker *maker = ViewMaker.new; 
    maker.viewClass = self.viewClass; 
    return maker; 
} 
@end

这样就有效防止了，.with.with.with这样的语法。但是实际上，我们要根据真实的需要来进行开发，使用 DSL 的用户是为了更好的表达性，所以并不会写出.with.with.with这样的代码，这样的防护性措施就显得有点不必要了。

不过使用类来区分助词还有另外几个小好处，就是它可以确保在语法提示的时候，ViewClassHelper这个类只有.with这样一个语法提示，而ViewMaker不出现.with语法提示;并且同时确保.with一定要出现。

不过为了简化文章，我们都使用前者，既.with返回self来继续下文：

@interface ViewMaker : NSObject 
@property (nonatomic, strong) Class viewClass; 
@property (nonatomic, assign) CGPoint position; 
@property (nonatomic, assign) CGPoint size; 
@property (nonatomic, strong) UIColor *color; 
@property (nonatomic, readonly) ViewMaker *with; 
@end 
 
@implementation ViewMaker 
 
- (ViewMaker *)with 
{ 
    return self; 
} 
@end

(3) 修饰部分——定语

像例子中的position size bgColor这些都是定语部分，用来修饰UIView，他们以属性的形势存在于ViewMaker的实例中，为了支持链式表达，所以实现的时候，都会继续返回self。

我们来试着实现下：

@interface ViewMaker : NSObject 
// ... 
@property (nonatomic, copy) ViewMaker* (^position)(CGFloat x, CGFloat y); 
@property (nonatomic, copy) ViewMaker* (^size)(CGFloat x, CGFloat y); 
@property (nonatomic, copy) ViewMaker* (^bgColor)(UIColor *color); 
@end 
 
@implementation ViewMaker 
 
- (instancetype)init 
{ 
    if (self = [super init]) { 
        @weakify(self) 
        _position = ^ViewMaker *(CGFloat x, CGFloat y) { 
            @strongify(self) 
            self.position = CGPointMake(x, y); 
        }; 
        _size = ^ViewMaker *(CGFloat x, CGFloat y) { 
            @strongify(self) 
            self.size = CGPointMake(x, y); 
        }; 
        _bgColor = ^ViewMaker *(UIColor *color) { 
            @strongify(self) 
            self.color = color; 
        }; 
    } 
    return self; 
} 
@end

(4) 终结词

“终结词”这个实在是在现代语法里面找不到对应关系了，但是在 DSL 中，这一段尤为重要。ViewMaker的实例从头至尾收集了很多的修饰，需要***的一个表达词语来产生***的结果，这里就称为”终结词”。例如在 Expecta 这个开源库里面的 equal 就是把真正的行为表现出来的时候，to 和 notTo 都不会真正触发行为。

在我们的例子里，终结词.intoView(aSuperViwe)可以这样实现：

@interface ViewMaker : NSObject 
// ... 
@property (nonatomic, copy) UIView* (^intoView)(UIView *superView); 
@end 
 
@implementation ViewMaker 
 
- (instancetype)init 
{ 
    if (self = [super init]) { 
        @weakify(self) 
        // ... 
        _intoView = ^UIView *(UIView *superView) { 
            @strongify(self) 
            CGRect rect = CGRectMake(self.position.x, self.position.y, 
                         self.size.width, self.size.height); 
            UIView *view = [[UIView alloc] initWithFrame:rect]; 
            view.backgroundColor = self.color; 
            [superView addSubView:view]; 
            return view; 
        }; 
    } 
    return self; 
} 
@end

这样，一个终结词就写好了。

最终代码的汇总：

@interface ViewMaker : NSObject 
@property (nonatomic, strong) Class viewClass; 
@property (nonatomic, assign) CGPoint position; 
@property (nonatomic, assign) CGPoint size; 
@property (nonatomic, strong) UIColor *color; 
@property (nonatomic, readonly) ViewMaker *with; 
@property (nonatomic, copy) ViewMaker* (^position)(CGFloat x, CGFloat y); 
@property (nonatomic, copy) ViewMaker* (^size)(CGFloat x, CGFloat y); 
@property (nonatomic, copy) ViewMaker* (^bgColor)(UIColor *color); 
@property (nonatomic, copy) UIView* (^intoView)(UIView *superView); 
@end 
 
@implementation ViewMaker 
 
- (instancetype)init 
{ 
    if (self = [super init]) { 
        @weakify(self) 
        _position = ^ViewMaker *(CGFloat x, CGFloat y) { 
            @strongify(self) 
            self.position = CGPointMake(x, y); 
        }; 
        _size = ^ViewMaker *(CGFloat x, CGFloat y) { 
            @strongify(self) 
            self.size = CGPointMake(x, y); 
        }; 
        _bgColor = ^ViewMaker *(UIColor *color) { 
            @strongify(self) 
            self.color = color; 
        }; 
        _intoView = ^UIView *(UIView *superView) { 
            @strongify(self) 
            CGRect rect = CGRectMake(self.position.x, self.position.y, 
                         self.size.width, self.size.height); 
            UIView *view = [[UIView alloc] initWithFrame:rect]; 
            view.backgroundColor = self.color; 
            [superView addSubView:view]; 
            return view; 
        }; 
    } 
    return self; 
} 
 
- (ViewMaker *)with 
{ 
    return self; 
} 
@end

总结

这种链式调用能够使程序更加清晰，在特定场景下使程序的可读性更强。这种手段在Swift也是相同道理，大家可以善加利用，让自己的代码更加美观。

其实，iOS 开发者要想不断精进，成长为真正的大牛高手，必须将自己的视野凌驾于业务需求之上，精简强化核心技能，提升自己对语言和工具的掌握层次，才能提高开发效率，提升技能水平。

这里为你准备了更多好玩的，让你事半功倍的 iOS 高阶黑魔法攻防术，斯达克学院(StuQ ) 特别邀请备受学员喜爱的资深 iOS 技术专家臧成威老师开设《 iOS 实战黑魔法》课程，6周12小时高效 Get iOS 必须掌握的高阶黑魔法攻防术，让你从普通的开发者中渐渐走出来，看到一个不一样的语言，感受不一样的开发!