React

关注公众号 jb51net

关闭
首页 > 网络编程 > JavaScript > javascript类库 > React > React自定义useWatch

React自定义实现useWatch的方式和特点

投稿:mrr

这篇文章主要介绍了React自定义实现useWatch的方式和特点,本文通过实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友参考下吧

如何实现类似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内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

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