[GLUTEN-5731][CORE] Fix the logic to calculate shuffle write time in …

…RssPartitionWriter (#5742)

Co-authored-by: yibing <[email protected]>

cpp/core/shuffle/rss/RssPartitionWriter.cc

@@ -56,8 +56,6 @@ arrow::Status RssPartitionWriter::evict(
     bool reuseBuffers,
     bool hasComplexType) {
   rawPartitionLengths_[partitionId] += inMemoryPayload->getBufferSize();
-
-  ScopedTimer timer(&spillTime_);
   auto payloadType = (codec_ && inMemoryPayload->numRows() >= options_.compressionThreshold)
       ? Payload::Type::kCompressed
       : Payload::Type::kUncompressed;
@@ -69,6 +67,7 @@ arrow::Status RssPartitionWriter::evict(
   payload = nullptr; // Invalidate payload immediately.
 
   // Push.
+  ScopedTimer timer(&spillTime_);
   ARROW_ASSIGN_OR_RAISE(auto buffer, rssBufferOs->Finish());
   bytesEvicted_[partitionId] += rssClient_->pushPartitionData(
       partitionId, reinterpret_cast<char*>(const_cast<uint8_t*>(buffer->data())), buffer->size());