重構的藝術:五個小妙招助你寫出好代碼!
全文共8912字,預計學習時長14分鐘

糟糕的代碼可以運作,但早晚會讓我們付出代價。你有沒有遇到過這樣的問題:幾周后,你無法理解自己的代碼,于是不得不花上幾個小時,甚至幾天的時間來弄清楚到底發(fā)生了什么。
解決這個常見問題的方法是使代碼盡可能清晰。如果做得更好的話,即使是非技術人員也應該能理解你的代碼。
是時候停止尋找借口,提高我們的代碼質(zhì)量了!
編寫清晰的代碼并沒有那么復雜。本教程將向你展示五種改進代碼的簡單技巧,并提供一些實例:
1. 不用switch語句
我們通常使用switch語句來代替大型if-else-if語句。但是,switch語句非常冗長,很難維護,甚至很難調(diào)試。這些switch語句把我們的代碼弄得亂七八糟,而且這些語句的語法很奇怪,很不舒服。在添加更多的case時,我們不得不必須手動添加每個case和break語句,而這就很容易出錯。
接下來看一個switch語句的例子:
function getPokemon(type) { let pokemon; switch (type) { case 'Water': pokemon = 'Squirtle'; break; case 'Fire': pokemon = 'Charmander'; break; case 'Plant': pokemon = 'Bulbasur'; break; case 'Electric': pokemon = 'Pikachu'; break; default: pokemon = 'Mew'; } return pokemon; }
console.log(getPokemon('Fire')); // Result: Charmander
如果需要在switch語句中添加更多的case的話,需要編寫的代碼量是相當大的。我們可能最終會復制粘貼代碼,而其實我們都知道這種行為的后果是什么。
那么,如何避免使用switch語句呢?可以通過使用對象文本。對象文本簡單,易于編寫,方便讀取,維護輕松。我們都習慣用javascript處理對象,對象文本語法比switch語句更新鮮。下面舉個例子:
const pokemon = { Water: 'Squirtle', Fire: 'Charmander', Plant: 'Bulbasur', Electric: 'Pikachu' };
function getPokemon(type) { return pokemon[type] || 'Mew'; } console.log(getPokemon('Fire')); // Result: Charmander
// If the type isn't found in the pokemon object, the function will return the default value 'Mew' console.log(getPokemon('unknown')); // Result: Mew
如你所見,可以使用運算符 || 添加默認值。如果在pokemon對象中找不到type,getpokemon函數(shù)將使mew返回為默認值。
小貼士:你可能已經(jīng)注意到,我們在函數(shù)外部而不是內(nèi)部聲明pokemon對象。這樣做是為了避免每次執(zhí)行函數(shù)時都重新創(chuàng)建pokemon。
用映射也能達到同樣的效果。映射就像對象一樣是鍵-值對的集合。不同的是映射允許任何類型的鍵,而對象只允許字符串作為鍵。此外,映射還有一系列有趣的屬性和方法。
以下是使用映射的方法:
const pokemon = new Map([ ['Water', 'Squirtle'], ['Fire', 'Charmander'], ['Plant', 'Bulbasur'], ['Electric', 'Pikachu'] ]);
function getPokemon(type) { return pokemon.get(type) || 'Mew'; }
console.log(getPokemon('Fire')); // Result: Charmander console.log(getPokemon('unknown')); // Result: Mew
如你所見,當用對象文本或映射替換switch語句時,代碼看起來更清楚、更直接。
2. 把條件語句寫的更有描述性
在編寫代碼時,條件語句是絕對必要的。然而,他們很快就會失控,最終讓我們無法理解這些語句。這導致我們要么必須編寫注釋來解釋語句的作用,要么必須花費寶貴的時間來一條一條檢查代碼來了解發(fā)生了什么。這很糟糕。
看一下下面的語句:
function checkGameStatus() { if ( remaining === 0 || (remaining === 1 && remainingPlayers === 1) || remainingPlayers === 0 ) { quitGame(); } }
如果只看前面函數(shù)里if語句中的代碼,很難理解發(fā)生了什么。代碼表意不清楚,不清楚的代碼只會導致技術錯誤,還會讓人們非常頭痛。
怎樣改善條件語句呢?可以把它寫成一個函數(shù)。以下是具體操作方法:
function isGameLost() { return ( remaining === 0 || (remaining === 1 && remainingPlayers === 1) || remainingPlayers === 0 ); }
// Our function is now much easier to understand: function checkGameStatus() { if (isGameLost()) { quitGame(); } }
通過將條件提取到具有描述性名稱isGameLost()的函數(shù)中,checkGameStatus函數(shù)現(xiàn)在就變得很容易理解。為什么?因為代碼表意更清晰。它能夠告訴我們發(fā)生了什么,這是我們應該一直努力的方向。

3. 用衛(wèi)語句(Guard Clauses)代替嵌套的if語句
嵌套if語句是在代碼中可能遇到的最可怕的事情之一。你要是想能夠完全掌握代碼的情況,這絕對會讓你精疲力竭。下面是一個嵌套if語句的例子(這個嵌套有三層):
function writeTweet() { const tweet = writeSomething();
if (isLoggedIn()) { if (tweet) { if (isTweetDoubleChecked()) { tweetIt(); } else { throw new Error('Dont publish without double checking your tweet'); } } else { throw new Error("Your tweet is empty, can't publish it"); } } else { throw new Error('You need to log in before tweeting'); } }
嵌套的if語句
你可能需要花幾分鐘時間上下閱讀,才能了解函數(shù)運作的流程。嵌套的if語句很難讓人閱讀和理解。那么,如何才能擺脫討厭的嵌套if語句呢?可以反向思考,使用衛(wèi)語句來替代這些句子。
“在計算機程序設計中,衛(wèi)語句是一個布爾表達式,如果程序要在有問題的分支里繼續(xù)運行的話,它的求值必須為true。”——維基百科
通過顛倒函數(shù)的邏輯,并在函數(shù)開始時放置導致早期退出的條件,它們將變?yōu)楸Wo程序,并且只允許函數(shù)在滿足所有條件時繼續(xù)執(zhí)行。這樣可以避免else語句。下面是如何重構之前的函數(shù)以使用衛(wèi)語句的方法:
function writeTweet() { const tweet = writeSomething();
if (!isLoggedIn()) { throw new Error('You need to log in before tweeting'); } if (!tweet) { throw new Error("Your tweet is empty, can't publish it"); } if (!isTweetDoubleChecked()) { throw new Error('Dont publish without double checking your tweet'); }
tweetIt(); }
如你所見,代碼更清晰,更容易理解。我們可以簡單向下閱讀來了解函數(shù)的作用,遵循函數(shù)的自然流動,不像以前那樣上下閱讀。
4. 不要寫重復的代碼
寫重復的代碼總是以失敗告終。它會導致如下情況:“我在這里修復了這個bug,但是忘記在那里修復”或“我需要在五個不同的地方更改/添加一個新功能”。
正如DRY(Don’t repeat yourself不要重復)原則所說:
每一部分知識或邏輯都必須在一個系統(tǒng)中有單一的、明確的表示。
因此,代碼越少越好:它節(jié)省了時間和精力,更易于維護,并減少了錯誤的出現(xiàn)。
那么,如何避免重復代碼呢?這有點難,但是將邏輯提取到函數(shù)/變量通常效果很好。讓我們看看下面的代碼,我在重構應用程序時看到了這些代碼:
function getJavascriptNews() { const allNews = getNewsFromWeb(); const news = []; for (let i = allNews.length - 1; i >= 0; i--){ if (allNews[i].type === "javascript") { news.push(allNews[i]); } } return news; } function getRustNews() { const allNews = getNewsFromWeb(); const news = []; for (let i = allNews.length - 1; i >= 0; i--){ if (allNews[i].type === "rust") { news.push(allNews[i]); } } return news; }
function getGolangNews() { const news = []; const allNews = getNewsFromWeb();
for (let i = allNews.length - 1; i >= 0; i--) { if (allNews[i].type === 'golang') { news.push(allNews[i]); } }
return news; }
你可能已經(jīng)注意到for循環(huán)在這兩個函數(shù)中完全相同,除了一個小細節(jié):我們想要的新聞類型,即javascript或rust新聞。為了避免這種重復,可以將for循環(huán)提取到一個函數(shù)中,然后從getJavascriptNews,getRustNews和getGolangNews 函數(shù)調(diào)用該函數(shù)。以下是具體操作方法:
function getJavascriptNews() { const allNews = getNewsFromWeb(); return getNewsContent(allNews, 'javascript'); }
function getRustNews() { const allNews = getNewsFromWeb(); return getNewsContent(allNews, 'rust'); }
function getGolangNews() { const allNews = getNewsFromWeb(); return getNewsContent(allNews, 'golang'); }
function getNewsContent(newsList, type) { const news = []; for (let i = newsList.length - 1; i >= 0; i--) { if (newsList[i].type === type) { news.push(newsList[i].content); } } return news; }
在將for循環(huán)提取到getNewsContent函數(shù)中之后,getJavaScriptNews, getRustNews和getGolangNews函數(shù)變成了簡單、清晰的程序。
進一步重構
但是,你是否意識到,除了傳遞給getNewsContent的類型字符串之外,這兩個函數(shù)完全相同?這是重構代碼時經(jīng)常發(fā)生的事情。通常情況下,更改一個會導致另一個更改,以此類推,直到重構后的代碼最終變成原始代碼的一半大小。代碼告訴你它需要什么:
function getNews(type) { const allNews = getNewsFromWeb(); return getNewsContent(allNews, type); }
function getNewsContent(newsList, type) { const news = []; for (let i = newsList.length - 1; i >= 0; i--) { if (newsList[i].type === type) { news.push(newsList[i].content); } } return news; }
補充:使用es6特性使for循環(huán)更具可讀性
for循環(huán)并不完全可讀。通過引入es6數(shù)組函數(shù),可以有95%的機會避免使用它們。在本例中可以使用array.filter和array.map組合來替換原始循環(huán):
function getNews(type) { const allNews = getNewsFromWeb(); return getNewsContent(allNews, type); }
function getNewsContent(newsList, type) { return newsList .filter(newsItem => newsItem.type === type) .map(newsItem => newsItem.content); }
? 用Array.filter只返回其類型等于作為參數(shù)傳遞的類型的元素。
? 用Array.map只返回item對象的content屬性,而不是整個item。
恭喜你,經(jīng)過三次簡單的重構,最初的三個函數(shù)已經(jīng)縮減為兩個,這更容易理解和維護。另外,抽象讓getNews函數(shù)變得可以重新利用。

5. 一個函數(shù)只用來做一件事
一個函數(shù)應該只做一件事。做不止一件事的函數(shù)是所有罪惡的根源,也是代碼中可能遇到的最糟糕的事情之一(嵌套的if語句也是)。它們很混亂,搞得代碼難以理解。下面是一個來自實際應用程序的復雜函數(shù)示例:
function startProgram() { if (!window.indexedDB) { window.alert("Browser not support indexeDB"); } else { let openRequest = indexedDB.open("store", 1); openRequest.onupgradeneeded = () => {}; openRequest.onerror = () => { console.error("Error", openRequest.error); }; openRequest.onsuccess = () => { let db = openRequest.result; }; document.getElementById('stat-op').addEventListener('click', () => {}); document.getElementById('pre2456').addEventListener('click', () => {}); document.getElementById('cpTagList100').addEventListener('change', () => {}); document.getElementById('cpTagList101').addEventListener('click', () => {}); document.getElementById('gototop').addEventListener('click', () => {}); document.getElementById('asp10').addEventListener('click', () => {}); fetch("employeList.json") .then(res => res.json()) .then(employes => { document.getElementById("employesSelect").innerHTML = employes.reduce( (content, employe) => employe.name + "<br>", "" ); }); document.getElementById("usernameLoged").innerHTML = `Welcome, ${username}`; } }
小貼士:由于本例不需要事件偵聽器的處理程序,所以刪除了它們。
如你所見,這讓人困惑,也很難理解里面發(fā)生了什么。如果有錯誤出現(xiàn),都很難找到并修復它們。如何改進startProgram函數(shù)?可以將公共邏輯提取到函數(shù)中。以下是具體操作方法:
function startProgram() { if (!window.indexedDB) { throw new Error("Browser doesn't support indexedDB"); }
initDatabase(); setListeners(); printEmployeeList(); }
function initDatabase() { let openRequest = indexedDB.open('store', 1);
openRequest.onerror = () => { console.error('Error', openRequest.error); }; openRequest.onsuccess = () => { let db = openRequest.result; }; }
function setListeners() { document.getElementById('stat-op').addEventListener('click', () => {}); document.getElementById('pre2456').addEventListener('click', () => {}); document.getElementById('cpTagList100').addEventListener('change', () => {}); document.getElementById('cpTagList101').addEventListener('click', () => {}); document.getElementById('gototop').addEventListener('click', () => {}); document.getElementById('asp10').addEventListener('click', () => {}); }
async function printEmployeeList() { const employees = await getEmployeeList();
document.getElementById('employeeSelect').innerHTML = formatEmployeeList(employees); }
function formatEmployeeList(employees) { return employees.reduce( (content, employee) => content + employee.name + '<br>', '' ); }
function getEmployeeList() { return fetch('employeeList.json').then(res => res.json()); }
仔細看看startProgram函數(shù)的變化:
首先,通過使用一個衛(wèi)語句替換掉if-else語句。然后,啟動數(shù)據(jù)庫所需的邏輯提取到initDatabase函數(shù)中,并將事件偵聽器添加到setListeners函數(shù)中。
打印員工列表的邏輯稍微復雜一些,因此創(chuàng)建了三個函數(shù):printEmployeeList, formatEmployeeList和getEmployeeList。
getEmployeeList負責向employeeList.json發(fā)出GET請求并以json格式返回響應。
然后由printEmployeeList函數(shù)調(diào)用getEmployeeList,該函數(shù)獲取員工列表并將其傳遞給formatEmployeeList函數(shù),formatEmployeeList函數(shù)格式化并返回該列表。然后輸出列表。
如你所見,每個功能只負責做一件事。
我們?nèi)匀豢梢詫瘮?shù)進行一些修改,其實,應用程序很需要把視圖從控制器中分離出來,但總體而言,startprogram函數(shù)現(xiàn)在信息很容易懂,理解它的功能絕對沒有困難。如果幾個月后必須重新用這段代碼,那也不是什么難事。
小結
程序員是唯一負責編寫高質(zhì)量代碼的人。我們都應該養(yǎng)成從第一行就寫好代碼的習慣。編寫清晰易懂的代碼并不難,這樣做對你和你的同事都有好處。
