diff --git a/pkg/sourcer/examples/simple_source/impl/simple_source.go b/pkg/sourcer/examples/simple_source/impl/simple_source.go index 87ec9dcb..33423ccf 100644 --- a/pkg/sourcer/examples/simple_source/impl/simple_source.go +++ b/pkg/sourcer/examples/simple_source/impl/simple_source.go @@ -76,8 +76,9 @@ func (s *SimpleSource) Read(_ context.Context, readRequest sourcesdk.ReadRequest func (s *SimpleSource) Ack(_ context.Context, request sourcesdk.AckRequest) { s.lock.Lock() defer s.lock.Unlock() - offset := deserializeOffset(request.Offset().Value()) - delete(s.toAckSet, offset) + for _, offset := range request.Offsets() { + delete(s.toAckSet, deserializeOffset(offset.Value())) + } } func (s *SimpleSource) Partitions(_ context.Context) []int32 {