MLHR-1416 

Provide api to track the kafka consumer offset and fix offset reset bug in SimpleKafkaConsumer
When exactly will this be called? Will the map contain all current partitions?