jrmccluskey commented on code in PR #17334:
URL: https://github.com/apache/beam/pull/17334#discussion_r847536784


##########
sdks/go/pkg/beam/core/runtime/exec/sdf.go:
##########
@@ -343,7 +348,12 @@ func (n *ProcessSizedElementsAndRestrictions) 
ProcessElement(_ context.Context,
                defer func() {
                        <-n.SU
                }()
-               return n.PDo.processSingleWindow(mainIn)
+               continuation, processResult := n.PDo.processSingleWindow(mainIn)
+               if continuation != nil {
+                       n.source.pc = continuation
+                       n.source.selfSu = n

Review Comment:
   I think that works up at the DataSource level if we wanted to (that's how we 
init the SplittableUnit.) That's cleaner with how I was thinking about doing 
the actual split, too (trying to keep all of the logic within DataSource and 
just returning the split to the harness if we do so.) I'll mock up the code for 
that after I go through the rest of the comments



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: github-unsubscr...@beam.apache.org

For queries about this service, please contact Infrastructure at:
us...@infra.apache.org

Reply via email to