Skip to content
Closed
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
rebase and implement #24776 changes
  • Loading branch information
manu-olx committed Jun 4, 2019
commit 0c3bc55320716dc2cc801c7384154d0af1b98f30
Original file line number Diff line number Diff line change
Expand Up @@ -259,7 +259,7 @@ case class AlterTableAddColumnsCommand(

// Text format doesn't need ADD COLUMNS as text datasource can have only one column
override val supportedDatasourceFormats = Seq("ParquetFileFormat", "OrcFileFormat",
"OrcDataSourceV2", "JsonDataSourceV2", "CSVDataSourceV2")
"OrcDataSourceV2", "JsonFileFormat", "JsonDataSourceV2", "CSVFileFormat", "CSVDataSourceV2")

override def run(sparkSession: SparkSession): Seq[Row] = {
val catalog = sparkSession.sessionState.catalog
Expand Down Expand Up @@ -296,7 +296,7 @@ case class AlterTableReplaceColumnsCommand(
// Csv format not supported by REPLACE COLUMNS as csv datasource is read positionally
// and a new replacement column would reference an old replaced column's data
override val supportedDatasourceFormats = Seq("ParquetFileFormat", "OrcFileFormat",
"OrcDataSourceV2", "JsonDataSourceV2", "TextDataSourceV2")
"OrcDataSourceV2", "JsonFileFormat", "JsonDataSourceV2", "TextFileFormat", "TextDataSourceV2")

override def run(sparkSession: SparkSession): Seq[Row] = {
val catalog = sparkSession.sessionState.catalog
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2678,13 +2678,20 @@ abstract class DDLSuite extends QueryTest with SQLTestUtils {
}
}

val supportedNativeFileFormatsForAlterTableReplaceColumns = Seq("parquet", "orc", "json", "text")
val supportedNativeFileFormatsForAlterTableReplaceColumns = Seq("parquet", "orc", "json", "text",
"org.apache.spark.sql.execution.datasources.parquet.ParquetFileFormat",
"org.apache.spark.sql.execution.datasources.orc.OrcFileFormat",
"org.apache.spark.sql.execution.datasources.json.JsonFileFormat",
"org.apache.spark.sql.execution.datasources.text.TextFileFormat")
supportedNativeFileFormatsForAlterTableReplaceColumns.foreach { provider =>
test(s"alter datasource table replace columns - $provider") {
if (provider == "text") testReplaceColumnTextProvider() else testReplaceColumn(provider)
if (provider == "text" | provider.endsWith("TextFileFormat")) testReplaceColumnTextProvider()
else testReplaceColumn(provider)
}
test(s"alter datasource table replace columns - partitioned - $provider") {
if (provider == "text") testReplaceColumnPartitionedTextProvider()
if (provider == "text" | provider.endsWith("TextFileFormat")) {
testReplaceColumnPartitionedTextProvider()
}
else testReplaceColumn(provider)
}
}
Expand Down