-
Notifications
You must be signed in to change notification settings - Fork 17
/
server.go
63 lines (51 loc) · 1.31 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
package grpcserver
import (
"crypto/tls"
"net"
"os"
"code.cloudfoundry.org/lager/v3"
"code.cloudfoundry.org/locket/models"
"google.golang.org/grpc"
"google.golang.org/grpc/credentials"
)
type grpcServerRunner struct {
listenAddress string
handler models.LocketServer
logger lager.Logger
tlsConfig *tls.Config
}
func NewGRPCServer(logger lager.Logger, listenAddress string, tlsConfig *tls.Config, handler models.LocketServer) grpcServerRunner {
return grpcServerRunner{
listenAddress: listenAddress,
handler: handler,
logger: logger,
tlsConfig: tlsConfig,
}
}
func (s grpcServerRunner) Run(signals <-chan os.Signal, ready chan<- struct{}) error {
logger := s.logger.Session("grpc-server")
logger.Info("started")
defer logger.Info("complete")
lis, err := net.Listen("tcp", s.listenAddress)
if err != nil {
logger.Error("failed-to-listen", err)
return err
}
server := grpc.NewServer(grpc.Creds(credentials.NewTLS(s.tlsConfig)))
models.RegisterLocketServer(server, s.handler)
errCh := make(chan error)
go func() {
errCh <- server.Serve(lis)
}()
close(ready)
select {
case sig := <-signals:
logger.Info("signalled", lager.Data{"signal": sig})
break
case err = <-errCh:
logger.Error("failed-to-serve", err)
break
}
server.GracefulStop()
return err
}