enhance: Support to trace restful request and request error (#28685)

issue: #28348 Signed-off-by: SimFG <[email protected]>
milvus-io · Nov 27, 2023 · 9c46788 · 9c46788
1 parent eaabe02
commit 9c46788
Show file tree

Hide file tree

Showing 12 changed files with 509 additions and 207 deletions.
diff --git a/internal/distributed/proxy/httpserver/handler.go b/internal/distributed/proxy/httpserver/handler.go
@@ -1,6 +1,7 @@
 package httpserver
 
 import (
+	"context"
 	"fmt"
 
 	"github.com/gin-gonic/gin"
@@ -10,9 +11,12 @@ import (
 	"github.com/milvus-io/milvus/internal/types"
 )
 
+type RestRequestInterceptor func(ctx context.Context, ginCtx *gin.Context, req any, handler func(reqCtx context.Context, req any) (any, error)) (any, error)
+
 // Handlers handles http requests
 type Handlers struct {
-	proxy types.ProxyComponent
+	proxy        types.ProxyComponent
+	interceptors []RestRequestInterceptor
 }
 
 // NewHandlers creates a new Handlers