main.go

package main

import (
	"fmt"
	"net/http"
	"net/http/pprof"
	"time"

	"github.com/arl/statsviz"
)

func Mux() *http.ServeMux {
	mux := http.NewServeMux()

	mux.HandleFunc("/debug/pprof/", pprof.Index)
	mux.HandleFunc("/debug/pprof/cmdline", pprof.Cmdline)
	mux.HandleFunc("/debug/pprof/profile", pprof.Profile)
	mux.HandleFunc("/debug/pprof/symbol", pprof.Symbol)
	mux.HandleFunc("/debug/pprof/trace", pprof.Trace)

	statsviz.Register(mux)

	// Liveness handler
	mux.HandleFunc("/liveness", func(w http.ResponseWriter, r *http.Request) {
		time.Sleep(time.Millisecond * 10)
		fmt.Fprintln(w, "OK")
	})

	return mux
}

func main() {
	if err := http.ListenAndServe("0.0.0.0:3010", Mux()); err != nil {
		fmt.Println("Server Error")
	}
}