catch error at the end

Author: aboitreaud

dd-java-agent/instrumentation/spark/spark-common/src/main/java/datadog/trace/instrumentation/spark/AbstractSparkInstrumentation.java

@@ -106,6 +106,20 @@ public static void exit(@Advice.Thrown Throwable throwable) {
       if (AbstractDatadogSparkListener.listener != null) {
         AbstractDatadogSparkListener.listener.finishApplication(
             System.currentTimeMillis(), throwable, 0, null);
+      } else {
+        try {
+          Class<?> adviceClass =
+              Class.forName("datadog.trace.instrumentation.spark.SparkLauncherAdvice");
+          Object span = adviceClass.getDeclaredField("launcherSpan").get(null);
+          if (span != null) {
+            if (throwable != null) {
+              span.getClass().getMethod("addThrowable", Throwable.class).invoke(span, throwable);
+            }
+            span.getClass().getMethod("finish").invoke(span);
+            adviceClass.getDeclaredField("launcherSpan").set(null, null);
+          }
+        } catch (Exception ignored) {
+        }
       }
     }
   }