1 files changed, 36 insertions, 0 deletions
diff --git a/core/setup/pprof.go b/core/setup/pprof.go
new file mode 100644
index 000000000..e202bfc00
--- /dev/null
+++ b/core/setup/pprof.go
@@ -0,0 +1,36 @@
+package setup
+
+import (
+	"sync"
+
+	"github.com/miekg/coredns/middleware"
+	"github.com/miekg/coredns/middleware/pprof"
+)
+
+var pprofOnce sync.Once
+
+// PProf returns a new instance of a pprof handler. It accepts no arguments or options.
+func PProf(c *Controller) (middleware.Middleware, error) {
+	found := false
+	for c.Next() {
+		if found {
+			return nil, c.Err("pprof can only be specified once")
+		}
+		if len(c.RemainingArgs()) != 0 {
+			return nil, c.ArgErr()
+		}
+		if c.NextBlock() {
+			return nil, c.ArgErr()
+		}
+		found = true
+	}
+	handler := &pprof.Handler{}
+	pprofOnce.Do(func() {
+		c.Startup = append(c.Startup, handler.Start)
+	})
+
+	return func(next middleware.Handler) middleware.Handler {
+		handler.Next = next
+		return handler
+	}, nil
+}