40行代碼把Vue3的響應式集成進React作狀態管理

前言

vue-next是Vue3的源碼倉庫，Vue3採用lerna作package的劃分，而響應式能力@vue/reactivity被劃分到了單獨的一個package中。

若是咱們想把它集成到React中，可行嗎？來試一試吧。

使用示例

話很少說，先看看怎麼用的解解饞吧。

// store.ts
import { reactive, computed, effect } from '@vue/reactivity';

export const state = reactive({
  count: 0,
});

const plusOne = computed(() => state.count + 1);

effect(() => {
  console.log('plusOne changed: ', plusOne);
});

const add = () => (state.count += 1);

export const mutations = {
  // mutation
  add,
};

export const store = {
  state,
  computed: {
    plusOne,
  },
};

export type Store = typeof store;
複製代碼

// Index.tsx
import { Provider, useStore } from 'rxv'
import { mutations, store, Store } from './store.ts'
function Count() {
  const countState = useStore((store: Store) => {
    const { state, computed } = store;
    const { count } = state;
    const { plusOne } = computed;

    return {
      count,
      plusOne,
    };
  });

  return (
    <Card hoverable style={{ marginBottom: 24 }}> <h1>計數器</h1> <div className="chunk"> <div className="chunk">store中的count如今是 {countState.count}</div> <div className="chunk">computed值中的plusOne如今是 {countState.plusOne.value}</div> <Button onClick={mutations.add}>add</Button> </div> </Card>
  );
}

export default () => {
  return (
    <Provider value={store}> <Count /> </Provider>
  );
};
複製代碼

能夠看出，store的定義只用到了@vue/reactivity，而rxv只是在組件中作了一層橋接，連通了Vue3和React，而後咱們就能夠盡情的使用Vue3的響應式能力啦。

預覽

能夠看到，完美的利用了reactive、computed的強大能力。

分析

從這個包提供的幾個核心api來分析：

effect（重點）

effect實際上是響應式庫中一個通用的概念：觀察函數，就像Vue2中的Watcher，mobx中的autorun，observer同樣，它的做用是收集依賴。

它接受的是一個函數，它會幫你執行這個函數，而且開啓依賴收集，

這個函數內部對於響應式數據的訪問均可以收集依賴，那麼在響應式數據被修改後，就會觸發更新。

最簡單的用法

const data = reactive({ count: 0 })
effect(() => {
    // 就是這句話 訪問了data.count
    // 從而收集到了依賴
    console.log(data.count)
})

data.count = 1
// 控制檯打印出1
複製代碼

那麼若是把這個簡單例子中的

() => {
    // 就是這句話 訪問了data.count
    // 從而收集到了依賴
    console.log(data.count)
}
複製代碼

這個函數，替換成React的組件渲染，是否是就能達成響應式更新組件的目的了？

reactive（重點）

響應式數據的核心api，這個api返回的是一個proxy，對上面全部屬性的訪問都會被劫持，從而在get的時候收集依賴（也就是正在運行的effect），在set的時候觸發更新。

ref

對於簡單數據類型好比number，咱們不可能像這樣去作：

let data = reactive(2)
// 😭oops
data = 5
複製代碼

這是不符合響應式的攔截規則的，沒有辦法能攔截到data自己的改變，只能攔截到data身上的屬性的改變，因此有了ref。

const data = ref(2)
// 💕ok
data.value= 5
複製代碼

computed

計算屬性，依賴值更新之後，它的值也會隨之自動更新。其實computed內部也是一個effect。

擁有在computed中觀察另外一個computed數據、effect觀察computed改變之類的高級特性。

實現

從這幾個核心api來看，只要effect能接入到React系統中，那麼其餘的api都沒什麼問題，由於它們只是去收集effect的依賴，去通知effect觸發更新。

effect接受的是一個函數，並且effect還支持經過傳入schedule參數來自定義依賴更新的時候須要觸發什麼函數，若是咱們把這個schedule替換成對應組件的更新呢？要知道在hook的世界中，實現當前組件強制更新但是很簡單的：

useForceUpdate

export const useForceUpdate = () => {
  const [, forceUpdate] = useReducer(s => s + 1, 0);
  return forceUpdate;
};
複製代碼

這是一個很經典的自定義hook，經過不斷的把狀態+1來強行讓組件渲染。

而rxv的核心api: useStore接受的也是一個函數selector，它會讓用戶本身選擇在組件中須要訪問的數據。

那麼思路就顯而易見了：

1. 把selector包裝在effect中執行，去收集依賴。
2. 指定依賴發生更新時，須要調用的函數是當前正在使用useStore的這個組件的forceUpdate強制渲染函數。

這樣不就實現了數據變化，組件自動更新嗎？

簡單的看一下核心實現

useStore和Provider

import React, { useContext } from 'react';
import { useForceUpdate, useEffection } from './share';

type Selector<T, S> = (store: T) => S;

const StoreContext = React.createContext<any>(null);

const useStoreContext = () => {
  const contextValue = useContext(StoreContext);
  if (!contextValue) {
    throw new Error(
      'could not find store context value; please ensure the component is wrapped in a <Provider>',
    );
  }
  return contextValue;
};

/** * 在組件中讀取全局狀態 * 須要經過傳入的函數收集依賴 */
export const useStore = <T, S>(selector: Selector<T, S>): S => {
  const forceUpdate = useForceUpdate();
  const store = useStoreContext();

  const effection = useEffection(() => selector(store), {
    scheduler: forceUpdate,
    lazy: true,
  });

  const value = effection();
  return value;
};

export const Provider = StoreContext.Provider;

複製代碼

這個option是傳遞給Vue3的effectapi，

scheduler規定響應式數據更新之後應該作什麼操做，這裏咱們使用forceUpdate去讓組件從新渲染。

lazy表示延遲執行，後面咱們手動調用effection來執行

{
  scheduler: forceUpdate,
  lazy: true,
}
複製代碼

再來看下useEffection和useForceUpdate

import { useEffect, useReducer, useRef } from 'react';
import { effect, stop, ReactiveEffect } from '@vue/reactivity';

export const useEffection = (...effectArgs: Parameters<typeof effect>) => {
  // 用一個ref存儲effection
  // effect函數只須要初始化執行一遍
  const effectionRef = useRef<ReactiveEffect>();
  if (!effectionRef.current) {
    effectionRef.current = effect(...effectArgs);
  }

  // 卸載組件後取消effect
  const stopEffect = () => {
    stop(effectionRef.current!);
  };
  useEffect(() => stopEffect, []);

  return effectionRef.current
};

export const useForceUpdate = () => {
  const [, forceUpdate] = useReducer(s => s + 1, 0);
  return forceUpdate;
};
複製代碼

也很簡單，就是把傳入的函數交給effect，而且在組件銷燬的時候中止effect而已。

流程

1. 先經過useForceUpdate在當前組件中註冊一個強制更新的函數。
2. 經過useContext讀取用戶從Provider中傳入的store。
3. 再經過Vue的effect去幫咱們執行selector(store)，而且指定scheduler爲forceUpdate，這樣就完成了依賴收集。
4. 那麼在store裏的值更新了之後，觸發了scheduler也就是forceUpdate，咱們的React組件就自動更新啦。

就簡單的幾行代碼，就實現了在React中使用@vue/reactivity中的全部能力。

優勢：

1. 直接引入@vue/reacivity，徹底使用Vue3的reactivity能力，擁有computed, effect等各類能力，而且對於Set和Map也提供了響應式的能力。後續也會隨着這個庫的更新變得更加完善的和強大。
2. vue-next倉庫內部完整的測試用例。
3. 完善的TypeScript類型支持。
4. 徹底複用@vue/reacivity實現超強的全局狀態管理能力。
5. 狀態管理中組件級別的精確更新。
6. Vue3老是要學的嘛，提早學習防止失業！

缺點：

1. 因爲須要精確的收集依賴全靠useStore，因此selector函數必定要精確的訪問到你關心的數據。甚至若是你須要觸發數組內部某個值的更新，那你在useStore中就不能只返回這個數組自己。

舉一個例子：

function Logger() {
  const logs = useStore((store: Store) => {
    return store.state.logs.map((log, idx) => (
      <p className="log" key={idx}> {log} </p>
    ));
  });

  return (
    <Card hoverable> <h1>控制檯</h1> <div className="logs">{logs}</div> </Card>
  );
}
複製代碼

這段代碼直接在useStore中返回了整段jsx，是由於map的過程當中回去訪問數組的每一項來收集依賴，只有這樣才能達到響應式的目的。

源碼地址

github.com/sl1673495/r…

若是你喜歡這個庫，歡迎給出你的star✨，你的支持就是我最大的動力~