netmaker/controllers/debug.go

34 lines
710 B
Go
Raw Normal View History

2021-12-09 05:52:32 +08:00
//go:build debug
package controller
import (
"context"
"net/http"
_ "net/http/pprof"
"os"
"os/signal"
"github.com/gravitl/netmaker/logger"
)
func init() {
srv := &http.Server{Addr: "0.0.0.0:6060", Handler: nil}
go func() {
logger.Log(0, "Debug mode active")
err := srv.ListenAndServe()
if err != nil {
logger.Log(0, err.Error())
}
c := make(chan os.Signal)
// Relay os.Interrupt to our channel (os.Interrupt = CTRL+C)
// Ignore other incoming signals
signal.Notify(c, os.Interrupt)
// Block main routine until a signal is received
// As long as user doesn't press CTRL+C a message is not passed and our main routine keeps running
<-c
srv.Shutdown(context.TODO())
}()
}