/
main.go
53 lines (41 loc) · 1.03 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
package main
import (
"context"
"fmt"
"log"
"net"
"net/http"
"github.com/eviltomorrow/developer-kit/grpc-name-resolver/pb"
"golang.org/x/net/trace"
"google.golang.org/grpc"
)
// HelloService hello
type HelloService struct {
pb.UnimplementedGreeterServer
}
// SayHello hello
func (hs *HelloService) SayHello(ctx context.Context, request *pb.HelloRequest) (*pb.HelloReply, error) {
return &pb.HelloReply{Message: "Hello " + request.Name}, nil
}
func main() {
listen, err := net.Listen("tcp", fmt.Sprintf(":%d", 8080))
if err != nil {
log.Fatalf("failed to listen: %v", err)
}
grpc.EnableTracing = true
startTrace()
s := grpc.NewServer(grpc.NumStreamWorkers(3))
pb.RegisterGreeterServer(s, &HelloService{})
log.Println("Server start")
if err := s.Serve(listen); err != nil {
log.Fatalf("failed to serve: %v", err)
}
s.Stop()
}
func startTrace() {
trace.AuthRequest = func(req *http.Request) (any, sensitive bool) {
return true, true
}
go http.ListenAndServe(":8090", nil)
log.Printf("Trace listen on 8090\r\n")
}