antd
Version:
An enterprise-class UI design language and React components implementation
20 lines • 716 B
JavaScript
import * as React from 'react';
import { composeRef } from "rc-util/es/ref";
import { FormContext } from '../context';
export default function useItemRef() {
const {
itemRef
} = React.useContext(FormContext);
const cacheRef = React.useRef({});
function getRef(name, children) {
const childrenRef = children && typeof children === 'object' && children.ref;
const nameStr = name.join('_');
if (cacheRef.current.name !== nameStr || cacheRef.current.originRef !== childrenRef) {
cacheRef.current.name = nameStr;
cacheRef.current.originRef = childrenRef;
cacheRef.current.ref = composeRef(itemRef(name), childrenRef);
}
return cacheRef.current.ref;
}
return getRef;
}