From b3e66ec4f7b342e28d9edf0dbaa7f0af864c4b18 Mon Sep 17 00:00:00 2001
From: ShuNing <nolouch@gmail.com>
Date: Thu, 26 Oct 2023 12:18:33 +0800
Subject: [PATCH] This is an automated cherry-pick of #7255

close tikv/pd#7253

Signed-off-by: ti-chi-bot <ti-community-prow-bot@tidb.io>
---
 server/util.go | 55 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 55 insertions(+)

diff --git a/server/util.go b/server/util.go
index 6a060779d70..f57dbfb9632 100644
--- a/server/util.go
+++ b/server/util.go
@@ -16,8 +16,13 @@ package server
 
 import (
 	"context"
+<<<<<<< HEAD
 	"fmt"
 	"math/rand"
+=======
+	"net/http"
+	"net/http/pprof"
+>>>>>>> bfc988bb3 (api: fix cannot dump trace (#7255))
 	"path/filepath"
 	"strings"
 	"time"
@@ -160,6 +165,56 @@ func checkBootstrapRequest(clusterID uint64, req *pdpb.BootstrapRequest) error {
 	return nil
 }
 
+<<<<<<< HEAD
+=======
+func combineBuilderServerHTTPService(ctx context.Context, svr *Server, serviceBuilders ...HandlerBuilder) (map[string]http.Handler, error) {
+	userHandlers := make(map[string]http.Handler)
+	registerMap := make(map[string]http.Handler)
+
+	apiService := negroni.New()
+	recovery := negroni.NewRecovery()
+	apiService.Use(recovery)
+	router := mux.NewRouter()
+
+	for _, build := range serviceBuilders {
+		handler, info, err := build(ctx, svr)
+		if err != nil {
+			return nil, err
+		}
+		if !info.IsCore && len(info.PathPrefix) == 0 && (len(info.Name) == 0 || len(info.Version) == 0) {
+			return nil, errs.ErrAPIInformationInvalid.FastGenByArgs(info.Name, info.Version)
+		}
+
+		if err := apiutil.RegisterUserDefinedHandlers(registerMap, &info, handler); err != nil {
+			return nil, err
+		}
+	}
+
+	// Combine the pd service to the router. the extension service will be added to the userHandlers.
+	for pathPrefix, handler := range registerMap {
+		if strings.Contains(pathPrefix, apiutil.CorePath) || strings.Contains(pathPrefix, apiutil.ExtensionsPath) {
+			router.PathPrefix(pathPrefix).Handler(handler)
+			if pathPrefix == apiutil.CorePath {
+				// Deprecated
+				router.Path("/pd/health").Handler(handler)
+				// Deprecated
+				router.Path("/pd/ping").Handler(handler)
+			}
+		} else {
+			userHandlers[pathPrefix] = handler
+		}
+	}
+
+	apiService.UseHandler(router)
+	userHandlers[pdAPIPrefix] = apiService
+
+	// fix issue https://github.com/tikv/pd/issues/7253
+	// FIXME: remove me after upgrade
+	userHandlers["/debug/pprof/trace"] = http.HandlerFunc(pprof.Trace)
+	return userHandlers, nil
+}
+
+>>>>>>> bfc988bb3 (api: fix cannot dump trace (#7255))
 func isPathInDirectory(path, directory string) bool {
 	absPath, err := filepath.Abs(path)
 	if err != nil {