From e60d2173335f409711b017740453972aec8907ec Mon Sep 17 00:00:00 2001 From: Jason Tackaberry Date: Sun, 2 Aug 2020 01:05:17 -0400 Subject: [PATCH] Include MillisBehindLatest in Record for ScanFunc (#124) --- consumer.go | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/consumer.go b/consumer.go index a01d9da..763b0e2 100644 --- a/consumer.go +++ b/consumer.go @@ -20,7 +20,8 @@ import ( // extends to include the shard id. type Record struct { *kinesis.Record - ShardID string + ShardID string + MillisBehindLatest *int64 } // New creates a kinesis consumer with default settings. Use Option to override @@ -184,7 +185,7 @@ func (c *Consumer) ScanShard(ctx context.Context, shardID string, fn ScanFunc) e case <-ctx.Done(): return nil default: - err := fn(&Record{r, shardID}) + err := fn(&Record{r, shardID, resp.MillisBehindLatest}) if err != nil && err != ErrSkipCheckpoint { return err }