How can I clear streaming DataSet records after foreach together with flatMapGroupsWithState?

Question

I\'m using flatMapGroupsWithState to collect messages streamed from kafka based on a time window key (timeBucket), which are then returned from the group state update function (