55package route
66
77import (
8+ "fmt"
9+ "log"
810 "net/http"
911 "net/http/pprof"
12+ "strings"
1013
1114 "github.com/gin-gonic/gin"
12- "github.com/sirupsen/logrus"
1315 "golang.design/x/ssaplayground/src/config"
1416)
1517
16- // Register routers
18+ // Register routers for the ssa service.
1719func Register () * gin.Engine {
18- r := & Router {
19- Engine : gin .Default (),
20- }
20+ r := & Router {Engine : gin .Default ()}
2121 r .SetupAPI ()
2222 r .SetupApp ()
23- if config .Get ().Mode != "debug" {
23+ if config .Get ().Mode != gin . DebugMode {
2424 return r .Engine
2525 }
2626 r .SetupProfile ()
2727 return r .Engine
2828}
2929
30- // Router .. .
30+ // Router is a router engine .
3131type Router struct {
3232 Engine * gin.Engine
3333}
3434
35- // SetupAPI .. .
35+ // SetupAPI serves the API endpoints of the gossa service .
3636func (r * Router ) SetupAPI () {
3737 v1 := r .Engine .Group ("/gossa/api/v1" )
3838 {
@@ -41,46 +41,54 @@ func (r *Router) SetupAPI() {
4141 }
4242}
4343
44- // SetupApp .. .
44+ // SetupApp serves the static website of Go SSA Playground .
4545func (r * Router ) SetupApp () {
4646 r .Engine .Use (static ("/gossa" ))
47- logrus . Infof ("GoSSAWeb is on: http://%s, static: %s" , config .Get ().Addr , config .Get ().Static )
47+ log . Printf ("GoSSAWeb is on: http://%s, static: %s" , config .Get ().Addr , config .Get ().Static )
4848}
4949
50- // SetupProfile profiles the standard HandlerFuncs from the net/http/pprof package with
50+ // SetupProfile the standard HandlerFuncs from the net/http/pprof package with
5151// the provided gin.Engine. prefixOptions is a optional. If not prefixOptions,
5252// the default path prefix is used, otherwise first prefixOptions will be path prefix.
5353//
5454// Basic Usage:
5555//
5656// - use the pprof tool to look at the heap profile:
57- // go tool pprof http://localhost:9999 /debug/pprof/heap
57+ // go tool pprof localhost:8080/midgard/api/v1 /debug/pprof/heap
5858// - look at a 30-second CPU profile:
59- // go tool pprof http://localhost:9999 /debug/pprof/profile
59+ // go tool pprof localhost:8080/midgard/api/v1 /debug/pprof/profile
6060// - look at the goroutine blocking profile, after calling runtime.SetBlockProfileRate:
61- // go tool pprof http://localhost:9999 /debug/pprof/block
61+ // go tool pprof localhost:8080/midgard/api/v1 /debug/pprof/block
6262// - collect a 5-second execution trace:
63- // wget http://localhost:9999 /debug/pprof/trace?seconds=5
63+ // go tool pprof localhost:8080/midgard/api/v1 /debug/pprof/trace?seconds=5
6464//
6565func (r * Router ) SetupProfile () {
6666 pprofHandler := func (h http.HandlerFunc ) gin.HandlerFunc {
6767 handler := http .HandlerFunc (h )
6868 return func (c * gin.Context ) {
69+
70+ fmt .Println (c .Request .Host )
71+ if ! strings .Contains (c .Request .Host , "localhost" ) {
72+ c .AbortWithStatus (http .StatusUnauthorized )
73+ return
74+ }
75+
6976 handler .ServeHTTP (c .Writer , c .Request )
7077 }
7178 }
72- prefixRouter := r .Engine .Group ("/debug/pprof" )
79+ rr := r .Engine .Group ("/debug/pprof" )
7380 {
74- prefixRouter .GET ("/" , pprofHandler (pprof .Index ))
75- prefixRouter .GET ("/cmdline" , pprofHandler (pprof .Cmdline ))
76- prefixRouter .GET ("/profile" , pprofHandler (pprof .Profile ))
77- prefixRouter .POST ("/symbol" , pprofHandler (pprof .Symbol ))
78- prefixRouter .GET ("/symbol" , pprofHandler (pprof .Symbol ))
79- prefixRouter .GET ("/trace" , pprofHandler (pprof .Trace ))
80- prefixRouter .GET ("/block" , pprofHandler (pprof .Handler ("block" ).ServeHTTP ))
81- prefixRouter .GET ("/goroutine" , pprofHandler (pprof .Handler ("goroutine" ).ServeHTTP ))
82- prefixRouter .GET ("/heap" , pprofHandler (pprof .Handler ("heap" ).ServeHTTP ))
83- prefixRouter .GET ("/mutex" , pprofHandler (pprof .Handler ("mutex" ).ServeHTTP ))
84- prefixRouter .GET ("/threadcreate" , pprofHandler (pprof .Handler ("threadcreate" ).ServeHTTP ))
81+ rr .GET ("/" , pprofHandler (pprof .Index ))
82+ rr .GET ("/cmdline" , pprofHandler (pprof .Cmdline ))
83+ rr .GET ("/profile" , pprofHandler (pprof .Profile ))
84+ rr .POST ("/symbol" , pprofHandler (pprof .Symbol ))
85+ rr .GET ("/symbol" , pprofHandler (pprof .Symbol ))
86+ rr .GET ("/trace" , pprofHandler (pprof .Trace ))
87+ rr .GET ("/allocs" , pprofHandler (pprof .Handler ("allocs" ).ServeHTTP ))
88+ rr .GET ("/block" , pprofHandler (pprof .Handler ("block" ).ServeHTTP ))
89+ rr .GET ("/goroutine" , pprofHandler (pprof .Handler ("goroutine" ).ServeHTTP ))
90+ rr .GET ("/heap" , pprofHandler (pprof .Handler ("heap" ).ServeHTTP ))
91+ rr .GET ("/mutex" , pprofHandler (pprof .Handler ("mutex" ).ServeHTTP ))
92+ rr .GET ("/threadcreate" , pprofHandler (pprof .Handler ("threadcreate" ).ServeHTTP ))
8593 }
8694}
0 commit comments