Add support for zpages

cmd/census.go

@@ -1,8 +1,14 @@
 package cmd
 
 import (
+	"context"
+	"net/http"
+	"os"
+
 	"github.com/pkg/errors"
+	"github.com/virtual-kubelet/virtual-kubelet/log"
 	"go.opencensus.io/trace"
+	"go.opencensus.io/zpages"
 )
 
 var (
@@ -49,3 +55,14 @@ func AvailableTraceExporters() []string {
 	}
 	return out
 }
+
+func setupZpages() {
+	ctx := context.TODO()
+	p := os.Getenv("ZPAGES_PORT")
+	if p == "" {
+		log.G(ctx).Error("Missing ZPAGES_PORT env var, cannot setup zpages endpoint")
+	}
+	mux := http.NewServeMux()
+	zpages.Handle(mux, "/debug")
+	http.ListenAndServe(p, mux)
+}

cmd/root.go

@@ -275,6 +275,10 @@ func initConfig() {
 	userTraceConfig.Tags["provider"] = provider
 	userTraceConfig.Tags["nodeName"] = nodeName
 	for _, e := range userTraceExporters {
+		if e == "zpages" {
+			go setupZpages()
+			continue
+		}
 		exporter, err := GetTracingExporter(e, userTraceConfig)
 		if err != nil {
 			log.L.WithError(err).WithField("exporter", e).Fatal("Cannot initialize exporter")