Expose java beam error handler API

spotify · Aug 21, 2024 · 513118f · 513118f
1 parent bc2c965
commit 513118f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/scio-core/src/main/scala/com/spotify/scio/ScioContext.scala b/scio-core/src/main/scala/com/spotify/scio/ScioContext.scala
@@ -51,6 +51,7 @@ import scala.reflect.ClassTag
 import scala.util.control.NoStackTrace
 import scala.util.{Failure, Success, Try}
 import org.apache.beam.runners.dataflow.options.DataflowPipelineOptions
+import org.apache.beam.sdk.transforms.errorhandling.{BadRecord, ErrorHandler}
 
 /** Runner specific context. */
 trait RunnerContext {
@@ -854,6 +855,11 @@ class ScioContext private[scio] (
   // =======================================================================
   // Error handler
   // =======================================================================
+  def registerBadRecordErrorHandler[T <: POutput](
+    sinkTransform: PTransform[PCollection[BadRecord], T]
+  ): ErrorHandler[BadRecord, T] =
+    pipeline.registerBadRecordErrorHandler(sinkTransform)
+
   def errorSink(): ErrorSink = ErrorSink(this)
 
   // =======================================================================