master

分支 (1)

管理

管理

master

yasrpc
/
debug.go

package yasrpc

import (
	"fmt"
	"html/template"
	"net/http"
)

const debugText = `<html>
<body>
    <title>YasRPC Services</title>
    {{range .}}
    <hr>
    Service {{.Name}}
    <hr>
    <table>
        <th align=center>Method</th>
        <th align=center>Calls</th>
        {{range $name, $mtype := .Method}}
        <tr>
            <td align=left font=fixed>{{$name}}({{$mtype.ArgType}}, {{$mtype.ReplyType}}) error</td>
            <td align=center>{{$mtype.NumCalls}}</td>
        </tr>
        {{end}}
    </table>
    {{end}}
</body>
</html>
`

var debug = template.Must(template.New("RPC debug").Parse(debugText))

type debugHTTP struct {
	*Server
}

type debugService struct {
	Name   string
	Method map[string]*methodType
}

// Runs at /debug/yasrpc
func (s debugHTTP) ServeHTTP(w http.ResponseWriter, req *http.Request) {
	var services []debugService
	s.serviceMap.Range(func(namei, svci interface{}) bool {
		svc := svci.(*service)
		services = append(services, debugService{
			Name:   namei.(string),
			Method: svc.method,
		})
		return true
	})
	if err := debug.Execute(w, services); err != nil {
		_, _ = fmt.Fprintln(w, "rpc: error executing template:", err.Error())
	}
}

// Runs at /debug/yasrpc
func (s debugHTTP) HandleWeb(ctx *Context) {
	var services []debugService
	s.serviceMap.Range(func(namei, svci interface{}) bool {
		svc := svci.(*service)
		services = append(services, debugService{
			Name:   namei.(string),
			Method: svc.method,
		})
		return true
	})
	if err := debug.Execute(ctx.Writer, services); err != nil {
		_, _ = fmt.Fprintln(ctx.Writer, "rpc: error executing template:", err.Error())
	}
}