update metrics feature

Author: JacksonMei

api-service/controller/mcp_proxy.go

@@ -61,7 +61,7 @@ func NewMCPGateway(router *gin.RouterGroup) *MCPGateway {
 	gateway := &MCPGateway{
 		router: router,
 		transport: &http.Transport{
-			MaxIdleConns:        100,
+			MaxIdleConns:        2000,
 			MaxIdleConnsPerHost: 10,
 			IdleConnTimeout:     90 * time.Second,
 		},

api-service/middleware/metrics.go

@@ -90,24 +90,33 @@ func MCPMetricsMiddleware() gin.HandlerFunc {
 
 		var rpcMethod string
 		if c.Request.Method == "POST" && c.Request.Body != nil {
-			// LimitReader: only read up to 8KB for JSON-RPC header extraction
-			limited := io.LimitReader(c.Request.Body, 8192)
+			// Read up to 8KB+1 to detect oversized bodies; JSON-RPC method
+			// field is always in the first few dozen bytes.
+			const maxPeek = 8192
+			limited := io.LimitReader(c.Request.Body, maxPeek+1)
 			if body, err := io.ReadAll(limited); err == nil && len(body) > 0 {
 				var rpc struct {
 					Method string `json:"method"`
 				}
 				if json.Unmarshal(body, &rpc) == nil {
 					rpcMethod = rpc.Method
 				}
-				// Rebuild full body: read portion + any unread remainder
-				remaining, _ := io.ReadAll(c.Request.Body)
-				fullBody := append(body, remaining...)
+
+				// Fast path (99%+ requests): body fits within limit, skip
+				// second ReadAll and append entirely.
+				var fullBody []byte
+				if len(body) <= maxPeek {
+					fullBody = body
+				} else {
+					remaining, _ := io.ReadAll(c.Request.Body)
+					fullBody = append(body, remaining...)
+				}
+
 				c.Request.Body = io.NopCloser(bytes.NewReader(fullBody))
 				c.Request.GetBody = func() (io.ReadCloser, error) {
 					return io.NopCloser(bytes.NewReader(fullBody)), nil
 				}
 				c.Request.ContentLength = int64(len(fullBody))
-				// Store body in context for LoggingMiddleware reuse
 				c.Set("_req_body", fullBody)
 			}
 		}

go.work.sum

@@ -301,6 +301,7 @@ github.com/klauspost/cpuid/v2 v2.2.7/go.mod h1:Lcz8mBdAVJIBVzewtcLocK12l3Y+JytZY
 github.com/konsorten/go-windows-terminal-sequences v1.0.3 h1:CE8S1cTafDpPvMhIxNJKvHsGVBgn1xWYf1NbHQhywc8=
 github.com/kr/logfmt v0.0.0-20140226030751-b84e30acd515 h1:T+h1c/A9Gawja4Y9mFVWj2vyii2bbUNDw3kt9VxK2EY=
 github.com/kr/pretty v0.2.0 h1:s5hAObm+yFO5uHYt5dYjxi2rXrsnmRpJx4OYvIWUaQs=
+github.com/kr/pretty v0.3.0 h1:WgNl7dwNpEZ6jJ9k1snq4pZsg7DOEN8hP9Xw0Tsjwk0=
 github.com/kr/pty v1.1.1 h1:VkoXIwSboBpnk99O/KFauAEILuNHv5DVFKZMBN/gUgw=
 github.com/lyft/protoc-gen-star/v2 v2.0.3/go.mod h1:amey7yeodaJhXSbf/TlLvWiqQfLOSpEk//mLlc+axEk=
 github.com/minio/asm2plan9s v0.0.0-20200509001527-cdd76441f9d8/go.mod h1:mC1jAcsrzbxHt8iiaC+zU4b1ylILSosueou12R++wfY=
@@ -318,9 +319,11 @@ github.com/pelletier/go-toml/v2 v2.2.2/go.mod h1:1t835xjRzz80PqgE6HHgN2JOsmgYu/h
 github.com/peterbourgon/diskv v2.0.1+incompatible h1:UBdAOUP5p4RWqPBg048CAvpKN+vxiaj6gdUUzhl4XmI=
 github.com/peterbourgon/diskv v2.0.1+incompatible/go.mod h1:uqqh8zWWbv1HBMNONnaR/tNboyR3/BZd58JJSHlUSCU=
 github.com/pierrec/lz4/v4 v4.1.15/go.mod h1:gZWDp/Ze/IJXGXf23ltt2EXimqmTUXEy0GFuRQyBid4=
+github.com/pkg/diff v0.0.0-20210226163009-20ebb0f2a09e h1:aoZm08cpOy4WuID//EZDgcC4zIxODThtZNPirFr42+A=
 github.com/prometheus/client_model v0.3.0/go.mod h1:LDGWKZIo7rky3hgvBe+caln+Dr3dPggB5dvjtD7w9+w=
 github.com/rogpeppe/fastuuid v1.2.0/go.mod h1:jVj6XXZzXRy/MSR5jhDC/2q6DgLz+nrA6LYCDYWNEvQ=
 github.com/rogpeppe/go-internal v1.3.0 h1:RR9dF3JtopPvtkroDZuVD7qquD0bnHlKSqaQhgwt8yk=
+github.com/rogpeppe/go-internal v1.8.0 h1:FCbCCtXNOY3UtUuHUYaghJg4y7Fd14rXifAYUAtL9R8=
 github.com/soheilhy/cmux v0.1.5 h1:jjzc5WVemNEDTLwv9tlmemhC73tI08BNOIGwBOo10Js=
 github.com/soheilhy/cmux v0.1.5/go.mod h1:T7TcVDs9LWfQgPlPsdngu6I6QIoyIFZDDC6sNE1GqG0=
 github.com/spf13/afero v1.3.3/go.mod h1:5KUK8ByomD5Ti5Artl0RtHeI5pTF7MIDuXL3yY520V4=