statistics HTML page for rpc

R=rsc DELTA=121 (115 added, 0 deleted, 6 changed) OCL=32427 CL=32429
2025-12-08 06:10:04 +00:00 · 2009-07-29 13:26:49 -07:00 · 2009-07-29 13:26:49 -07:00 · 75e0569b99
commit 75e0569b99
parent 9271c6402e
3 changed files with 121 additions and 5 deletions
--- a/src/pkg/rpc/debug.go
+++ b/src/pkg/rpc/debug.go
@ -0,0 +1,100 @@
+// Copyright 2009 The Go Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+
+package rpc
+
+/*
+	Some HTML presented at http://machine:port/debug/rpc
+	Lists services, their methods, and some statistics, still rudimentary.
+*/
+
+import (
+	"fmt";
+	"gob";
+	"http";
+	"io";
+	"log";
+	"os";
+	"rpc";
+	"sort";
+	"template";
+)
+
+const debugText =
+	`<html>
+	<body>
+	<title>Services</title>
+	{.repeated section @}
+	<hr>
+	Service {name}
+	<hr>
+		<table>
+		<th align=center>Method</th><th align=center>Calls</th>
+		{.repeated section meth}
+			<tr>
+			<td align=left font=fixed>{name}({.section m}{argType}, {replyType}) os.Error</td>
+			<td align=center>{numCalls}</td>{.end}
+			</tr>
+		{.end}
+		</table>
+	{.end}
+	</body>
+	</html>`
+
+var debug *template.Template
+
+type debugMethod struct {
+	m	*methodType;
+	name	string;
+}
+
+type methodArray []debugMethod
+
+type debugService struct {
+	s	*service;
+	name	string;
+	meth	methodArray;
+}
+
+type serviceArray []debugService
+
+func (s serviceArray) Len() int { return len(s) }
+func (s serviceArray) Less(i, j int) bool { return s[i].name < s[j].name }
+func (s serviceArray) Swap(i, j int) { s[i], s[j] = s[j], s[i] }
+
+func (m methodArray) Len() int { return len(m) }
+func (m methodArray) Less(i, j int) bool { return m[i].name < m[j].name }
+func (m methodArray) Swap(i, j int) { m[i], m[j] = m[j], m[i] }
+
+// Runs at /debug/rpc
+func debugHTTP(c *http.Conn, req *http.Request) {
+	var err os.Error;
+	if debug == nil {
+		debug, err = template.Parse(debugText, nil);
+		if err != nil {
+			fmt.Fprintln(c, "rpc can't create debug HTML template:", err.String());
+			return;
+		}
+	}
+	// Build a sorted version of the data.
+	var services = make(serviceArray, len(server.serviceMap));
+	i := 0;
+	server.Lock();
+	for sname, service := range server.serviceMap {
+		services[i] = debugService{service, sname, make(methodArray, len(service.method))};
+		j := 0;
+		for mname, method := range service.method {
+			services[i].meth[j] = debugMethod{method, mname};
+			j++;
+		}
+		sort.Sort(services[i].meth);
+		i++;
+	}
+	server.Unlock();
+	sort.Sort(services);
+	err = debug.Execute(services, c);
+	if err != nil {
+		fmt.Fprintln(c, "rpc: error executing template:", err.String());
+	}
+}