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
Fix
  • Loading branch information
maropu committed Jun 14, 2018
commit 614c87aada943c16faa3f7a718837de0e4dcaeab
Original file line number Diff line number Diff line change
Expand Up @@ -309,8 +309,7 @@ class DataFrameReader private[sql](sparkSession: SparkSession) extends Logging {
val parts: Array[Partition] = predicates.zipWithIndex.map { case (part, i) =>
JDBCPartition(part, i) : Partition
}
val schema = JDBCRelation.getSchema(sparkSession.sessionState.conf.resolver, options)
val relation = JDBCRelation(schema, parts, options)(sparkSession)
val relation = JDBCRelation(parts, options)(sparkSession)
sparkSession.baseRelationToDataFrame(relation)
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -145,6 +145,17 @@ private[sql] object JDBCRelation extends Logging {
case None => tableSchema
}
}

/**
* Resolves a Catalyst schema of a JDBC table and returns [[JDBCRelation]] with the schema.
*/
def apply(
parts: Array[Partition],
jdbcOptions: JDBCOptions)(
sparkSession: SparkSession): JDBCRelation = {
val schema = JDBCRelation.getSchema(sparkSession.sessionState.conf.resolver, jdbcOptions)
JDBCRelation(schema, parts, jdbcOptions)(sparkSession)
}
}

private[sql] case class JDBCRelation(
Expand Down