From 5cd092114f1f6d8fcc3e389f147dfeec52897d38 Mon Sep 17 00:00:00 2001
From: "miriam.baglioni" <miriam.baglioni@isti.cnr.it>
Date: Fri, 17 Apr 2020 17:13:18 +0200
Subject: [PATCH] use mergeFrom method to add the new community contexts

---
 .../SparkResultToCommunityFromOrganizationJob2.java        | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/dhp-workflows/dhp-propagation/src/main/java/eu/dnetlib/dhp/resulttocommunityfromorganization/SparkResultToCommunityFromOrganizationJob2.java b/dhp-workflows/dhp-propagation/src/main/java/eu/dnetlib/dhp/resulttocommunityfromorganization/SparkResultToCommunityFromOrganizationJob2.java
index 3116161e9..ff94edbf9 100644
--- a/dhp-workflows/dhp-propagation/src/main/java/eu/dnetlib/dhp/resulttocommunityfromorganization/SparkResultToCommunityFromOrganizationJob2.java
+++ b/dhp-workflows/dhp-propagation/src/main/java/eu/dnetlib/dhp/resulttocommunityfromorganization/SparkResultToCommunityFromOrganizationJob2.java
@@ -98,15 +98,20 @@ public class SparkResultToCommunityFromOrganizationJob2 {
                     if(rcl.isPresent()){
                         ArrayList<String> communitySet = rcl.get().getCommunityList();
                         List<String> contextList = ret.getContext().stream().map(con -> con.getId()).collect(Collectors.toList());
+                        Result res = new Result();
+                        res.setId(ret.getId());
+                        List<Context> propagatedContexts = new ArrayList<>();
                         for(String cId:communitySet){
                             if(!contextList.contains(cId)){
                                 Context newContext = new Context();
                                 newContext.setId(cId);
                                 newContext.setDataInfo(Arrays.asList(getDataInfo(PROPAGATION_DATA_INFO_TYPE,
                                         PROPAGATION_RESULT_COMMUNITY_ORGANIZATION_CLASS_ID, PROPAGATION_RESULT_COMMUNITY_ORGANIZATION_CLASS_NAME)));
-                                ret.getContext().add(newContext);
+                                propagatedContexts.add(newContext);
                             }
                         }
+                        res.setContext(propagatedContexts);
+                        ret.mergeFrom(res);
                     }
                     return ret;
                 }, Encoders.bean(resultClazz))