不到30行代码，有5层嵌套的if语句？避免陷入“if-else地狱”！

IT界那些事儿

于 2024-09-30 16:30:05 发布

阅读量25

点赞数

文章标签： java 前端 javascript

原文链接：https://mp.weixin.qq.com/s/PVX3pS3L8DSk-GmYE7AwqQ

版权

多年前，我参与了一个至今仍被广泛使用的打车移动应用的开发。我不知道现在他们运行的代码是什么情况，但如果我没记错的话，早期关于司机分配这部分的代码，很大程度上类似于下面这个极度简化的示例：

async function assignDriver(rider, availableDrivers) {
    const driverDistances = await calculateDistances(rider.location, availableDrivers);
    let assignedDriver = null;

    for (let driver of availableDrivers) {
        if (driverDistances[driver.id] <= 5) {
            if (!rider.preferredVehicle || rider.preferredVehicle === driver.vehicle) {
                if (driver.rating >= 4.5) {
                    if (rider.preferences.includes('Premium Driver')) {
                        if (driver.isPremiumDriver) {
                            assignedDriver = driver;
                            break;
                        } else {
                            continue;
                        }
                    } else {
                        assignedDriver = driver;
                        break;
                    }
                } else if (driver.rating >= 4.0) {
                    assignedDriver = driver;
                    break;
                }
            }
        }
    }

    return assignedDriver;
}

在这段不到 30 行的代码中，有五层嵌套的 if 语句。也许有人会说，这看起来还不算太糟，但不难想象，如果再加上诸如高峰期加价、忠诚度计划等更多检查条件，这些代码会变得多么复杂。

幸运的是，我们有办法将代码扁平化处理，而最终的结果可能会让你大吃一惊：当我们完成重构后，就不会再有 if 语句了。

利用保护子句

让我们先从简单的开始。第一个 if 语句用于检查司机和乘客之间允许的最大距离，由于它适用于所有情况，因此可以将其转换为保护子句，从而移除一层嵌套。同样，我们还可以为首选车辆检查添加另一个保护子句，以去除另一层嵌套。由此修改后的代码如下所示：

async function assignDriver(rider, availableDrivers) {
    const driverDistances = await calculateDistances(rider.location, availableDrivers);
    let assignedDriver = null;

    for (let driver of availableDrivers) {
    if (driverDistances[driver.id] > 5) {
      continue;
    }
    if (rider.preferredVehicle && rider.preferredVehicle !== driver.vehicle) {
        continue;
    }
    if (driver.rating >= 4.5) {
            if (rider.preferences.includes('Premium Driver')) {
                if (driver.isPremiumDriver) {
                assignedDriver = driver;
                break;
            } else {
                continue;
            }
        } else {
            assignedDriver = driver;
            break;
        }
    } else if (driver.rating >= 4.0) {
        assignedDriver = driver;
        break;
    }
    }

    return assignedDriver;
}

决策表

与其在函数内部硬编码逻辑，我们可以将每个 if-else 模块放在它自己的函数中，并将这些函数放入一个数组，形成一个决策表。然后，我们遍历决策表中的每个函数，直到得到一个肯定的响应。当然了，表中的条目必须从最具体的到最不具体的顺序排列，这样才能在我们的案例中正常运行。

const conditions = [
    (rider, driver) => driver.rating >= 4.5 && rider.preferences.includes('Premium Driver') && driver.isPremiumDriver,
    (rider, driver) => driver.rating >= 4.5 && !rider.preferences.includes('Premium Driver'),
    (rider, driver) => driver.rating >= 4.0 && driver.rating < 4.5,
];

async function assignDriver(rider, availableDrivers, conditions) {
    const driverDistances = await calculateDistances(rider.location, availableDrivers);
    let assignedDriver = null;

    for (let driver of availableDrivers) {
        if (driverDistances[driver.id] > 5) {
            continue;
        }
        if (rider.preferredVehicle && rider.preferredVehicle !== driver.vehicle) {
            continue;
        }
        if (conditions.find((condition) => condition(rider, driver))) {
            assignedDriver = driver;
            break;
        }
    }

    return assignedDriver;
}

有了决策表，我们就彻底消除了嵌套的 if 语句，还有一个额外的好处：现在只需简单编辑条件数组，就可以更改驱动程序的赋值逻辑。