From c64b40b4adc4a2f67f1ba8a139fafebd4b78a5a1 Mon Sep 17 00:00:00 2001
From: Harlow Ward <harlow@hward.com>
Date: Wed, 22 Nov 2017 14:21:19 -0800
Subject: [PATCH] Increment counts in correct usage points

---
 consumer.go | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/consumer.go b/consumer.go
index 5016067..fce6d30 100644
--- a/consumer.go
+++ b/consumer.go
@@ -185,18 +185,17 @@ loop:
 						break loop
 					default:
 						lastSeqNum = *r.SequenceNumber
+						c.counter.Add("records", 1)
 						if ok := fn(r); !ok {
 							break loop
 						}
 					}
 				}
 
-				c.counter.Add("records", int64(len(resp.Records)))
-				c.counter.Add("checkpoints", 1)
-
 				if err := c.checkpoint.Set(shardID, lastSeqNum); err != nil {
 					c.logger.Printf("set checkpoint error: %v", err)
 				}
+				c.counter.Add("checkpoints", 1)
 			}
 
 			if resp.NextShardIterator == nil || shardIterator == resp.NextShardIterator {