diff --git a/sql/core/src/main/scala/org/apache/spark/sql/execution/streaming/state/RocksDB.scala b/sql/core/src/main/scala/org/apache/spark/sql/execution/streaming/state/RocksDB.scala index fcefc1666f3ab..c6fb9699cf336 100644 --- a/sql/core/src/main/scala/org/apache/spark/sql/execution/streaming/state/RocksDB.scala +++ b/sql/core/src/main/scala/org/apache/spark/sql/execution/streaming/state/RocksDB.scala @@ -886,6 +886,8 @@ class RocksDB( colFamilyNameToHandleMap.values.map(handle => handle.close) colFamilyNameToHandleMap.clear() + // Cancel and wait until all background work finishes + db.cancelAllBackgroundWork(true) // Close the DB instance db.close() db = null