feat(httpc): added MetricsInterceptor in DoRequest

zeromicro · Aug 6, 2024 · f1d8a19 · f1d8a19
1 parent 4196ddb
commit f1d8a19
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/rest/httpc/requests.go b/rest/httpc/requests.go
@@ -39,7 +39,11 @@ func Do(ctx context.Context, method, url string, data any) (*http.Response, erro
 
 // DoRequest sends an HTTP request and returns an HTTP response.
 func DoRequest(r *http.Request) (*http.Response, error) {
-	return request(r, defaultClient{})
+	interceptor := internal.MetricsInterceptor("", nil)
+	r, handler := interceptor(r)
+	resp, err := request(r, defaultClient{})
+	handler(resp, err)
+	return resp, err
 }
 
 type (

diff --git a/rest/httpc/service.go b/rest/httpc/service.go
@@ -5,6 +5,7 @@ import (
 	"net/http"
 
 	"github.com/zeromicro/go-zero/core/breaker"
+	"github.com/zeromicro/go-zero/rest/httpc/internal"
 )
 
 type (
@@ -54,7 +55,11 @@ func (s namedService) Do(ctx context.Context, method, url string, data any) (*ht
 
 // DoRequest sends an HTTP request to the service.
 func (s namedService) DoRequest(r *http.Request) (*http.Response, error) {
-	return request(r, s)
+	interceptor := internal.MetricsInterceptor(s.name, nil)
+	r, handler := interceptor(r)
+	resp, err := request(r, s)
+	handler(resp, err)
+	return resp, err
 }
 
 func (s namedService) do(r *http.Request) (resp *http.Response, err error) {