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