Added some comments about message state

mahendra · mahendra · commit 99da57f98a65 · 2013-06-25T17:06:16.000+05:30
diff --git a/kafka/consumer.py b/kafka/consumer.py
@@ -365,6 +365,13 @@ def __iter_partition__(self, partition, offset):
             next_offset = None
             for message in resp.messages:
                 next_offset = message.offset
+
+                # update the offset before the message is yielded. This is
+                # so that the consumer state is not lost in certain cases.
+                # For eg: the message is yielded and consumed by the caller,
+                # but the caller does not come back into the generator again.
+                # The message will be consumed but the status will not be
+                # updated in the consumer
                 self.offsets[partition] = message.offset
                 yield message
             if next_offset is None:
