-
Notifications
You must be signed in to change notification settings - Fork 3
/
server.go
92 lines (74 loc) · 2.75 KB
/
server.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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
package test
import (
"context"
"errors"
"github.com/alexfalkowski/go-service/telemetry/tracer"
v1 "github.com/alexfalkowski/go-service/test/greet/v1"
"github.com/alexfalkowski/go-service/transport"
tgrpc "github.com/alexfalkowski/go-service/transport/grpc"
gtracer "github.com/alexfalkowski/go-service/transport/grpc/telemetry/tracer"
shttp "github.com/alexfalkowski/go-service/transport/http"
htracer "github.com/alexfalkowski/go-service/transport/http/telemetry/tracer"
"github.com/urfave/negroni/v3"
"go.opentelemetry.io/otel/metric"
"go.uber.org/fx"
"go.uber.org/zap"
"google.golang.org/grpc"
)
// ErrInvalidToken ...
var ErrInvalidToken = errors.New("invalid token")
// NewServer ...
func NewServer(verifyAuth bool) *Server {
return &Server{verifyAuth: verifyAuth}
}
// Server ...
type Server struct {
verifyAuth bool
v1.UnimplementedGreeterServiceServer
}
// SayHello ...
func (s *Server) SayHello(ctx context.Context, req *v1.SayHelloRequest) (*v1.SayHelloResponse, error) {
if s.verifyAuth && Test(ctx) != "auth" {
return nil, ErrInvalidToken
}
return &v1.SayHelloResponse{Message: "Hello " + req.GetName()}, nil
}
// SayStreamHello ...
func (s *Server) SayStreamHello(stream v1.GreeterService_SayStreamHelloServer) error {
if s.verifyAuth && Test(stream.Context()) != "auth" {
return ErrInvalidToken
}
req, err := stream.Recv()
if err != nil {
return err
}
return stream.Send(&v1.SayStreamHelloResponse{Message: "Hello " + req.GetName()})
}
// NewHTTPServer for test.
func NewHTTPServer(lc fx.Lifecycle, logger *zap.Logger, cfg *tracer.Config, tcfg *transport.Config, meter metric.Meter, handlers []negroni.Handler) *shttp.Server {
tracer, _ := htracer.NewTracer(htracer.Params{Lifecycle: lc, Config: cfg, Version: Version})
server, _ := shttp.NewServer(shttp.ServerParams{
Shutdowner: NewShutdowner(), Config: tcfg.HTTP, Logger: logger,
Tracer: tracer, Meter: meter, Handlers: handlers,
})
return server
}
// NewGRPCServer for test.
func NewGRPCServer(
lc fx.Lifecycle, logger *zap.Logger, cfg *tracer.Config, tcfg *transport.Config,
verifyAuth bool, meter metric.Meter,
unary []grpc.UnaryServerInterceptor, stream []grpc.StreamServerInterceptor,
) *tgrpc.Server {
tracer, _ := gtracer.NewTracer(gtracer.Params{Lifecycle: lc, Config: cfg, Version: Version})
server, _ := tgrpc.NewServer(tgrpc.ServerParams{
Shutdowner: NewShutdowner(), Config: tcfg.GRPC, Logger: logger,
Tracer: tracer, Meter: meter,
Unary: unary, Stream: stream,
})
v1.RegisterGreeterServiceServer(server.Server, NewServer(verifyAuth))
return server
}
// RegisterTransport for test.
func RegisterTransport(lc fx.Lifecycle, gs *tgrpc.Server, hs *shttp.Server) {
transport.Register(transport.RegisterParams{Lifecycle: lc, Servers: []transport.Server{gs, hs}})
}