Skip to content

Commit f7fa5c9

Browse files
committed
Do not rely on StreamExecution class
1 parent 5477fa0 commit f7fa5c9

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

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

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,6 @@
5858
import org.apache.spark.sql.execution.SQLExecution;
5959
import org.apache.spark.sql.execution.SparkPlanInfo;
6060
import org.apache.spark.sql.execution.metric.SQLMetricInfo;
61-
import org.apache.spark.sql.execution.streaming.StreamExecution;
6261
import org.apache.spark.sql.execution.ui.SparkListenerSQLExecutionEnd;
6362
import org.apache.spark.sql.execution.ui.SparkListenerSQLExecutionStart;
6463
import org.apache.spark.sql.streaming.SourceProgress;
@@ -1264,7 +1263,7 @@ private static String getStreamingBatchKey(Properties properties) {
12641263
return null;
12651264
}
12661265

1267-
Object queryId = properties.get(StreamExecution.QUERY_ID_KEY());
1266+
Object queryId = properties.get("sql.streaming.queryId");
12681267
Object batchId = properties.get("streaming.sql.batchId");
12691268

12701269
if (queryId == null || batchId == null) {

0 commit comments

Comments
 (0)