fixes (#2)

Exafunction · Jun 12, 2024 · 69589db · 69589db
1 parent 3143acb
commit 69589db
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/interceptor.go b/interceptor.go
@@ -72,12 +72,12 @@ func (i *Interceptor) WrapUnary(next connect.UnaryFunc) connect.UnaryFunc {
 
 		resp, err := next(ctx, req)
 		code = codeOf(err)
-		if err == nil {
+		if err == nil && reporter != nil {
 			var bytes *prom.CounterVec
 			if reporter.isClient {
-				bytes = reporter.bytesSent
-			} else {
 				bytes = reporter.bytesReceived
+			} else {
+				bytes = reporter.bytesSent
 			}
 			if bytes != nil {
 				bytes.WithLabelValues(callType, callPackage, callMethod).Add(float64(proto.Size(resp.Any().(proto.Message))))

diff --git a/metrics.go b/metrics.go
@@ -205,11 +205,11 @@ type Metrics struct {
 func (m *Metrics) Reset() {
 	m.requestStarted.Reset()
 	m.requestHandled.Reset()
-	m.streamMsgSent.Reset()
-	m.streamMsgReceived.Reset()
 	if m.requestHandledSeconds != nil {
 		m.requestHandledSeconds.Reset()
 	}
+	m.streamMsgSent.Reset()
+	m.streamMsgReceived.Reset()
 	if m.bytesSent != nil {
 		m.bytesSent.Reset()
 	}