React 将多个ref同时赋给一个element

需求:

有一个组件中有两个ref,一个f是orwardRef暴露出的ref,另一个是内部维护的ref

如何将这个两个ref都赋到 一个element上?

const Element = forwardRef((props, ref) => {
    const insiderRef = useRef(null);

    return (
        <div></div>
    )
})

解决方案:

        ref可以接收一个回调函数,这个回调函数在组件挂载或者卸载的时候被调用,函数参数为组件实例或DOM元素。

可以利用此机制,实现一个函数为两个ref赋值。

hooks实现如下:

import { MutableRefObject, Ref, useMemo } from 'react';

function setRef<T>(
  ref: MutableRefObject<T | null> | ((instance: T | null) => void) | null | undefined,
  value: T | null,
): void {
  if (typeof ref === 'function') {
    ref(value);
  } else if (ref) {
    ref.current = value;
  }
}

export default function useForkRef<InstanceA, InstanceB>(
  refA: Ref<InstanceA> | null | undefined,
  refB: Ref<InstanceB> | null | undefined,
): Ref<InstanceA & InstanceB> | null {
  /**
   * This will create a new function if the ref props change and are defined.
   * This means react will call the old forkRef with `null` and the new forkRef
   * with the ref. Cleanup naturally emerges from this behavior.
   */
  return useMemo(() => {
    if (refA == null && refB == null) {
      return null;
    }
    return (refValue) => {
      setRef(refA, refValue);
      setRef(refB, refValue);
    };
  }, [refA, refB]);
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值