fix grpc: write call statistics after run middlewares

commit_hash:024b0b11851daf92ff459d22c21331dd9e0e27f1

Author: kpavlov00

grpc/include/userver/ugrpc/client/impl/async_stream_methods.hpp

@@ -29,9 +29,9 @@ void CheckOk(
 
 void CheckFinishStatus(CallState& state);
 
-void ProcessFinish(CallState& state, const google::protobuf::Message* final_response);
+void ProcessFinish(CallState& state, const grpc::Status& status, const google::protobuf::Message* final_response);
 
-void ProcessFinishAbandoned(CallState& state) noexcept;
+void ProcessFinishAbandoned(CallState& state, const grpc::Status& status) noexcept;
 
 void ProcessCancelled(CallState& state, std::string_view stage) noexcept;
 
@@ -65,7 +65,7 @@ void Finish(
         case ugrpc::impl::AsyncMethodInvocation::WaitStatus::kOk:
             state.GetStatsScope().SetFinishTime(invocation.GetFinishTime());
             try {
-                ProcessFinish(state, final_response);
+                ProcessFinish(state, state.GetStatus(), final_response);
             } catch (const std::exception& ex) {
                 if (throw_on_error) {
                     throw;
@@ -117,7 +117,7 @@ void FinishAbandoned(GrpcStream& stream, StreamingCallState& state) noexcept try
     state.GetStatsScope().SetFinishTime(invocation.GetFinishTime());
 
     if (ok) {
-        ProcessFinishAbandoned(state);
+        ProcessFinishAbandoned(state, state.GetStatus());
     } else {
         ProcessNetworkError(state, "Finish");
     }

grpc/src/ugrpc/client/impl/async_stream_methods.cpp

@@ -70,17 +70,18 @@ void CheckFinishStatus(CallState& state) {
     }
 }
 
-void ProcessFinish(CallState& state, const google::protobuf::Message* final_response) {
-    const auto& status = state.GetStatus();
+void ProcessFinish(CallState& state, const grpc::Status& status, const google::protobuf::Message* response) {
+    RunMiddlewarePipeline(state, MiddlewareHooks::FinishHooks(status, status.ok() ? response : nullptr));
 
     HandleCallStatistics(state, status);
 
-    RunMiddlewarePipeline(state, MiddlewareHooks::FinishHooks(status, status.ok() ? final_response : nullptr));
-
     SetStatusAndResetSpan(state, status);
 }
 
-void ProcessFinishAbandoned(CallState& state) noexcept { SetStatusAndResetSpan(state, state.GetStatus()); }
+void ProcessFinishAbandoned(CallState& state, const grpc::Status& status) noexcept {
+    // Nothing to do with statistics, `RpcStatisticsScope` automatically accounts "abandoned-error"
+    SetStatusAndResetSpan(state, status);
+}
 
 void ProcessCancelled(CallState& state, std::string_view stage) noexcept {
     state.GetStatsScope().OnCancelled();