如何在一个react组件中同时使用forwardRef和useRef

利器之路2024-03-15 16:50

在一个子组件中，有一种场景是子组件用forwardRef包了一层，传到子组件的outerRef需要绑定到子组件上，而与此同时，子组件内部也有一个通过useRef创建的innerRef，也需要绑定到子组件上，咋办呢？我做了几次尝试，如果你想直接参考成功的尝试，请跳转

尝试一

如果是两个useRef，可以这样绑定：

jsx 复制代码

const ref1 = useRef(null);
const ref2 = useRef(null);

return (
    <div ref={(el) => {ref1(el); ref2(el)}}>
        ...
    </div>
);

可是这是forwardedRef，我试了之后不成功。

尝试二

写一个mergeRefs方法：

jsx 复制代码

const mergeRefs = (...refs) => {
    return node => {
      for (const ref of refs) {
        ref.current = node
      }
    }
}

return(
    <div ref={mergeRefs(ref1, ref2)}>
        ...
    </div>
);

对于简单的情景，这样也可以，但我的场景还是不work。

尝试三

也就是优化mergeRefs方法，当ref是个function时，则调用上一个ref：

jsx 复制代码

const setRef = (node, ...refs) => {
    refs.forEach((ref) => {
      if (typeof ref === 'function') {
        ref(node);
      } else if (ref != null) {
        ref.current = node;
      }
    });
}

const mergeRefs = (...refs) => {
    return node => {
      setRef(node, ...refs);
    }
}

return(
    <div ref={mergeRefs(innerRef, forwardedRef)}>
        ...
    </div>
);

成功啦！

结尾

我是在开发带动画效果的draggable list时遇到的这个问题，每一个list item既是droppable的，要绑定dropRef，又是draggable的，要绑定dragRef，又需要从父组件那接收forwardedRef，通过以上第三个尝试，我就可以实现多重功能啦。如果你对这个draggable list感兴趣，可以参考这篇文章：《用react-dnd和react-flip-move做一个带动画的draggable list》。