changed the logic to verify if a community is contained in the list of context of a result

This commit is contained in:
Miriam Baglioni 2021-07-13 14:22:23 +02:00
parent f5486ffb14
commit 2f66fedfec
1 changed files with 4 additions and 7 deletions

View File

@ -57,16 +57,13 @@ public class CommunitySplit implements Serializable {
Dataset<CommunityResult> community_products = result
.filter((FilterFunction<CommunityResult>) r -> containsCommunity(r, c));
try {
community_products.first();
community_products
.write()
.option("compression", "gzip")
.mode(SaveMode.Overwrite)
.json(outputPath + "/" + c);
} catch (Exception e) {
}
}
@ -75,9 +72,9 @@ public class CommunitySplit implements Serializable {
return r
.getContext()
.stream()
.filter(con -> con.getCode().equals(c))
.map(con -> con.getCode())
.collect(Collectors.toList())
.size() > 0;
.contains(c)
}
return false;
}