Golang中gRPC內(nèi)置Trace的實現(xiàn)

更新時間：2025年03月03日 10:46:41 作者：兩片空白

gRPC內(nèi)置了基于golang.org/x/net/trace包的客戶端和服務端請求追蹤,默認開啟狀態(tài),可以查看事件和請求日志,幫助調(diào)試,具有一定的參考價值,感興趣的可以了解一下

gRPC內(nèi)置了客戶端和服務端的請求追蹤，基于golang.org/x/net/trace包實現(xiàn)，默認是開啟狀態(tài)，可以查看事件和請求日志，對于基本的請求狀態(tài)查看調(diào)試也是很有幫助的，客戶端與服務端基本一致，這里以服務端開啟trace server為例，代碼如下。

目錄結構

代碼

服務端

package main

import (
	"context"
	"fmt"
	"log"
	"net"
	"net/http"
	hello "sample-app/grpc/proto"

	"golang.org/x/net/trace"
	"google.golang.org/grpc"
)

// gprc服務器地址
var Addr = "127.0.0.1:8080"

// 定義結構 實現(xiàn)約定接口
type helloService struct{}

// 服務
var HelloService = helloService{}

// 實現(xiàn)服務接口
func (h helloService) SayHello(c context.Context, req *hello.HelloRequest) (*hello.HelloResponse, error) {
	resp := new(hello.HelloResponse)
	resp.Message = fmt.Sprintf("Hello %s", req.Name)

	return resp, nil
}

func init() {
	//使用trace
	grpc.EnableTracing = true
}

func main() {
	ls, err := net.Listen("tcp", Addr)
	if err != nil {
		fmt.Println(err)
		return
	}
	//實例化grpc server
	server := grpc.NewServer()
	//注冊HelloService
	hello.RegisterHelloServer(server, HelloService)
	//開啟trace
	go startTrace()

	fmt.Println("Listen on " + Addr)
	server.Serve(ls)
}

func startTrace() {
	trace.AuthRequest = func(req *http.Request) (any, sensitive bool) {
		return true, true
	}

	go http.ListenAndServe(":8081", nil)
	log.Println("trace listen on 8081")
}

這里我們開啟一個http服務監(jiān)聽8081端口，用來查看grpc請求的trace信息。

運行：

客戶端

package main

import (
	"context"
	"fmt"
	hello "sample-app/grpc/proto"

	"google.golang.org/grpc"
)

const (
	Addr = "127.0.0.1:8080"
)

func main() {
	conn, err := grpc.Dial(Addr, grpc.WithInsecure())
	if err != nil {
		fmt.Println("Dial fail", err)
		return
	}
	defer conn.Close()
	//初始化服務器
	c := hello.NewHelloClient(conn)
	//單項請求
	req := new(hello.HelloRequest)
	req.Name = "gRPC"
	resp, err := c.SayHello(context.Background(), req)
	if err != nil {
		fmt.Println("say hello fail", err)
		return
	}
	fmt.Println(resp.Message)
}