Add PdfDocument constructor to try-with-resources

DEVSIX-3504
itext · May 3, 2021 · 6a8d6be · 6a8d6be
1 parent e563ebe
commit 6a8d6be
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/pdfocr-api/src/test/java/com/itextpdf/pdfocr/helpers/PdfHelper.java b/pdfocr-api/src/test/java/com/itextpdf/pdfocr/helpers/PdfHelper.java
@@ -252,12 +252,13 @@ public static ExtractionStrategy getExtractionStrategy(String pdfPath,
     public static ExtractionStrategy getExtractionStrategy(String pdfPath,
             String layerName, boolean useActualText)
             throws IOException {
-        PdfDocument pdfDocument = new PdfDocument(new PdfReader(pdfPath));
-        ExtractionStrategy strategy = new ExtractionStrategy(layerName);
-        strategy.setUseActualText(useActualText);
-        PdfCanvasProcessor processor = new PdfCanvasProcessor(strategy);
-        processor.processPageContent(pdfDocument.getFirstPage());
-        pdfDocument.close();
-        return strategy;
+        try (PdfReader readerPdf = new PdfReader(pdfPath);
+                PdfDocument pdfDocument = new PdfDocument(readerPdf)) {
+            ExtractionStrategy strategy = new ExtractionStrategy(layerName);
+            strategy.setUseActualText(useActualText);
+            PdfCanvasProcessor processor = new PdfCanvasProcessor(strategy);
+            processor.processPageContent(pdfDocument.getFirstPage());
+            return strategy;
+        }
     }
 }