fix in jobNurn collection (#2)

nubank · Jul 11, 2024 · ee9496c · ee9496c
1 parent 6381cb5
commit ee9496c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/...park/app/src/main/java/io/openlineage/spark/agent/lifecycle/SparkSQLExecutionContext.java b/...park/app/src/main/java/io/openlineage/spark/agent/lifecycle/SparkSQLExecutionContext.java
@@ -236,6 +236,12 @@ public void setActiveJob(ActiveJob activeJob) {
   @Override
   public void start(SparkListenerJobStart jobStart) {
     log.debug("SparkListenerJobStart - executionId: {}", executionId);
+    try {
+      jobName = jobStart.properties().getProperty("spark.job.name");
+    } catch (RuntimeException e) {
+      log.info("spark.job.name property not found in the context");
+    }
+    olContext.setJobNurn(jobName);
     if (!olContext.getQueryExecution().isPresent()) {
       log.info(NO_EXECUTION_INFO, olContext);
       return;