Marmelator/ocr-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 64 Wiki Activity

RED-7669: optimize OCR-module performance

Browse Source 
* fix thread handling for PDFs without any images
This commit is contained in:
Kilian Schuettler 2023-12-22 15:04:52 +01:00
parent 80d38fb785
commit 6f29270e66
2 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ocr-service-v1/ocr-service-processor/src/main/java/com/knecon/fforesight/service/ocr/processor/service/OcrImageFactory.java
View File

@ -96,6 +96,7 @@ public class OcrImageFactory {
}
imageProcessingThread.setAllImagesExtracted(true);
imageProcessingThread.interrupt();
imageProcessingThread.join();

13
ocr-service-v1/ocr-service-processor/src/main/java/com/knecon/fforesight/service/ocr/processor/service/threads/ImageProcessingThread.java
View File

@ -57,13 +57,16 @@ public class ImageProcessingThread extends Thread {
@Override
public void run() {
while (!allImagesExtracted) {
final UnprocessedImage image = imageInputQueue.take();
var ocrImage = this.process(image);
imageOutputQueue.put(ocrImage);
try {
while (!allImagesExtracted) {
final UnprocessedImage image = imageInputQueue.take();
var ocrImage = this.process(image);
imageOutputQueue.put(ocrImage);
}
} catch (InterruptedException e) {
log.info("All images extracted, emptying processing queue and stopping");
}
try {
while (true) {
final UnprocessedImage image = imageInputQueue.remove();