dotweb_sysgroup.go

package dotweb

import (
	"fmt"
	"github.com/devfeel/dotweb/core"
	jsonutil "github.com/devfeel/dotweb/framework/json"
	"runtime"
	"runtime/debug"
	"runtime/pprof"
	"strings"
)

// initDotwebGroup init Dotweb route group which start with /dotweb/
func initDotwebGroup(server *HttpServer) {
	gInner := server.Group("/dotweb")
	gInner.GET("/debug/pprof/:key", showPProf)
	gInner.GET("/debug/freemem", freeMemory)
	gInner.GET("/state", showServerState)
	gInner.GET("/state/interval", showIntervalData)
	gInner.GET("/query/:key", showQuery)
	gInner.GET("/routers", showRouters)
}

// query pprof debug info
// key:heap goroutine threadcreate block
func showPProf(ctx Context) error {
	querykey := ctx.GetRouterName("key")
	runtime.GC()
	return pprof.Lookup(querykey).WriteTo(ctx.Response().Writer(), 1)
}

func freeMemory(ctx Context) error {
	debug.FreeOSMemory()
	return nil
}

func showIntervalData(ctx Context) error {
	if ctx.Request().ExistsQueryKey("pretty") {
		return showIntervalDataPretty(ctx)
	} else {
		return showIntervalDataJson(ctx)
	}
}

func showIntervalDataJson(ctx Context) error {
	type data struct {
		Time         string
		RequestCount uint64
		ErrorCount   uint64
	}
	queryKey := ctx.QueryString("querykey")

	d := new(data)
	d.Time = queryKey
	d.RequestCount = ctx.HttpServer().StateInfo().QueryIntervalRequestData(queryKey)
	d.ErrorCount = ctx.HttpServer().StateInfo().QueryIntervalErrorData(queryKey)
	return ctx.WriteJson(d)
}

func showIntervalDataPretty(ctx Context) error {
	type data struct {
		Time         string
		RequestCount uint64
		ErrorCount   uint64
	}
	queryKey := ctx.QueryString("querykey")
	d := new(data)
	d.Time = queryKey
	d.RequestCount = ctx.HttpServer().StateInfo().QueryIntervalRequestData(queryKey)
	d.ErrorCount = ctx.HttpServer().StateInfo().QueryIntervalErrorData(queryKey)
	tableData := "<tr><td>" + d.Time + "</td><td>" + fmt.Sprint(d.RequestCount) + "</td><td>" + fmt.Sprint(d.ErrorCount) + "</td></tr>"
	col := `<colgroup>
		  <col width="40%">
		  <col width="30%">
          <col width="30%">
		</colgroup>`
	header := `<tr>
          <th>Time</th>
          <th>RequestCount</th>
          <th>ErrorCount</th>
        </tr>`
	html := core.CreateTableHtml(col, "IntervalData", header, tableData)
	return ctx.WriteHtml(html)
}

// snow server status
func showServerState(ctx Context) error {
	return ctx.WriteHtml(ctx.HttpServer().StateInfo().ShowHtmlTableData(Version, ctx.HttpServer().DotApp.GlobalUniqueID()))
}

// query server information
func showQuery(ctx Context) error {
	querykey := ctx.GetRouterName("key")
	switch querykey {
	case "state":
		return ctx.WriteString(jsonutil.GetJsonString(ctx.HttpServer().StateInfo()))
	case "":
		return ctx.WriteString("please input key")
	default:
		return ctx.WriteString("not support key => " + querykey)
	}
}

func showRouters(ctx Context) error {
	data := ""
	routerCount := len(ctx.HttpServer().router.GetAllRouterExpress())
	for k, _ := range ctx.HttpServer().router.GetAllRouterExpress() {
		method := strings.Split(k, routerExpressSplit)[0]
		router := strings.Split(k, routerExpressSplit)[1]
		data += "<tr><td>" + method + "</td><td>" + router + "</td></tr>"
	}
	col := `<colgroup>
		  <col width="40%">
		  <col width="60%">
		</colgroup>`
	header := `<tr>
          <th>Method</th>
          <th>Router</th>
        </tr>`
	html := core.CreateTableHtml(col, "Routers:"+fmt.Sprint(routerCount), header, data)

	return ctx.WriteHtml(html)
}