chore: update side input example (#164)

Signed-off-by: Yashash H L <[email protected]>
numaproj · Oct 30, 2024 · 8f4062f · 8f4062f
1 parent 7ec3f2c
commit 8f4062f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pkg/sideinput/examples/simple_source_with_sideinput/impl/simple_source_sideinput.go b/pkg/sideinput/examples/simple_source_with_sideinput/impl/simple_source_sideinput.go
@@ -75,7 +75,9 @@ func (s *SimpleSource) Read(_ context.Context, readRequest sourcesdk.ReadRequest
 }
 
 func (s *SimpleSource) Ack(_ context.Context, request sourcesdk.AckRequest) {
-	delete(s.toAckSet, deserializeOffset(request.Offset().Value()))
+	for _, offset := range request.Offsets() {
+		delete(s.toAckSet, deserializeOffset(offset.Value()))
+	}
 }
 
 func (s *SimpleSource) Partitions(_ context.Context) []int32 {