Merge pull request #224 from intelops/pprof

added liveness and pprof

Author: vijeyash1

agent/kubviz/k8smetrics_agent.go

@@ -29,6 +29,7 @@ import (
 	_ "k8s.io/client-go/plugin/pkg/client/auth/oidc"
 
 	//  _ "k8s.io/client-go/plugin/pkg/client/auth/openstack"
+	"github.com/intelops/kubviz/agent/server"
 	"k8s.io/client-go/tools/cache"
 	"k8s.io/client-go/tools/clientcmd"
 )
@@ -104,7 +105,7 @@ func main() {
 		clientset = getK8sClient(config)
 	}
 	go publishMetrics(clientset, js, clusterMetricsChan)
-
+	go server.StartServer()
 	collectAndPublishMetrics := func() {
 		err := outDatedImages(config, js)
 		LogErr(err)

agent/server/server.go

@@ -0,0 +1,37 @@
+package server
+
+import (
+	"log"
+	"net/http"
+	_ "net/http/pprof"
+
+	"github.com/gin-gonic/gin"
+)
+
+func EnableProfile(r *gin.Engine) {
+	pprofGroup := r.Group("/debug/pprof")
+	{
+		pprofGroup.GET("/", gin.WrapH(http.DefaultServeMux))
+		pprofGroup.GET("/cmdline", gin.WrapH(http.DefaultServeMux))
+		pprofGroup.GET("/profile", gin.WrapH(http.DefaultServeMux))
+		pprofGroup.POST("/symbol", gin.WrapH(http.DefaultServeMux))
+		pprofGroup.GET("/symbol", gin.WrapH(http.DefaultServeMux))
+		pprofGroup.GET("/trace", gin.WrapH(http.DefaultServeMux))
+		pprofGroup.GET("/allocs", gin.WrapH(http.DefaultServeMux))
+		pprofGroup.GET("/block", gin.WrapH(http.DefaultServeMux))
+		pprofGroup.GET("/goroutine", gin.WrapH(http.DefaultServeMux))
+		pprofGroup.GET("/heap", gin.WrapH(http.DefaultServeMux))
+		pprofGroup.GET("/mutex", gin.WrapH(http.DefaultServeMux))
+		pprofGroup.GET("/threadcreate", gin.WrapH(http.DefaultServeMux))
+	}
+
+	r.GET("/liveness", func(c *gin.Context) {
+		c.String(http.StatusOK, "Alive")
+	})
+}
+
+func StartServer() {
+	r := gin.Default()
+	EnableProfile(r)
+	log.Fatal(r.Run(":8080"))
+}