From 82b531fbfbd83433a4fd158fe044b838b4709663 Mon Sep 17 00:00:00 2001 From: John Date: Wed, 14 Oct 2020 21:18:11 +0800 Subject: [PATCH] add domain support for domain of ghttp.Server --- .example/net/ghttp/server/pprof.go | 5 +++-- net/ghttp/ghttp_server_pprof.go | 9 +++++++-- 2 files changed, 10 insertions(+), 4 deletions(-) diff --git a/.example/net/ghttp/server/pprof.go b/.example/net/ghttp/server/pprof.go index fd16f2766..79f96ebdc 100644 --- a/.example/net/ghttp/server/pprof.go +++ b/.example/net/ghttp/server/pprof.go @@ -1,12 +1,13 @@ package main import ( + "github.com/gogf/gf/frame/g" "github.com/gogf/gf/net/ghttp" ) func main() { - s := ghttp.GetServer() - s.EnablePProf() + s := g.Server() + s.Domain("localhost").EnablePProf() s.BindHandler("/", func(r *ghttp.Request) { r.Response.Writeln("哈喽世界！") }) diff --git a/net/ghttp/ghttp_server_pprof.go b/net/ghttp/ghttp_server_pprof.go index b85d8a6cf..714e600f6 100644 --- a/net/ghttp/ghttp_server_pprof.go +++ b/net/ghttp/ghttp_server_pprof.go @@ -24,14 +24,19 @@ const ( // EnablePProf enables PProf feature for server. func (s *Server) EnablePProf(pattern ...string) { + s.Domain(gDEFAULT_DOMAIN).EnablePProf(pattern...) +} + +// EnablePProf enables PProf feature for server of specified domain. +func (d *Domain) EnablePProf(pattern ...string) { p := gDEFAULT_PPROF_PATTERN if len(pattern) > 0 && pattern[0] != "" { p = pattern[0] } up := &utilPProf{} - _, _, uri, _ := s.parsePattern(p) + _, _, uri, _ := d.server.parsePattern(p) uri = strings.TrimRight(uri, "/") - s.Group(uri, func(group *RouterGroup) { + d.Group(uri, func(group *RouterGroup) { group.ALL("/*action", up.Index) group.ALL("/cmdline", up.Cmdline) group.ALL("/profile", up.Profile)