/
server.go
66 lines (52 loc) · 2.41 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
package services
import (
v0 "github.com/authzed/authzed-go/proto/authzed/api/v0"
v1 "github.com/authzed/authzed-go/proto/authzed/api/v1"
"github.com/authzed/authzed-go/proto/authzed/api/v1alpha1"
"github.com/authzed/grpcutil"
"google.golang.org/grpc"
healthpb "google.golang.org/grpc/health/grpc_health_v1"
"google.golang.org/grpc/reflection"
"github.com/authzed/spicedb/internal/datastore"
"github.com/authzed/spicedb/internal/dispatch"
"github.com/authzed/spicedb/internal/namespace"
v0svc "github.com/authzed/spicedb/internal/services/v0"
v1svc "github.com/authzed/spicedb/internal/services/v1"
v1alpha1svc "github.com/authzed/spicedb/internal/services/v1alpha1"
)
// SchemaServiceOption defines the options for enabled or disabled the V1 Schema service.
type SchemaServiceOption int
const (
// V1SchemaServiceDisabled indicates that the V1 schema service is disabled.
V1SchemaServiceDisabled SchemaServiceOption = 0
// V1SchemaServiceEnabled indicates that the V1 schema service is enabled.
V1SchemaServiceEnabled SchemaServiceOption = 1
)
// RegisterGrpcServices registers all services to be exposed on the GRPC server.
func RegisterGrpcServices(
srv *grpc.Server,
ds datastore.Datastore,
nsm namespace.Manager,
dispatch dispatch.Dispatcher,
maxDepth uint32,
prefixRequired v1alpha1svc.PrefixRequiredOption,
schemaServiceOption SchemaServiceOption,
) {
healthSrv := grpcutil.NewAuthlessHealthServer()
v0.RegisterACLServiceServer(srv, v0svc.NewACLServer(ds, nsm, dispatch, maxDepth))
healthSrv.SetServicesHealthy(&v0.ACLService_ServiceDesc)
v0.RegisterNamespaceServiceServer(srv, v0svc.NewNamespaceServer(ds))
healthSrv.SetServicesHealthy(&v0.NamespaceService_ServiceDesc)
v0.RegisterWatchServiceServer(srv, v0svc.NewWatchServer(ds, nsm))
healthSrv.SetServicesHealthy(&v0.WatchService_ServiceDesc)
v1alpha1.RegisterSchemaServiceServer(srv, v1alpha1svc.NewSchemaServer(ds, prefixRequired))
healthSrv.SetServicesHealthy(&v1alpha1.SchemaService_ServiceDesc)
v1.RegisterPermissionsServiceServer(srv, v1svc.NewPermissionsServer(ds, nsm, dispatch, maxDepth))
healthSrv.SetServicesHealthy(&v1.PermissionsService_ServiceDesc)
if schemaServiceOption == V1SchemaServiceEnabled {
v1.RegisterSchemaServiceServer(srv, v1svc.NewSchemaServer(ds))
healthSrv.SetServicesHealthy(&v1.SchemaService_ServiceDesc)
}
healthpb.RegisterHealthServer(srv, healthSrv)
reflection.Register(grpcutil.NewAuthlessReflectionInterceptor(srv))
}