File tree Expand file tree Collapse file tree
iped-engine/src/main/java/iped/engine/task Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -177,6 +177,36 @@ public boolean hasAnyResult() {
177177 }
178178 }
179179
180+ static boolean shouldReplaceReviewPriority (String currentPriority , String newPriority ) {
181+ if (newPriority == null || newPriority .isBlank ()) {
182+ return false ;
183+ }
184+ if (currentPriority == null || currentPriority .isBlank ()) {
185+ return true ;
186+ }
187+ return getReviewPriorityRank (newPriority ) > getReviewPriorityRank (currentPriority );
188+ }
189+
190+ static int getReviewPriorityRank (String reviewPriority ) {
191+ if (reviewPriority == null ) {
192+ return Integer .MIN_VALUE ;
193+ }
194+ switch (reviewPriority ) {
195+ case "VERY_HIGH" :
196+ return 5 ;
197+ case "HIGH" :
198+ return 4 ;
199+ case "MEDIUM" :
200+ return 3 ;
201+ case "LOW" :
202+ return 2 ;
203+ case "VERY_LOW" :
204+ return 1 ;
205+ default :
206+ return Integer .MIN_VALUE ;
207+ }
208+ }
209+
180210 /**
181211 * Represents a Zip archive holding files to be sent for classification.
182212 */
@@ -432,7 +462,8 @@ private void processResult(InputStream responseStream) throws IOException {
432462
433463 // New API payload
434464 JsonNode reviewPriority = item .get ("reviewPriority" );
435- if (reviewPriority != null && !reviewPriority .isNull () && !reviewPriority .asText ().isBlank ()) {
465+ if (reviewPriority != null && !reviewPriority .isNull () && !reviewPriority .asText ().isBlank ()
466+ && shouldReplaceReviewPriority (res .getReviewPriority (), reviewPriority .asText ())) {
436467 res .setReviewPriority (reviewPriority .asText ());
437468 }
438469 JsonNode embeddings = item .get ("embeddings" );
You can’t perform that action at this time.
0 commit comments