Bulk Load CDK: Restore additional properties to checkpoint output (#4…

…6560)
airbytehq · Oct 7, 2024 · bff4613 · bff4613
1 parent 2abba2d
commit bff4613
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/airbyte-cdk/bulk/core/load/src/main/kotlin/io/airbyte/cdk/message/DestinationMessage.kt b/airbyte-cdk/bulk/core/load/src/main/kotlin/io/airbyte/cdk/message/DestinationMessage.kt
@@ -193,6 +193,9 @@ data class StreamCheckpoint(
                             AirbyteStateStats()
                                 .withRecordCount(destinationStats.recordCount.toDouble())
                     }
+                    it.additionalProperties.forEach { (key, value) ->
+                        it.withAdditionalProperty(key, value)
+                    }
                 }
         return AirbyteMessage().withType(AirbyteMessage.Type.STATE).withState(stateMessage)
     }
@@ -232,6 +235,9 @@ data class GlobalCheckpoint(
                             AirbyteStateStats()
                                 .withRecordCount(destinationStats.recordCount.toDouble())
                     }
+                    it.additionalProperties.forEach { (key, value) ->
+                        it.withAdditionalProperty(key, value)
+                    }
                 }
         return AirbyteMessage().withType(AirbyteMessage.Type.STATE).withState(stateMessage)
     }