Augmented updateStateByKey API

Augment updateStateByKey API by exposing both key and current batch timestamp to applications.