William Wong created SPARK-27772:
------------------------------------
Summary: SQLTestUtils Refactoring
Key: SPARK-27772
URL: https://issues.apache.org/jira/browse/SPARK-27772
Project: Spark
Issue Type: Improvement
Components: SQL
Affects Versions: 3.0.0
Reporter: William Wong
The current `SQLTestUtils` created many `withXXX` utility functions to clean up
tables/views/caches created for testing purpose.
Some of those `withXXX` functions would ignore certain exceptions, like
`NoSuchTableException` in the clean up block (finally block).
```
/**
* Drops temporary view `viewNames` after calling `f`.
*/
protected def withTempView(viewNames: String*)(f: => Unit): Unit = {
try f finally {
// If the test failed part way, we don't want to mask the failure by
failing to remove
// temp views that never got created.
try viewNames.foreach(spark.catalog.dropTempView) catch {
case _: NoSuchTableException =>
}
}
}
```
Maybe it is not the best option. If a test hit an exception in the 'f' closure,
no matter what exception we hit in the finally block, we should not mask that
exception with any other exception hit in the finally block. The exception
caught in the finally block should be reattached to the original exception as a
suppressed exception. The idea is similar to how java handle
'try-with-resources' statement.
A proposed to create following function to standardise those 'withXXX'
functions.
```
/**
* Executes the given tryBlock and then the given finallyBlock no matter
whether tryBlock throws
* an exception. If both tryBlock and finallyBlock throw exceptions, the
exception thrown
* from the finallyBlock with be added to the exception thrown from tryBlock as
a
* suppress exception. It helps to avoid masking the exception from tryBlock
with exception
* from finallyBlock
*/
private def withFinallyBlock(tryBlock: => Unit)(finallyBlock: => Unit): Unit = {
var fromTryBlock : Throwable = null
try tryBlock catch {
case cause : Throwable =>
fromTryBlock = cause
throw cause
} finally {
if (fromTryBlock != null) {
try finallyBlock catch {
case fromFinallyBlock : Throwable =>
fromTryBlock.addSuppressed(fromFinallyBlock)
throw fromTryBlock
}
} else {
finallyBlock
}
}
}
```
--
This message was sent by Atlassian JIRA
(v7.6.3#76005)
---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org
For additional commands, e-mail: issues-h...@spark.apache.org
