[ 
https://issues.apache.org/jira/browse/BEAM-11104?focusedWorklogId=755330&page=com.atlassian.jira.plugin.system.issuetabpanels:worklog-tabpanel#worklog-755330
 ]

ASF GitHub Bot logged work on BEAM-11104:
-----------------------------------------

                Author: ASF GitHub Bot
            Created on: 11/Apr/22 17:19
            Start Date: 11/Apr/22 17:19
    Worklog Time Spent: 10m 
      Work Description: jrmccluskey commented on code in PR #17334:
URL: https://github.com/apache/beam/pull/17334#discussion_r847566145


##########
sdks/go/pkg/beam/core/runtime/exec/fn.go:
##########
@@ -261,21 +274,34 @@ func (n *invoker) ret2(pn typex.PaneInfo, ws 
[]typex.Window, ts typex.EventTime,
 // ret3 handles processing of a trio of return values.
 func (n *invoker) ret3(pn typex.PaneInfo, ws []typex.Window, ts 
typex.EventTime, r0, r1, r2 interface{}) (*FullValue, error) {
        switch {
-       case n.outErrIdx >= 0:
-               if r2 != nil {
-                       return nil, r2.(error)
+       case n.outEtIdx >= 0:
+               if n.outErrIdx == 2 {
+                       if r2 != nil {
+                               return nil, r2.(error)
+                       }
+                       n.ret = FullValue{Windows: ws, Timestamp: 
r0.(typex.EventTime), Elm: r1, Pane: pn}
+                       return &n.ret, nil
                }
-               if n.outEtIdx < 0 {
-                       n.ret = FullValue{Windows: ws, Timestamp: ts, Elm: r0, 
Elm2: r1, Pane: pn}
+               if n.outPcIdx >= 0 {

Review Comment:
   Committed to explicit `==` checks since we know what indices they should be 
at



##########
sdks/go/pkg/beam/core/runtime/exec/fn.go:
##########
@@ -247,9 +253,16 @@ func (n *invoker) ret2(pn typex.PaneInfo, ws 
[]typex.Window, ts typex.EventTime,
                if r1 != nil {
                        return nil, r1.(error)
                }
+               if n.outPcIdx >= 0 {
+                       n.ret = FullValue{Windows: ws, Timestamp: ts, Pane: pn, 
Continuation: r0.(sdf.ProcessContinuation)}
+                       return &n.ret, nil
+               }
                n.ret = FullValue{Windows: ws, Timestamp: ts, Elm: r0, Pane: pn}
                return &n.ret, nil
        case n.outEtIdx == 0:
+               if n.outPcIdx >= 0 {
+                       panic("invoker.ret2: cannot return event time without a 
value")
+               }
                n.ret = FullValue{Windows: ws, Timestamp: r0.(typex.EventTime), 
Elm: r1, Pane: pn}
                return &n.ret, nil
        default:

Review Comment:
   Good catch! Missed that one. 





Issue Time Tracking
-------------------

    Worklog Id:     (was: 755330)
    Time Spent: 7h 20m  (was: 7h 10m)

> [Go SDK] DoFn Self Checkpointing
> --------------------------------
>
>                 Key: BEAM-11104
>                 URL: https://issues.apache.org/jira/browse/BEAM-11104
>             Project: Beam
>          Issue Type: Sub-task
>          Components: sdk-go
>            Reporter: Robert Burke
>            Assignee: Jack McCluskey
>            Priority: P3
>          Time Spent: 7h 20m
>  Remaining Estimate: 0h
>
> Allow SplittableDoFns to self checkpoint.
> Design doc: 
> https://docs.google.com/document/d/1_JbzjY9JR07ZK5v7PcZevUfzHPsqwzfV7W6AouNpMPk/edit?usp=sharing



--
This message was sent by Atlassian Jira
(v8.20.1#820001)

Reply via email to