getPropagationEntityIds improvements

This commit is contained in:
YevhenBondarenko 2021-01-25 11:32:15 +02:00 committed by Andrew Shvayka
parent d9f1872c8c
commit 4d4d189816

View File

@ -393,7 +393,9 @@ public class BaseAlarmService extends AbstractEntityService implements AlarmServ
private Set<EntityId> getPropagationEntityIds(Alarm alarm) {
if (alarm.isPropagate()) {
List<EntityRelation> relations = relationService.findByTo(alarm.getTenantId(), alarm.getId(), RelationTypeGroup.ALARM);
return relations.stream().map(EntityRelation::getFrom).collect(Collectors.toSet());
Set<EntityId> propagationEntityIds = relations.stream().map(EntityRelation::getFrom).collect(Collectors.toSet());
propagationEntityIds.add(alarm.getOriginator());
return propagationEntityIds;
} else {
return Collections.singleton(alarm.getOriginator());
}