[react] s3 파일 다운로드 버튼 구현

서연주·2021년 10월 29일
2

Solution

목록 보기
4/7

완전히 새로운 서비스로 넘어가기 전, 자료 소유자들이 pdf로 자료를 다운받을 수 있도록 다운로드 버튼을 만들어달라는 요청을 받았다.

이미 기존에 자료 주문 번호를 가지고 s3에 올라가있는 자료 주소를 받아오는 커스텀 Hook 있었으므로 그것을 활용해서 클라이언트 사이드에서 파일을 다운받을 수 있도록 해야했다.

import React, { useState, useEffect } from 'react';
import { usePdfRead } from '@src/hooks/pdf';

export default function DownloadButton({ orderItemId }: T) {
	const [downloadable, setDownloadable] = useState(false);
	const { data, loading } = usePdfRead<T>(orderItemId);

  // pdfUrl이 로드되었는지 확인
	useEffect(() => {
		if (data && data.pdfUrl) setDownloadable(true);
	}, [data]);

	const handleDownload = () => {
		if (downloadable) {
			fetch(data.pdfUrl, { method: 'GET' })
				.then(res => {
					return res.blob();
				})
				.then(blob => {
					const url = window.URL.createObjectURL(blob);
					const a = document.createElement('a');
					a.href = url;
					a.download = `{원하는 파일명}.pdf`;
					document.body.appendChild(a);
					a.click();
					setTimeout(_ => {
						window.URL.revokeObjectURL(url);
					}, 60000);
					a.remove();
					setOpen(false);
				})
				.catch(err => {
					console.error('err: ', err);
				});
		} else {
			alert(
				'PDF 다운에 실패했습니다. 다시 한 번 시도해주세요. 지속적인 실패 시 문의부탁드립니다.',
			);
		}
	};

	return (
		<Button onClick={handleDownload}>
  <div>
						다운로드
</div>
</Button>
	);
}

참고한 주소

profile
pizz@ttang

1개의 댓글

comment-user-thumbnail
2022년 1월 5일

몇일을 고민하던것을 덕분에 해결하였습니다ㅠㅠ 정말 감사합니다!

답글 달기