contrib/dimfeld/httptreemux.v5: only replace at end of path

DataDog · Sep 19, 2024 · c5d08e6 · c5d08e6
1 parent 1dac691
commit c5d08e6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/contrib/dimfeld/httptreemux.v5/httptreemux.go b/contrib/dimfeld/httptreemux.v5/httptreemux.go
@@ -145,7 +145,10 @@ func getRoute(router *httptreemux.TreeMux, w http.ResponseWriter, req *http.Requ
 		// replace parameter at end of the path, i.e. "../:param"
 		oldP = "/" + v
 		newP = "/:" + k
-		route = strings.Replace(route, oldP, newP, 1)
+		if strings.HasSuffix(route, oldP) {
+			endPos := strings.LastIndex(route, oldP)
+			route = route[:endPos] + newP
+		}
 	}
 	return route, true
 }