[jira] [Created] (SPARK-19223) InputFileBlockHolder doesn't work with Python UDF for datasource other than FileFormat

["Liang-Chi Hsieh (JIRA)" <ji...@apache.org>]

Liang-Chi Hsieh created SPARK-19223:
---------------------------------------

             Summary: InputFileBlockHolder doesn't work with Python UDF for datasource other than FileFormat
                 Key: SPARK-19223
                 URL: https://issues.apache.org/jira/browse/SPARK-19223
             Project: Spark
          Issue Type: Bug
          Components: PySpark, SQL
            Reporter: Liang-Chi Hsieh


For the datasource other than FileFormat, such as spark-xml which is based on BaseRelation and uses HadoopRDD, NewHadoopRDD, InputFileBlockHolder doesn't work with Python UDF.

The method to reproduce it is, running the following codes with {{bin/pyspark --packages com.databricks:spark-xml_2.11:0.4.1}}:


{code}
from pyspark.sql.functions import udf,input_file_name
from pyspark.sql.types import StringType
from pyspark.sql import SparkSession

def filename(path):
    return path

session = SparkSession.builder.appName('APP').getOrCreate()

session.udf.register('sameText',filename)
sameText = udf(filename, StringType())

df = session.read.format('xml').load('a.xml', rowTag='root').select('*',input_file_name().alias('file'))
df.select('file').show()  // works
df.select(sameText(df['file'])).show()  // returns empty content
{code}

a.xml:
{code}
<root>
  <x>TEXT</x>
  <y>TEXT2</y>
</root>
{code}





--
This message was sent by Atlassian JIRA
(v6.3.4#6332)

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