[UsageCount] add check in case the datasource is not matched against those present in the graph

(cherry picked from commit b42bdd5fb3)
This commit is contained in:
Miriam Baglioni 2024-04-02 16:28:27 +02:00 committed by Claudio Atzori
parent 10696f2a44
commit 191fc3a461
1 changed files with 5 additions and 2 deletions

View File

@ -111,8 +111,11 @@ public class SparkAtomicActionUsageJob implements Serializable {
resultModel
.joinWith(datasource, resultModel.col("datasourceId").equalTo(datasource.col("id")), "left")
.map((MapFunction<Tuple2<UsageStatsResultModel, Datasource>, UsageStatsResultModel>) t2 -> {
UsageStatsResultModel usrm = t2._1();
usrm.setDatasourceId(usrm.getDatasourceId() + "||" + t2._2().getOfficialname().getValue());
if(Optional.ofNullable(t2._2()).isPresent())
usrm.setDatasourceId(usrm.getDatasourceId() + "||" + t2._2().getOfficialname().getValue());
else
usrm.setDatasourceId(usrm.getDatasourceId() + "||NO_MATCH_FOUND");
return usrm;
return usrm;
}, Encoders.bean(UsageStatsResultModel.class))
.write()