Github user cloud-fan commented on a diff in the pull request:
https://github.com/apache/spark/pull/16386#discussion_r100586852
--- Diff:
sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/json/JacksonParser.scala
---
@@ -48,69 +47,98 @@ class JacksonParser(
// A `ValueConverter` is responsible for converting a value from
`JsonParser`
// to a value in a field for `InternalRow`.
- private type ValueConverter = (JsonParser) => Any
+ private type ValueConverter = JsonParser => AnyRef
// `ValueConverter`s for the root schema for all fields in the schema
- private val rootConverter: ValueConverter = makeRootConverter(schema)
+ private val rootConverter = makeRootConverter(schema)
private val factory = new JsonFactory()
options.setJacksonOptions(factory)
private val emptyRow: Seq[InternalRow] = Seq(new
GenericInternalRow(schema.length))
+ private val corruptFieldIndex =
schema.getFieldIndex(options.columnNameOfCorruptRecord)
+
+ @transient
+ private val printWarningForMalformedRecord = ExecuteOnce[() =>
UTF8String] { record =>
+ def sampleRecord: String = {
+ if (options.wholeFile) {
+ ""
+ } else {
+ s"Sample record: ${record()}\n"
+ }
+ }
+
+ def footer: String = {
+ s"""Code example to print all malformed records (scala):
+ |===================================================
+ |// The corrupted record exists in column
${options.columnNameOfCorruptRecord}.
+ |val parsedJson = spark.read.json("/path/to/json/file/test.json")
+ |
+ """.stripMargin
+ }
+
+ if (options.permissive) {
+ logWarning(
+ s"""Found at least one malformed record. The JSON reader will
replace
+ |all malformed records with placeholder null in current
$PERMISSIVE_MODE parser mode.
+ |To find out which corrupted records have been replaced with
null, please use the
+ |default inferred schema instead of providing a custom schema.
+ |
+ |${sampleRecord ++ footer}
+ |
+ """.stripMargin)
+ } else if (options.dropMalformed) {
+ logWarning(
+ s"""Found at least one malformed record. The JSON reader will drop
+ |all malformed records in current $DROP_MALFORMED_MODE parser
mode. To find out which
+ |corrupted records have been dropped, please switch the parser
mode to $PERMISSIVE_MODE
+ |mode and use the default inferred schema.
+ |
+ |${sampleRecord ++ footer}
+ |
+ """.stripMargin)
+ }
+ }
+
@transient
- private[this] var isWarningPrintedForMalformedRecord: Boolean = false
+ private val printWarningIfWholeFile = ExecuteOnce[Unit] { _ =>
+ if (options.wholeFile && corruptFieldIndex.isDefined) {
+ logWarning(
+ s"""Enabling wholeFile mode and defining columnNameOfCorruptRecord
may result
+ |in very large allocations or OutOfMemoryExceptions being
raised.
+ |
+ """.stripMargin)
+ }
+ }
/**
* This function deals with the cases it fails to parse. This function
will be called
* when exceptions are caught during converting. This functions also
deals with `mode` option.
*/
- private def failedRecord(record: String): Seq[InternalRow] = {
- // create a row even if no corrupt record column is present
- if (options.failFast) {
- throw new SparkSQLJsonProcessingException(s"Malformed line in
FAILFAST mode: $record")
- }
- if (options.dropMalformed) {
- if (!isWarningPrintedForMalformedRecord) {
- logWarning(
- s"""Found at least one malformed records (sample: $record). The
JSON reader will drop
- |all malformed records in current $DROP_MALFORMED_MODE parser
mode. To find out which
- |corrupted records have been dropped, please switch the
parser mode to $PERMISSIVE_MODE
- |mode and use the default inferred schema.
- |
- |Code example to print all malformed records (scala):
- |===================================================
- |// The corrupted record exists in column
${columnNameOfCorruptRecord}
- |val parsedJson =
spark.read.json("/path/to/json/file/test.json")
- |
- """.stripMargin)
- isWarningPrintedForMalformedRecord = true
- }
- Nil
- } else if (schema.getFieldIndex(columnNameOfCorruptRecord).isEmpty) {
- if (!isWarningPrintedForMalformedRecord) {
- logWarning(
- s"""Found at least one malformed records (sample: $record). The
JSON reader will replace
- |all malformed records with placeholder null in current
$PERMISSIVE_MODE parser mode.
- |To find out which corrupted records have been replaced with
null, please use the
- |default inferred schema instead of providing a custom schema.
- |
- |Code example to print all malformed records (scala):
- |===================================================
- |// The corrupted record exists in column
${columnNameOfCorruptRecord}.
- |val parsedJson =
spark.read.json("/path/to/json/file/test.json")
- |
- """.stripMargin)
- isWarningPrintedForMalformedRecord = true
- }
- emptyRow
- } else {
- val row = new GenericInternalRow(schema.length)
- for (corruptIndex <-
schema.getFieldIndex(columnNameOfCorruptRecord)) {
+ private def failedRecord(record: () => UTF8String): Seq[InternalRow] = {
+ corruptFieldIndex match {
+ case _ if options.failFast =>
+ if (options.wholeFile) {
+ throw new SparkSQLJsonProcessingException("Malformed line in
FAILFAST mode")
+ } else {
+ throw new SparkSQLJsonProcessingException(s"Malformed line in
FAILFAST mode: ${record()}")
+ }
+
+ case _ if options.dropMalformed =>
+ printWarningForMalformedRecord(record)
+ Nil
+
+ case None =>
+ printWarningForMalformedRecord(record)
+ emptyRow
+
+ case Some(corruptIndex) =>
+ printWarningIfWholeFile(())
--- End diff --
nit: extra brackets
---
If your project is set up for it, you can reply to this email and have your
reply appear on GitHub as well. If your project does not have this feature
enabled and wishes so, or if the feature is enabled but not working, please
contact infrastructure at infrastruct...@apache.org or file a JIRA ticket
with INFRA.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org
For additional commands, e-mail: reviews-h...@spark.apache.org
