diff --git a/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/redaction/model/Section.java b/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/redaction/model/Section.java index 3e534656..0d6d497b 100644 --- a/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/redaction/model/Section.java +++ b/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/redaction/model/Section.java @@ -2,15 +2,18 @@ package com.iqser.red.service.redaction.v1.server.redaction.model; import static com.iqser.red.service.redaction.v1.server.redaction.model.Dictionary.RECOMMENDATION_PREFIX; +import java.util.Collection; import java.util.HashMap; import java.util.HashSet; import java.util.Map; import java.util.Set; import java.util.regex.Matcher; import java.util.regex.Pattern; +import java.util.stream.Collectors; import org.apache.commons.lang3.StringUtils; +import com.iqser.red.service.redaction.v1.server.classification.model.TextBlock; import com.iqser.red.service.redaction.v1.server.redaction.utils.Patterns; import com.iqser.red.service.redaction.v1.server.redaction.utils.PositionUtil; @@ -292,8 +295,10 @@ public class Section { entity.setMatchedRule(ruleNumber); entity.setRedactionReason(reason); entity.setTargetSequences(value.getTextBlocks() - .get(0) - .getSequences()); // Make sure no other cells with same content are highlighted + .stream() + .map(TextBlock::getSequences) + .flatMap(Collection::stream) + .collect(Collectors.toList())); // Make sure no other cells with same content are highlighted entity.setLegalBasis(legalBasis); Set singleEntitySet = new HashSet<>();