linkmore-design
Version:
π πlmη»δ»ΆεΊγπ
19 lines β’ 431 B
JavaScript
import { useMemo } from 'react';
function setRef(ref, value) {
if (typeof ref === 'function') {
ref(value);
} else if (ref) {
ref.current = value;
}
}
export default function useForkRef(refA, refB) {
return useMemo(function () {
if (refA == null && refB == null) {
return null;
}
return function (refValue) {
setRef(refA, refValue);
setRef(refB, refValue);
};
}, [refA, refB]);
}