tidb http_handler 源码
tidb http_handler 代码
文件路径：/dumpling/export/http_handler.go
// Copyright 2020 PingCAP, Inc. Licensed under Apache-2.0.
package export
import (
"net"
"net/http"
"net/http/pprof"
"strings"
"time"
"github.com/pingcap/errors"
tcontext "github.com/pingcap/tidb/dumpling/context"
"github.com/pingcap/tidb/dumpling/log"
"github.com/prometheus/client_golang/prometheus/promhttp"
"github.com/soheilhy/cmux"
)
var cmuxReadTimeout = 10 * time.Second
func startHTTPServer(tctx *tcontext.Context, lis net.Listener) {
router := http.NewServeMux()
router.Handle("/metrics", promhttp.Handler())
router.HandleFunc("/debug/pprof/", pprof.Index)
router.HandleFunc("/debug/pprof/cmdline", pprof.Cmdline)
router.HandleFunc("/debug/pprof/profile", pprof.Profile)
router.HandleFunc("/debug/pprof/symbol", pprof.Symbol)
router.HandleFunc("/debug/pprof/trace", pprof.Trace)
httpServer := &http.Server{
Handler: router,
}
err := httpServer.Serve(lis)
err = errors.Cause(err)
if err != nil && !isErrNetClosing(err) && err != http.ErrServerClosed {
tctx.L().Info("dumpling http handler return with error", log.ShortError(err))
}
}
func startDumplingService(tctx *tcontext.Context, addr string) error {
rootLis, err := net.Listen("tcp", addr)
if err != nil {
return errors.Annotate(err, "start listening")
}
// create a cmux
m := cmux.New(rootLis)
m.SetReadTimeout(cmuxReadTimeout) // set a timeout, ref: https://github.com/pingcap/tidb-binlog/pull/352
httpL := m.Match(cmux.HTTP1Fast())
go startHTTPServer(tctx, httpL)
err = m.Serve() // start serving, block
if err != nil && isErrNetClosing(err) {
err = nil
}
return err
}
var useOfClosedErrMsg = "use of closed network connection"
// isErrNetClosing checks whether is an ErrNetClosing error
func isErrNetClosing(err error) bool {
if err == nil {
return false
}
return strings.Contains(err.Error(), useOfClosedErrMsg)
}
相关信息
相关文章
0
赞
热门推荐
-
2、 - 优质文章
-
3、 gate.io
-
8、 golang
-
9、 openharmony
-
10、 Vue中input框自动聚焦