Hyukjin Kwon created SPARK-23942:

             Summary: PySpark's collect doesn't trigger QueryExecutionListener
                 Key: SPARK-23942
                 URL: https://issues.apache.org/jira/browse/SPARK-23942
             Project: Spark
          Issue Type: Improvement
          Components: SQL
    Affects Versions: 2.4.0
            Reporter: Hyukjin Kwon

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  

>>> sql("SELECT * FROM range(1)").collect()

>>> spark.conf.set("spark.sql.execution.arrow.enabled", "true")
>>> sql("SELECT * FROM range(1)").toPandas()
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])

This message was sent by Atlassian JIRA

To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org
For additional commands, e-mail: issues-h...@spark.apache.org

Reply via email to