proper error handling for image clasification

redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/redaction/service/ImageClassificationService.java

@@ -35,7 +35,7 @@ public class ImageClassificationService {
                     var mockFile = new MockMultipartFile("file", "Image.png", "image/png", baos.toByteArray());
                     ImageClassificationResponse response = imageClassificationClient.classify(mockFile);
                     image.setImageType(ImageType.valueOf(response.getCategory()));
-                } catch (IOException e) {
+                } catch (Exception e) {
                     log.error("Could not classify image", e);
                 }
 

redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/segmentation/PdfSegmentationService.java

@@ -127,80 +127,86 @@ public class PdfSegmentationService {
 
 
     public Document parseDocument(InputStream documentInputStream) throws IOException {
+        PDDocument pdDocument = null;
+        try {
+            //create tempFile
+            File tempFile = File.createTempFile("document", ".pdf");
+            IOUtils.copy(documentInputStream, new FileOutputStream(tempFile));
 
-        //create tempFile
+            // initialize required variables
-        File tempFile = File.createTempFile("document", ".pdf");
+            Document document = new Document();
-        IOUtils.copy(documentInputStream, new FileOutputStream(tempFile));
+            List<Page> pages = new ArrayList<>();
-
-        // initialize required variables
-        Document document = new Document();
-        List<Page> pages = new ArrayList<>();
 
 
-        PDDocument pdDocument = reinitializePDDocument(tempFile, null);
+            pdDocument = reinitializePDDocument(tempFile, null);
-        long pageCount = pdDocument.getNumberOfPages();
+            long pageCount = pdDocument.getNumberOfPages();
+
+            for (int pageNumber = 1; pageNumber <= pageCount; pageNumber++) {
+
+                if (pageNumber % MAX_PAGES_BEFORE_GC == 0) {
+                    pdDocument = reinitializePDDocument(tempFile, pdDocument);
+                }
+
+                PDFLinesTextStripper stripper = new PDFLinesTextStripper();
+                PDPage pdPage = pdDocument.getPage(pageNumber - 1);
+                stripper.setPageNumber(pageNumber);
+                stripper.setStartPage(pageNumber);
+                stripper.setEndPage(pageNumber);
+                stripper.setPdpage(pdPage);
+                stripper.getText(pdDocument);
+
+                PDRectangle pdr = pdPage.getMediaBox();
+                boolean isLandscape = pdr.getWidth() > pdr.getHeight();
+
+                int rotation = pdPage.getRotation();
+                boolean isRotated = rotation != 0 && rotation != 360;
+
+                CleanRulings cleanRulings = rulingCleaningService.getCleanRulings(stripper.getRulings(), stripper.getMinCharWidth(), stripper
+                        .getMaxCharHeight());
+
+                Page page = blockificationService.blockify(stripper.getTextPositionSequences(), cleanRulings.getHorizontal(), cleanRulings
+                        .getVertical());
+                page.setRotation(rotation);
+
+                tableExtractionService.extractTables(cleanRulings, page);
+
+                buildPageStatistics(page);
+
+                page.setLandscape(isLandscape || isRotated);
+
+                page.setPageNumber(pageNumber);
+                increaseDocumentStatistics(page, document);
+
+                page.setImages(stripper.getImages());
+
+                imageClassificationService.classifyImages(page);
+
+                pages.add(page);
 
-        for (int pageNumber = 1; pageNumber <= pageCount; pageNumber++) {
 
-            if (pageNumber % MAX_PAGES_BEFORE_GC == 0) {
-                pdDocument = reinitializePDDocument(tempFile, pdDocument);
             }
 
-            PDFLinesTextStripper stripper = new PDFLinesTextStripper();
+            document.setPages(pages);
-            PDPage pdPage = pdDocument.getPage(pageNumber - 1);
-            stripper.setPageNumber(pageNumber);
-            stripper.setStartPage(pageNumber);
-            stripper.setEndPage(pageNumber);
-            stripper.setPdpage(pdPage);
-            stripper.getText(pdDocument);
 
-            PDRectangle pdr = pdPage.getMediaBox();
+            classificationService.classifyDocument(document);
-            boolean isLandscape = pdr.getWidth() > pdr.getHeight();
+            sectionsBuilderService.buildSections(document);
+            sectionsBuilderService.addImagesToSections(document);
 
-            int rotation = pdPage.getRotation();
+            pdDocument = reinitializePDDocument(tempFile, pdDocument);
-            boolean isRotated = rotation != 0 && rotation != 360;
 
-            CleanRulings cleanRulings = rulingCleaningService.getCleanRulings(stripper.getRulings(), stripper.getMinCharWidth(), stripper
+            // This can be improved an done in one pass, but it's complicated to do right away
-                    .getMaxCharHeight());
+            postProcessSections(pdDocument, document.getSectionText());
 
-            Page page = blockificationService.blockify(stripper.getTextPositionSequences(), cleanRulings.getHorizontal(), cleanRulings
+            IOUtils.close(pdDocument);
-                    .getVertical());
-            page.setRotation(rotation);
-
-            tableExtractionService.extractTables(cleanRulings, page);
-
-            buildPageStatistics(page);
-
-            page.setLandscape(isLandscape || isRotated);
-
-            page.setPageNumber(pageNumber);
-            increaseDocumentStatistics(page, document);
-
-            page.setImages(stripper.getImages());
-
-            imageClassificationService.classifyImages(page);
-
-            pages.add(page);
 
+            tempFile.delete();
 
+            return document;
+        } finally {
+            if (pdDocument != null) {
+                pdDocument.close();
+            }
         }
-
-        document.setPages(pages);
-
-        classificationService.classifyDocument(document);
-        sectionsBuilderService.buildSections(document);
-        sectionsBuilderService.addImagesToSections(document);
-
-        pdDocument = reinitializePDDocument(tempFile, pdDocument);
-
-        // This can be improved an done in one pass, but it's complicated to do right away
-        postProcessSections(pdDocument, document.getSectionText());
-
-        IOUtils.close(pdDocument);
-
-        tempFile.delete();
-
-        return document;
     }
 
     private PDDocument reinitializePDDocument(File tempFile, PDDocument pdDocument) throws IOException {

redaction-service-v1/redaction-service-server-v1/src/test/java/com/iqser/red/service/redaction/v1/server/RedactionIntegrationTest.java

@@ -19,7 +19,6 @@ import lombok.SneakyThrows;
 import org.apache.commons.io.IOUtils;
 import org.junit.After;
 import org.junit.Before;
-import org.junit.Ignore;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.kie.api.KieServices;
@@ -452,7 +451,6 @@ public class RedactionIntegrationTest {
 
 
     @Test
-    @Ignore
     public void testLargeScannedFileOOM() {
         AnalyzeRequest request = prepareStorage("scanned/VV-377031.pdf");
         MemoryStats.printMemoryStats();