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 admin
import (
"crypto/tls"
grpc_middleware "github.com/grpc-ecosystem/go-grpc-middleware"
grpc_auth "github.com/grpc-ecosystem/go-grpc-middleware/auth"
grpc_logrus "github.com/grpc-ecosystem/go-grpc-middleware/logging/logrus"
grpc_recovery "github.com/grpc-ecosystem/go-grpc-middleware/recovery"
grpc_prometheus "github.com/grpc-ecosystem/go-grpc-prometheus"
log "github.com/sirupsen/logrus"
"google.golang.org/grpc"
"google.golang.org/grpc/credentials"
"google.golang.org/grpc/health"
healthpb "google.golang.org/grpc/health/grpc_health_v1"
)
var logrusEntry *log.Entry
func init() {
logger := log.StandardLogger()
logrusEntry = log.NewEntry(logger)
grpc_logrus.ReplaceGrpcLogger(logrusEntry)
}
// NewServer creates a new unencrypted gRPC server for the gitlab-pages admin API.
func NewServer(secret string) *grpc.Server {
grpcServer := grpc.NewServer(serverOpts(secret)...)
registerServices(grpcServer)
return grpcServer
}
// NewTLSServer creates a new gRPC server with encryption for the gitlab-pages admin API.
func NewTLSServer(secret string, cert *tls.Certificate) *grpc.Server {
grpcServer := grpc.NewServer(append(
serverOpts(secret),
grpc.Creds(credentials.NewServerTLSFromCert(cert)),
)...)
registerServices(grpcServer)
return grpcServer
}
func serverOpts(secret string) []grpc.ServerOption {
return []grpc.ServerOption{
grpc.StreamInterceptor(grpc_middleware.ChainStreamServer(
grpc_prometheus.StreamServerInterceptor,
grpc_logrus.StreamServerInterceptor(logrusEntry),
grpc_auth.StreamServerInterceptor(authFunc(secret)),
grpc_recovery.StreamServerInterceptor(),
)),
grpc.UnaryInterceptor(grpc_middleware.ChainUnaryServer(
grpc_prometheus.UnaryServerInterceptor,
grpc_logrus.UnaryServerInterceptor(logrusEntry),
grpc_auth.UnaryServerInterceptor(authFunc(secret)),
grpc_recovery.UnaryServerInterceptor(),
)),
}
}
func registerServices(g *grpc.Server) {
healthpb.RegisterHealthServer(g, health.NewServer())
}
|