From 273f2a20396a0e199734ff724c6df05405641c27 Mon Sep 17 00:00:00 2001 From: lkd9125 Date: Thu, 21 Sep 2023 15:33:50 +0900 Subject: [PATCH] =?UTF-8?q?Exception=EC=B6=94=EA=B0=80?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../java/com/palnet/comn/utils/PdfUtils.java | 19 ++++++++++++++----- 1 file changed, 14 insertions(+), 5 deletions(-) diff --git a/pav-server/src/main/java/com/palnet/comn/utils/PdfUtils.java b/pav-server/src/main/java/com/palnet/comn/utils/PdfUtils.java index 2559af30..b88bea2f 100644 --- a/pav-server/src/main/java/com/palnet/comn/utils/PdfUtils.java +++ b/pav-server/src/main/java/com/palnet/comn/utils/PdfUtils.java @@ -2,6 +2,7 @@ package com.palnet.comn.utils; import java.io.File; import java.io.FileInputStream; +import java.io.FileNotFoundException; import java.io.FileOutputStream; import java.io.IOException; import java.io.InputStream; @@ -20,6 +21,7 @@ import javax.servlet.http.HttpServletResponse; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Value; +import org.springframework.security.access.method.P; import org.springframework.stereotype.Component; import org.thymeleaf.TemplateEngine; import org.thymeleaf.context.Context; @@ -206,13 +208,20 @@ public class PdfUtils { outputStream.write(buffer); } - } catch (Exception e) { + } catch (FileNotFoundException e) { + throw new CustomException(ErrorCode.DATA_NO); + } catch (IOException e){ e.printStackTrace(); - } finally { + } finally { try { - Objects.requireNonNull(outputStream).flush(); - outputStream.close(); - inputStream.close(); + if(outputStream != null){ + Objects.requireNonNull(outputStream).flush(); + outputStream.close(); + } + + if(inputStream != null){ + inputStream.close(); + } } catch (IOException e) { e.printStackTrace(); }