Can we just count the size when iterator.close is called (or the method in context)?