React

关注公众号 jb51net

关闭
首页 > 网络编程 > JavaScript > javascript类库 > React > React useWatch钩子

React自定义实现useWatch钩子的示例

作者:蓝精灵001

React中实现类似Vue watch的useWatch Hook,包含基础监听、立即执行、动态停止及深度比较四种方式,适用于不同场景,文中通过示例代码介绍的非常详细,需要的朋友们下面随着小编来一起学习学习吧

如何实现类似Vue中watch功能的useWatch自定义Hook,以下是几种常见的实现方式和特点:

1. 基础实现

import { useEffect, useRef } from 'react';
 
function useWatch<T>(value: T, callback: (prev?: T) => void) {
  const prev = useRef<T>();
 
  useEffect(() => {
    callback(prev.current);
    prev.current = value;
  }, [value]);
}

2. 增加immediate选项

import { useEffect, useRef } from 'react';
 
type Config = {
  immediate: boolean;
};
 
function useWatch<T>(value: T, callback: (prev?: T) => void, config: Config = { immediate: false }) {
  const { immediate } = config;
  const prev = useRef<T>();
  const inited = useRef(false);
 
  useEffect(() => {
    const execute = () => callback(prev.current);
 
    if (!inited.current) {
      inited.current = true;
      if (immediate) {
        execute();
      }
    } else {
      execute();
    }
    prev.current = value;
  }, [value]);
}

3. 增加停止监听功能

import { useEffect, useRef } from 'react';
 
type Config = {
  immediate: boolean;
};
 
function useWatch<T>(value: T, callback: (prev?: T) => void, config: Config = { immediate: false }) {
  const { immediate } = config;
  const prev = useRef<T>();
  const inited = useRef(false);
  const stop = useRef(false);
 
  useEffect(() => {
    const execute = () => callback(prev.current);
 
    if (!stop.current) {
      if (!inited.current) {
        inited.current = true;
        if (immediate) {
          execute();
        }
      } else {
        execute();
      }
      prev.current = value;
    }
  }, [value]);
 
  return () => {
    stop.current = true;
  };
}

4. 使用lodash进行深度比较

import { useEffect, useRef } from 'react';
import { isEqual } from 'lodash';
 
function useWatch<T>(value: T, callback: (prev?: T) => void, config: Config = { immediate: false }) {
  const { immediate } = config;
  const prev = useRef<T>();
  const inited = useRef(false);
 
  useEffect(() => {
    const execute = () => {
      if (!isEqual(prev.current, value)) {
        callback(prev.current);
      }
    };
 
    if (!inited.current) {
      inited.current = true;
      if (immediate) {
        execute();
      }
    } else {
      execute();
    }
    prev.current = value;
  }, [value]);
}

使用示例

import React, { useState } from 'react';
import { useWatch } from './useWatch';
 
function App() {
  const [count, setCount] = useState(0);
 
  useWatch(count, (oldCount) => {
    console.log(`count changed from ${oldCount} to ${count}`);
  }, { immediate: true });
 
  return (
    <div>
      <p>Count: {count}</p>
      <button onClick={() => setCount(count + 1)}>Increment</button>
    </div>
  );
}
 
export default App;

总结

这些实现方式各有特点,可以根据具体需求选择合适的版本。如果需要简单的监听功能,基础实现就足够了;如果需要更复杂的控制,如立即执行或停止监听,则可以选择带有immediate和停止功能的版本。

到此这篇关于React自定义实现useWatch钩子的示例的文章就介绍到这了,更多相关React useWatch钩子内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文