【golang-技巧】- pprof 添加开关

1.背景

由于直接引入 pprof 依赖 那么会导致 默认开启，且绑定默认listen端口

这样做 不安全，同时有一些 代码扫描 通过不了

2.代码

通过实现自定以 server 实现绑定 端口，不通过默认引入的方法 写到默认的 http server中

func AsyncOpenPprof(globalConfig *config.GlobalConfig, logger *component.TelemetryManager) {
	if globalConfig.Pprof != nil && globalConfig.Pprof.Enable {
		mux := http.NewServeMux()
		mux.HandleFunc("/debug/pprof/", pprof.Index)
		mux.HandleFunc("/debug/pprof/cmdline", pprof.Cmdline)
		mux.HandleFunc("/debug/pprof/profile", pprof.Profile)
		mux.HandleFunc("/debug/pprof/symbol", pprof.Symbol)
		mux.HandleFunc("/debug/pprof/trace", pprof.Trace)
		err := http.ListenAndServe(":"+globalConfig.Pprof.Port, mux)
		logger.Logger.Info("pprof  enable port :" + globalConfig.Pprof.Port)
		if err != nil {
			return
		}
	} else {
		logger.Logger.Info("pprof  disable")
	}
}

3.结论

通过这样实现 可以通过开关开启 pprof

实现规避性能影响 问题 ，代码安全扫描问题