Improve shuffle write time metrics under RSS

**Is your feature request related to a problem? Please describe.**
currently the shuffle write time does not count `RssPartitionWriterBase.close()`, leading incorrect time displayed in UI (smaller than actual).

**Describe the solution you'd like**
count `RssPartitionWriterBase.close()` in shuffle writer.

**Describe alternatives you've considered**

**Additional context**
