perf: change http clone cache key

Aur0ra-m · Apr 9, 2023 · ba721bc · ba721bc
1 parent fdccd9b
commit ba721bc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/core/ahttp/ahttp.go b/core/ahttp/ahttp.go
@@ -13,7 +13,7 @@ import (
 )
 
 type requestCacheBlock struct {
-	key   string //method-domain-url
+	key   *http.Request //method-domain-url
 	value string
 }
 
@@ -88,7 +88,10 @@ func RequestClone(src *http.Request) *http.Request {
 	// dump request
 	reqStr := ""
 	for _, c := range cache {
-		if c != nil && c.key == src.URL.String() {
+		if c == nil {
+			break
+		}
+		if c.key == src {
 			reqStr = c.value
 			break
 		}
@@ -99,7 +102,7 @@ func RequestClone(src *http.Request) *http.Request {
 		if cache[updatePoint] == nil {
 			cache[updatePoint] = &requestCacheBlock{}
 		}
-		cache[updatePoint].key = src.URL.String()
+		cache[updatePoint].key = src
 		cache[updatePoint].value = reqStr
 		updatePoint = (updatePoint + 1) % 128
 	}

diff --git a/core/ahttp/ahttpModify.go b/core/ahttp/ahttpModify.go
@@ -94,6 +94,8 @@ func ModifyURLPathMidPad(Url *url.URL, padding string) {
 	if strings.HasSuffix(Url.Path, "/") {
 		srcPath = strings.TrimRight(srcPath, "/")
 	}
+	// trim the first separator
+	srcPath = strings.TrimLeft(srcPath, "/")
 
 	// split path into fragments
 	splits := strings.Split(srcPath, "/")
@@ -108,7 +110,7 @@ func ModifyURLPathMidPad(Url *url.URL, padding string) {
 
 		newPath = firstPart + "/" + padding + "/" + thirdPart
 	} else {
-		newPath = padding + "/" + firstPart
+		newPath = "/" + padding + "/" + firstPart
 	}
 	Url.Path = newPath
 }