feat: add graphical visualization

Author: davidlhw

main.go

@@ -7,6 +7,8 @@ import (
 	"runtime/debug"
 	"strconv"
 
+	"github.com/arl/statsviz"
+	example "github.com/arl/statsviz/_example"
 	"github.com/docker/go-units"
 	"github.com/gin-gonic/gin"
 )
@@ -22,10 +24,14 @@ func main() {
 }
 
 func SetUpRouter(withRoutes bool) *gin.Engine {
+	// Force the GC to work to make the plots "move".
+	go example.Work()
+
 	r := gin.New()
 	if withRoutes {
 		r.GET("/config", readConfigHandler)
 		r.POST("/config", updateConfigHandler)
+		r.GET("/debug/statsviz/*filepath", statsvizHandler)
 	}
 	return r
 }
@@ -75,3 +81,11 @@ func updateConfigHandler(c *gin.Context) {
 
 	c.IndentedJSON(http.StatusOK, req)
 }
+
+func statsvizHandler(c *gin.Context) {
+	if c.Param("filepath") == "/ws" {
+		statsviz.Ws(c.Writer, c.Request)
+		return
+	}
+	statsviz.IndexAtRoot("/debug/statsviz").ServeHTTP(c.Writer, c.Request)
+}

main_test.go

@@ -132,3 +132,22 @@ func TestUpdateConfigHandler(t *testing.T) {
 		t.Errorf("TestUpdateConfigHandler: want %q, got %q", test.want, got)
 	}
 }
+
+func TestStatsvizHandler(t *testing.T) {
+	r := SetUpRouter(false)
+	r.GET("/debug/statsviz/*action", statsvizHandler)
+
+	req, _ := http.NewRequest("GET", "/debug/statsviz/", nil)
+	w := httptest.NewRecorder()
+
+	r.ServeHTTP(w, req)
+
+	statusOK := w.Code == http.StatusOK
+
+	p := w.Body.Bytes()
+	pageOK := strings.Index(string(p), "<title>Statsviz</title>") > 0
+
+	if !(statusOK && pageOK) {
+		t.Errorf("TestStatsvizHandler: want %d, got %d", http.StatusOK, w.Code)
+	}
+}