Советы по программированию: Рефакторинг if...else if...else...

JavaScript
Советы по программированию: Рефакторинг if...else if...else...

предисловие

В новой компании предшественники оставили проект, наполненный множеством if...else..., но он второй, в основном с небольшим количеством комментариев. Ввиду такого кода, который уже был запущен, я не хотел его рефакторить, потому что стоимость была слишком высока, поэтому мне пришлось уговаривать себя не писать такой код.

Столкнулись с проблемой?

Иногда мы можем столкнуться с такой бизнес-логикой (образовательные компании), если вы ответили на вопрос и прошли, если вы ответили на вопрос и не прошли, если вы не ответили на вопрос. Без использования определенного шаблона вы можете написать код, подобный следующему. Куча if...else выглядит очень неудобно и сложно в обслуживании.

/**
 * 初始化函数
 * if...else if... 的情况较为简单
 * @return undefined
 */
function init () {
    // 是否回答过题目 1-回答过, 通过 2-回答过, 没有通过 3-没有回答过
    let isAnswer
    // 是否是老用户 1-老用户 2-新用户
    let isOldUser

    if (isAnswer === 1) {
        // ...
    } else if (isAnswer === 2) {
        // ...
    } else if (isAnswer === 3) {
        // ...
    }

    if (isOldUser === 1) {
        // ...
    } else if (isOldUser === 2) {
        // ...
    }
}
/**
 * 初始化函数
 * if...else if... 嵌套的情况
 * @return undefined
 */
function init () {
    if (isAnswer === 1) {
        if (isOldUser === 1) {
            // ...
        } else if (isOldUser === 2) {
            // ...
        }
    } else if (isAnswer === 2) {
        if (isOldUser === 1) {
            // ...
        } else if (isOldUser === 2) {
            // ...
        }
    } else if (isAnswer === 3) {
        if (isOldUser === 1) {
            // ...
        } else if (isOldUser === 2) {
            // ...
        }
    }
}

Решение 1: справочная таблица, справочная таблица цепочки обязанностей

Хотя может показаться, что это лекарство от симптомов, а не от основной причины, это не так, сложность функции инициализации значительно снижается. Мы разделили сложную логику потока управления на функцию определения действия.

// 可以解决if...else if...简单的问题
const rules = {
    isAnswer1 () {
        return code
    },
    isAnswer2 () {
        return code
    },
    isAnswer3 () {
        return code
    }
}

function determineAction (type) {
    if (isAnswer === 1) {
        return 'isAnswer1'
    } else if (isAnswer === 2) {
        return 'isAnswer2'
    } else if (isAnswer === 3) {
        return 'isAnswer3'
    }
}

function init () {
    let key = determineAction(isAnswer)
    return rules[key]
}
// 面对if...else if...else 嵌套的复杂情况

const rules = [
    {
        match (an, old) {
            if (an === 1) {
                return true
            }
        },

        action (an, old) {
            if (old === 1) {
                // ...
            } else if (old === 2) {
                // ...
            }
        }
    },
    {
        match (an, old) {
            if (an === 2) {
                return true
            }
        },

        action (an, old) {
            if (old === 1) {
                // ...
            } else if (old === 2) {
                // ...
            }
        }
    },
    {
        match (an, old) {
            if (an === 3) {
                return true
            }
        },

        action (an, old) {
            if (old === 1) {
                // ...
            } else if (old === 2) {
                // ...
            }
        }
    }
]

function init (an, old) {
    for (let i = 0; i < rules.length; i++) {
        // 如果返回true
        if (rules[i].match(an, old)) {
            rules[i].action(an, old)
        }
    }
}

init(isAnswer, isOldUser)

⬆️В приведенной выше сложной ситуации вы также можете извлечь суждение о действии, но вам, возможно, придется написать три извлеченные функции, потому что есть три разных ситуации для значения


Решение 2. Аспектно-ориентированное программирование (АОП)

Для цепочки прототипов функции расширьте синтаксис after, если требования выполнены, функция выполняется напрямую и возвращается результат. Если условие не выполнено, верните «следующий», чтобы вызвать следующий узел в цепочке ответственности. Так называемый Function.prototype.after предназначен для выполнения функции, добавленной после выполнения этой функции.

// 可以解决if...else if...简单的问题
Function.prototype.after = function (nextFn) {
    let self = this
    return function (...rest) {
        let code = self(...rest)
        if (code === 'next') {
            return nextFn(...rest)
        }
        return code
    }
}

// 重构原函数

function isAnswer1 (type) {
    if (type === 1) {
        return code
    }
    return 'next'
}

function isAnswer2 () {
    if (type === 2) {
        return code
    }
    return 'next'
}

function isAnswer3 () {
    if (type === 3) {
        return code
    }
    return 'next'
}

let isAnswerFn = isAnswer1.after(isAnswer2).after(isAnswer3)

isAnswerFn(isAnswer)
// 面对if...else if...else 嵌套的复杂情况

function isAnswer1 (an, old) {
    if (an === 1) {
        return isOldUserFn1(an, old)
    }
    return 'next'
}

function isAnswer2 (an, old) {
    if (an === 2) {
        return isOldUserFn2(an, old)
    }
    return 'next'
}

function isAnswer3 (an, old) {
    if (an === 3) {
        return isOldUserFn3(an, old)
    }
    return 'next'
}

/**
 * isAnswer == 1 isOldUser == 1 的情况
 */
function isAnswer1IsOldUser1 (an, old) {
    if (old === 1) {
        return code
    }
    return 'next'
}

/**
 * isAnswer == 1 isOldUser == 2 的情况
 */
function isAnswer1IsOldUser2 (an, old) {
    if (old === 2) {
        return code
    }
    return 'next'
}

/**
 * isAnswer == 2 isOldUser == 1 的情况
 */
function isAnswer2IsOldUser1 (an, old) {
    if (old === 1) {
        return code
    }
    return 'next'
}

/**
 * isAnswer == 2 isOldUser == 2 的情况
 */
function isAnswer2IsOldUser2 (an, old) {
    if (old === 2) {
        return code
    }
    return 'next'
}

/**
 * isAnswer == 3 isOldUser == 1 的情况
 */
function isAnswer3IsOldUser1 (an, old) {
    if (old === 1) {
        return code
    }
    return 'next'
}

/**
 * isAnswer == 3 isOldUser == 2 的情况
 */
function isAnswer3IsOldUser2 (an, old) {
    if (old === 2) {
        return code
    }
    return 'next'
}

let isAnswerFn = isAnswer1.after(isAnswer2).after(isAnswer3)

// 三条职责链
let isOldUserFn1 = isAnswer1IsOldUser1.after(isAnswer1IsOldUser2)
let isOldUserFn2 = isAnswer2IsOldUser1.after(isAnswer2IsOldUser2)
let isOldUserFn3 = isAnswer3IsOldUser1.after(isAnswer3IsOldUser2)

isAnswerFn(isAnswer, isOldUser)

Решение 3. Функциональное программирование

Используйте библиотеки функционального программирования, такие как ramda, для решения такого рода проблем, 🔗 ссылка:ramda.cn/docs/#cond

import R from 'ramda'

var fn = R.cond([
  [R.equals(0),   R.always('water freezes at 0°C')],
  [R.equals(100), R.always('water boils at 100°C')],
  [R.T,           temp => 'nothing special happens at ' + temp + '°C']
]);

fn(0); //=> 'water freezes at 0°C'
fn(50); //=> 'nothing special happens at 50°C'
fn(100); //=> 'water boils at 100°C'