-
Notifications
You must be signed in to change notification settings - Fork 5
/
server.go
53 lines (45 loc) · 1.05 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
package test
import (
"context"
"net"
"sync"
pb "github.com/philip-bui/grpc-zerolog/protos"
"github.com/rs/zerolog/log"
"google.golang.org/grpc"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/status"
)
const (
address = "localhost:7070"
)
var (
server *grpc.Server
serverSync sync.Once
)
type TestServer struct{}
func (s *TestServer) TestUnary(ctx context.Context, t *pb.TestMessage) (*pb.TestMessage, error) {
if t.Test == "" {
return nil, status.Error(codes.InvalidArgument, "Empty message")
}
return t, nil
}
//func init() {
// log.Output(zerolog.ConsoleWriter{Out: os.Stderr})
//}
func StartServer(interceptor grpc.UnaryServerInterceptor) {
serverSync.Do(func() {
lis, err := net.Listen("tcp", address)
if err != nil {
log.Fatal().Err(err).Msg("start server")
} else {
log.Info().Msg("start server")
}
server = grpc.NewServer(
grpc.UnaryInterceptor(interceptor),
)
pb.RegisterTestServiceServer(server, &TestServer{})
if err := server.Serve(lis); err != nil {
log.Fatal().Err(err).Msg("start server")
}
})
}