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);