if - else 代码段的优化处理

概述

使用 if - else 的两种常见场景：

    // 1.非空异常判断
    id object;
    if (object == nil) {
        // code
    } else {
        // code
    }
    
    // 2.值状态判断
    if ([object isMemberOfClass:[NSObject class]]) {
        // code
    } else {
        // code
    }

对于第一种情况，可采取的优化方式有

• 合并条件表达式
• 减少嵌套
• 移除临时变量等
  原则是尽可能地维持正常流程代码在最外层，异常判断最先判断。

对于第二种情况

• 可以将每个状态对应的代码段封装成一个方法，这样代码可以更清晰
• 使用多态替代 if - else 代码段

值得注意的是，对于第一种使用 if-else 的情况，可以多多优化，但对于第二种情况，要慎重使用，不能为了优化而优化。

非空异常判断 if - else 代码段的优化

1.合并表达式，使代码更清晰易懂

    int number = 100;
    
    // 优化前
    if (number > 10) {
        return nil;
    }
    if (number < -10) {
        return nil;
    }
    if (number == 5) {
        return nil;
    }
    
    // 使用合并表达式条件优化后
    if (number > 10 || number < -10 || number == 5) {
        return nil;
    }

2.减少 if - else 嵌套，将包含关系改为平行关系，避免逻辑嵌套太深，使代码逻辑清晰明了，一目了然。

    int money = 6;
    
    // 优化前
    if (money < 0) {
        return @"食不果腹，衣不蔽体";
    } else if (money > 100) {
        return @"财务自由";
    } else if (money > 10) {
        return @"小康家庭";
    } else if (money > 5) {
        return @"温饱不愁";
    } else {
        return @"贫困家庭";
    }
    
    // 优化后
    if (money < 0) {
        return @"食不果腹，衣不蔽体";
    }
    if (money > 100) {
        return @"财务自由";
    }
    if (money > 10) {
        return @"小康家庭";
    }
    if (money > 5) {
        return @"温饱不愁";
    }
    return @"贫困家庭";

3.将异常判断放在最前面，且尽量减少临时变量改用return直接返回，使代码逻辑清晰。

     int number = 10;
    BOOL flag = NO;
    BOOL otherFlag = NO;
    
    // 优化前
    if (number > 10) {
        return nil;
    }

    if (otherFlag) {
        int result = 0;
        if (flag) {
            result = number * 2;
        } else {
            result = number *-2;
        }
        return result;
    }
    return nil;
    
    // 将上面的异常判断放在最前面，且较少临时变量的使用, 再合并表达式且将判断改为平行关系。异常条件先退出这一条可适用于请求网络数据后，对网络数据进行判断格式检查时，先将异常情况的处理放在最前面。
    if (number > 10 && !otherFlag) {
        return nil;
    }
    if (flag) {
        return number * 2;
    }
    return number *-2;

值状态 if-else 的优化

1.将每种状态下的长段代码提取成一个函数

    int number = 10;

    // 优化前
    if (number > 10) {
        // code balabala...
    } else {
        // code balabala...
    }
    
    // 优化前
    if (number > 10) {
        [self funationA];
    } else {
        [self funationB];
    }

2.使用多态替代 if-else

以上阐述其他的优化方式只能优化 if-else 代码段结构，但使用多态可以替代 if-else 。
这里使用多态 + map + 反射具体实现，当然还有其他几种实现方式,比如策略模式 + 工厂模式，不过我认为这种实现方式过于庞大。
举例来说，现在有个业务需求：根据汽车品牌，获取汽车配置。

使用 if-else 实现需求

    NSString *carBrand = @"BMW";
    
    if ([carBrand isEqualToString:@"BMW"]) {
        // BMW Configuration
    } else if ([carBrand isEqualToString:@"Benz"]) {
        // Benz Configuration
    }

使用多态 + map + 反射实现

//1. 声明汽车抽象类
@interface Car : NSObject
// 这个方法也可以提取到一个protocol中
- (NSString *)configuration;
@end

//2. 声明具体的汽车子类，并实现其相应方法
//2.1 声明 BenzCar 类
@interface BenzCar : Car

@end

@implementation BenzCar
- (NSString *)configuration {
    return @"BenzCar configuration";
}
@end

//2.2 声明 BMWCar 类
@interface BMWCar : Car
@end

@implementation BMWCar
- (NSString *)configuration {
    return @"BMWCar configuration";
}
@end

//3. 实现使用map + 反射机制的依托类

// 3.1 类的声明
@interface CarConfigurationHelper : NSObject
+ (NSString *)carConfigurationWithCarBrand:(NSString *)brand;
@end

// 3.2 类的实现
@implementation CarConfigurationHelper

+ (NSString *)carConfigurationWithCarBrand:(NSString *)brand {
    static NSDictionary *map = nil;
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        map = @{
                @"Benz" : [BenzCar class],
                @"BMW"  : [BMWCar class]
                };
    });
    if (!map[brand]) {
        return nil;
    }
    Car *car = [[map[brand] alloc] init];
    return [car configuration];
}

@end

//4.客户端的使用
- (void)function {
    NSString *configuration =  [CarConfigurationHelper carConfigurationWithCarBrand:@"BMW"];
    NSLog(@"Car configuration: %@",configuration);
}

参考：《重构_改善既有代码的设计》