oxia: add write latency for writeStream (#507)

streamnative · Aug 28, 2024 · a186246 · a186246
1 parent b0988d2
commit a186246
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/server/leader_controller.go b/server/leader_controller.go
@@ -873,11 +873,13 @@ func (lc *leaderController) handleWriteStream(stream proto.OxiaClient_WriteStrea
 			return
 		}
 
+		timer := lc.writeLatencyHisto.Timer()
 		slog.Debug("Got request in stream",
 			slog.Any("req", req))
 
 		offset, timestamp, err1 := lc.appendToWalStreamRequest(stream.Context(), req)
 		if err1 != nil {
+			timer.Done()
 			closeCh <- err1
 			return
 		}
@@ -886,14 +888,17 @@ func (lc *leaderController) handleWriteStream(stream proto.OxiaClient_WriteStrea
 			return lc.db.ProcessWrite(req, offset, timestamp, SessionUpdateOperationCallback)
 		})
 		if err2 != nil {
+			timer.Done()
 			closeCh <- err2
 			return
 		}
 
 		if err3 := stream.Send(resp); err3 != nil {
+			timer.Done()
 			closeCh <- err3
 			return
 		}
+		timer.Done()
 	}
 }