this should work

ericm-db · Dec 13, 2024 · 2306c7c · 2306c7c
1 parent 883e8ff
commit 2306c7c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/...main/scala/org/apache/spark/sql/execution/streaming/state/RocksDBStateStoreProvider.scala b/...main/scala/org/apache/spark/sql/execution/streaming/state/RocksDBStateStoreProvider.scala
@@ -398,9 +398,11 @@ private[sql] class RocksDBStateStoreProvider
     val avroEnc = getAvroEnc(
       stateStoreEncoding, avroEncCacheKey, keyStateEncoderSpec, valueSchema)
 
-    if (useColumnFamilies) {
+    val columnFamilyInfo = if (useColumnFamilies) {
       defaultColFamilyId = Some(rocksDB.createColFamilyIfAbsent(StateStore.DEFAULT_COL_FAMILY_NAME))
-      val columnFamilyInfo = Some(ColumnFamilyInfo(colFamilyName, defaultColFamilyId.get))
+      Some(ColumnFamilyInfo(colFamilyName, defaultColFamilyId.get))
+    } else {
+      None
     }
 
     val provider = RocksDBStateStoreProvider.this