[SPARK-23942] PySpark's collect doesn't trigger QueryExecutionListener

• Type: Sub-task
• Status: Resolved
• Priority: Major
• Resolution: Fixed
• Affects Version/s: 2.4.0
• Fix Version/s: 2.3.1, 2.4.0
• Component/s: PySpark, SQL
• Assignee: Hyukjin Kwon
• Reporter: Hyukjin Kwon
• Created: 09/Apr/18 09:50
• Updated: 12/Dec/22 18:10
• Resolved: 13/Apr/18 03:29

Description

For example, if you have an custom query execution listener:

package org.apache.spark.sql

import org.apache.spark.internal.Logging
import org.apache.spark.sql.execution.QueryExecution
import org.apache.spark.sql.util.QueryExecutionListener


class TestQueryExecutionListener extends QueryExecutionListener with Logging {
  override def onSuccess(funcName: String, qe: QueryExecution, durationNs: Long): Unit = {
    logError("Look at me! I'm 'onSuccess'")
  }

  override def onFailure(funcName: String, qe: QueryExecution, exception: Exception): Unit = { }
}

and set "spark.sql.queryExecutionListeners org.apache.spark.sql.TestQueryExecutionListener",

>>> sql("SELECT * FROM range(1)").collect()
[Row(id=0)]

>>> spark.conf.set("spark.sql.execution.arrow.enabled", "true")
>>> sql("SELECT * FROM range(1)").toPandas()
   id
0   0

Seems other actions like show and etc fine in Scala side too:

>>> sql("SELECT * FROM range(1)").show()
18/04/09 17:02:04 ERROR TestQueryExecutionListener: Look at me! I'm 'onSuccess'
+---+
| id|
+---+
|  0|
+---+

scala> sql("SELECT * FROM range(1)").collect()
18/04/09 16:58:41 ERROR TestQueryExecutionListener: Look at me! I'm 'onSuccess'
res1: Array[org.apache.spark.sql.Row] = Array([0])

Issue Links

• links to: [Github] Pull Request #21007 (HyukjinKwon)
• links to: [Github] Pull Request #21060 (HyukjinKwon)