Golang中gRPC内置Trace的实现

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"

// 定义结构 实现约定接口
type helloService struct{}

// 服务
var HelloService = helloService{}

// 实现服务接口
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")
}

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)
}