update: 查看路由列表和请求响应

Author: luopengift

server.go

@@ -3,6 +3,7 @@ package requests
 import (
 	"context"
 	"fmt"
+	"io"
 	"net/http"
 	"net/http/pprof"
 	"net/url"
@@ -83,17 +84,17 @@ func (node *Node) paths() []string {
 }
 
 // Print print trie tree struct
-func (node *Node) Print() {
-	node.print(0)
+func (node *Node) Print(w io.Writer) {
+	node.print(0, w)
 }
 
-func (node *Node) print(m int) {
+func (node *Node) print(m int, w io.Writer) {
 	paths := node.paths()
 	for method, handler := range node.methods {
-		fmt.Printf("%spath=%s, method=%s, handler=%v, next=%#v\n", strings.Repeat("    ", m), node.path, method, handler, paths)
+		fmt.Fprintf(w, "%spath=%s, method=%s, handler=%v, next=%#v\n", strings.Repeat("    ", m), node.path, method, handler, paths)
 	}
 	for _, p := range paths {
-		node.next[p].print(m + 1)
+		node.next[p].print(m+1, w)
 	}
 }
 
@@ -112,8 +113,8 @@ func NewServeMux(opts ...Option) *ServeMux {
 }
 
 // Print print trie tree struct.
-func (mux *ServeMux) Print() {
-	mux.root.Print()
+func (mux *ServeMux) Print(w io.Writer) {
+	mux.root.Print(w)
 }
 
 // HandleFunc set func pattern path to handle

server_test.go

@@ -260,7 +260,7 @@ func Test_Use(t *testing.T) {
 	rec := httptest.NewRecorder()
 	req := httptest.NewRequest("GET", "/test", nil)
 	mux.ServeHTTP(rec, req)
-	mux.Print()
+	mux.Print(os.Stdout)
 	expected := []string{
 		"before_global1", "before_global2", "before_global3", "before_local",
 		"handler",
@@ -285,7 +285,7 @@ func Test_Node(t *testing.T) {
 	r.Add("/abc/def/", f)
 	r.Add("/abc/def/", f)
 	r.Add("/", f)
-	r.Print()
+	r.Print(os.Stdout)
 	//go ListenAndServe(context.Background(), r, URL("0.0.0.0:1234"))
 	//fmt.Println(r)
 }
@@ -566,7 +566,7 @@ func TestNode_PathsAndPrint(t *testing.T) {
 
 	// 测试打印功能
 	// 因为打印到标准输出，这里只验证不会 panic
-	node.Print()
+	node.Print(os.Stdout)
 }
 
 func Test_Methods(t *testing.T) {

stat.go

@@ -45,8 +45,17 @@ type Stat struct {
 
 // String implement fmt.Stringer interface.
 func (stat *Stat) String() string {
-	b, _ := json.Marshal(stat)
-	return string(b)
+	return a2s(stat)
+}
+
+// RequestBody return request body as string
+func (stat *Stat) RequestBody() string {
+	return a2s(stat.Request.Body)
+}
+
+// ResponseBody return response body as string
+func (stat *Stat) ResponseBody() string {
+	return a2s(stat.Response.Body)
 }
 
 // Print is used for server side
@@ -160,3 +169,11 @@ func serveLoad(w *ResponseWriter, r *http.Request, start time.Time, buf *bytes.B
 	stat.Response.Body = w.Content.String()
 	return stat
 }
+
+func a2s(v any) string {
+	b, err := json.Marshal(v)
+	if err != nil {
+		return fmt.Sprintf("%v", v)
+	}
+	return string(b)
+}