react-redux的connect与React.forwardRef结合ref失效的解决-前端开发

这篇文章主要介绍了react-redux的connect与React.forwardRef结合ref失效的解决方案，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教

react-redux的connect与React.forwardRef结合ref失效

开发中遇到connect包裹forwardRef组件后，ref引用失效，返回undefined的问题。

const CompA = React.forwardRef<OutputInterface, IProps>((props, ref) => {
  // ...
})
export default connect(mapStateToProps, mapDispatchToProps)(CompA);

原因

源码：React.forwardRef

从下面截取的源码中，可以看到forwardRef返回的是一个元素对象，包含了元素的类型REACT_FORWARD_REF_TYPE，以及渲染组件函数render方法，而connect需要的是组件，故透传出现问题。

import {REACT_FORWARD_REF_TYPE, REACT_MEMO_TYPE} from 'shared/ReactSymbols';
export function forwardRef<Props, ElementType: React$ElementType>(
  render: (props: Props, ref: React$Ref<ElementType>) => React$Node,
) {
  // 省略边界条件判断
  const elementType = {
    $$typeof: REACT_FORWARD_REF_TYPE,
    render,
  };
  if (__DEV__) {
    let ownName;
    Object.defineProperty(elementType, 'displayName', {
      enumerable: false,
      configurable: true,
      get: function() {
        return ownName;
      },
      set: function(name) {
        ownName = name;
        if (!render.name && !render.displayName) {
          render.displayName = name;
        }
      },
    });
  }
  return elementType;
}

解决办法

方法一：调用forwardRef返回值中的render方法

const CompA = React.forwardRef<OutputInterface, IProps>((props, ref) => {
  // ...
})
// 直接通过render获取渲染组件
export default connect(mapStateToProps, mapDispatchToProps)(CompA.render);

方法二：connect第4个参数可以设置{forwardRef: true}实现透传

const CompA = React.forwardRef<OutputInterface, IProps>((props, ref) => {
  // ...
})
// 直接通过render获取渲染组件
export default connect(mapStateToProps, mapDispatchToProps, null, {
  forwardRef: true
})(CompA);