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