log StateSourceOptions optionally

eason-yuchen-liu · Jun 13, 2024 · 4d4cd70 · 4d4cd70
1 parent 20e1b9c
commit 4d4cd70
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/.../src/main/scala/org/apache/spark/sql/execution/datasources/v2/state/StateDataSource.scala b/.../src/main/scala/org/apache/spark/sql/execution/datasources/v2/state/StateDataSource.scala
@@ -123,8 +123,9 @@ case class StateSourceOptions(
 
   override def toString: String = {
     s"StateSourceOptions(checkpointLocation=$resolvedCpLocation, batchId=$batchId, " +
-      s"operatorId=$operatorId, storeName=$storeName, joinSide=$joinSide, " +
-      s"snapshotStartBatchId=$snapshotStartBatchId, snapshotPartitionId=$snapshotPartitionId)"
+      s"operatorId=$operatorId, storeName=$storeName, joinSide=$joinSide" +
+      snapshotStartBatchId.map(", snapshotStartBatchId=" + _).getOrElse("") +
+      snapshotPartitionId.map(", snapshotPartitionId=" + _).getOrElse("") + ")"
   }
 }