From 1e10ccb1cbaa670ae9a4c15ce15b50979e0510fa Mon Sep 17 00:00:00 2001
From: Nick Jiang <njiang747@gmail.com>
Date: Tue, 11 Jun 2024 21:11:23 -0700
Subject: [PATCH] fixes

---
 interceptor.go | 6 +++---
 metrics.go     | 4 ++--
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/interceptor.go b/interceptor.go
index b77a74a..ed30953 100644
--- 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
index ad4037a..0c479f8 100644
--- 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()
 	}