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

This commit is contained in:
Miriam Baglioni 2024-04-02 16:28:27 +02:00
parent 64cbd8abe9
commit b42bdd5fb3
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();
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()