Fix: use image length instead of fixed value for imageStageRefs

tim-soft · Aug 27, 2019 · 2bdc424 · 2bdc424
1 parent 915b90d
commit 2bdc424
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/components/ImageStage/components/ImagePager/index.js b/src/components/ImageStage/components/ImagePager/index.js
@@ -21,7 +21,9 @@ import Image from '../Image';
  */
 const ImagePager = ({ images, currentIndex, onPrev, onNext, onClose }) => {
   const firstRender = useRef(true);
-  const imageStageRef = useRef([...Array(3)].map(() => React.createRef()));
+  const imageStageRef = useRef(
+    [...Array(images.length)].map(() => React.createRef())
+  );
   const { width: pageWidth } = useWindowSize();
   const [disableDrag, setDisableDrag] = useState(false);