Merge pull request #2 from skit-ai/custom-labels

Fix for requests started metric not reporting custom labels
skit-ai · Jul 5, 2023 · e2976f6 · e2976f6
2 parents 6ecd634 + a879144
commit e2976f6
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/server_reporter.go b/server_reporter.go
@@ -17,22 +17,27 @@ type serverReporter struct {
 	startTime   time.Time
 	clientUuid  string
 	flowUuid    string
+	first       bool
 }
 
 func newServerReporter(m *ServerMetrics, rpcType grpcType, fullMethod string) *serverReporter {
 	r := &serverReporter{
 		metrics: m,
 		rpcType: rpcType,
+		first:   true,
 	}
 	if r.metrics.serverHandledHistogramEnabled {
 		r.startTime = time.Now()
 	}
 	r.serviceName, r.methodName = splitMethodName(fullMethod)
-	r.metrics.serverStartedCounter.WithLabelValues(string(r.rpcType), r.serviceName, r.methodName, r.clientUuid, r.flowUuid).Inc()
 	return r
 }
 
 func (r *serverReporter) ReceivedMessage() {
+	if r.first {
+		r.metrics.serverStartedCounter.WithLabelValues(string(r.rpcType), r.serviceName, r.methodName, r.clientUuid, r.flowUuid).Inc()
+		r.first = false
+	}
 	r.metrics.serverStreamMsgReceived.WithLabelValues(string(r.rpcType), r.serviceName, r.methodName, r.clientUuid, r.flowUuid).Inc()
 	if r.metrics.serverHandledHistogramEnabled {
 		r.startTime = time.Now()