React源碼中如何實現受控組件

作者：卡頌 2020-10-21 08:38:47

今天我們站在框架開發者的角度來聊聊如何實現受控組件。

在React中一個簡單的受控組件如下：

function App() { 
  const [num, updateNum] = React.useState(0); 
   
  const onChange = ({target: {value}}) => { 
    updateNum(value); 
  } 
 
  return ( 
    <input value={num} onChange={onChange}/> 
  ) 
}

在onChange中會更新num，num作為value prop傳遞給<input/>，達到value受控的目的。

如果讓你來設計，你會怎么做？

我相信大部分同學第一個想法是：將value prop與其他attribute prop一樣處理就行。

我們知道React內部運行有3個階段：

schedule 調度更新階段
render 進行diff算法的階段
commit 進行DOM操作的階段

假設我們要在onChange中觸發更新改變className，只需要在render階段記錄要改變的className，在commit階段執行對應的addClass DOM操作。

同樣的，如果我們要在onChange中觸發更新改變value，只需要在render階段記錄要改變的value，在commit階段執行對應的inputDOM.setAttribute('value', value)操作。

這樣邏輯非常通順。那么事實上呢？

直接改變value的問題
className只是inputDOM上的一個普通屬性。而value則涉及到輸入框光標的位置。

如果我們直接修改value，那么屬性改變后input的光標輸入位置也會丟失，光標會跳到輸入框的最后。

想想我們將1234修改為12534。

1234 --> 12534

需要先將光標位置移動到2之后，再輸入5。

如果setAttribute('value', '12534')，那么光標不會保持在5后面而是跳到4后面。

那么React如何解決這個問題呢？

用非受控的形式實現受控組件

你沒有看錯，React用非受控形式實現了受控組件的邏輯。

簡單的說，不同于className在commit階段受控更新，value則完全是非受控的形式，只在必要的時候受控更新。

因為一旦更新value，那么光標位置就會丟失。

我們稍微修改下Demo，input為受控組件，value始終為1：

function App() { 
  const num = 1; 
 
  return ( 
    <input value={num}/> 
  ) 
}

當我們在源碼中打上斷點，輸入2后，實際上會先顯示12，再刪掉2。

只不過這個刪除的過程是同步的所以看起來輸入框內始終只有1。

所以，不同于React其他組件props的更新會經歷schedule - render - commit流程。

對于input、textarea、select，React有一條單獨的更新路徑，這條路徑觸發的更新被稱為discreteUpdate。

這條路徑的工作流程如下：

先以非受控的形式更新表單DOM
以同步的優先級開啟一次更新
更新后的value在commit階段并不會像其他props一樣作用于DOM
調用restoreStateOfTarget方法，比較DOM的實際value（即步驟1中的非受控value）與步驟3中更新的value，如果相同則退出，如果不同則用步驟3的value更新DOM

什么情況下這2個value會相同呢？

我們正常的受控組件就是相同的情況：

function App() { 
  const [num, updateNum] = React.useState(0); 
   
  const onChange = ({target: {value}}) => { 
    updateNum(value); 
  } 
 
  return ( 
    <input value={num} onChange={onChange}/> 
  ) 
}

什么情況下這2個value會不同呢？

上面的Demo中，雖然受控，但是沒有調用updateNum更新value的情況：

function App() { 
  const num = 1; 
 
  return ( 
    <input value={num}/> 
  ) 
}

在這種情況下，步驟1的非受控value變為了12，步驟3的受控value還是1，所以最終會用1再更新下DOM的value。

總結
可以看到，要實現一個完備的前端框架，是有非常多細節的。

為了實現受控組件，就得脫離整體更新流程，單獨實現一套流程。

責任編輯：姜華來源：魔術師卡頌

React源碼