-
Notifications
You must be signed in to change notification settings - Fork 4
/
key_access_server_registry.go
94 lines (81 loc) · 3.18 KB
/
key_access_server_registry.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
93
94
package kasregistry
import (
"context"
"log/slog"
"github.com/grpc-ecosystem/grpc-gateway/v2/runtime"
"github.com/opentdf/platform/pkg/serviceregistry"
kasr "github.com/opentdf/platform/protocol/go/kasregistry"
"github.com/opentdf/platform/services"
kasDb "github.com/opentdf/platform/services/kasregistry/db"
)
type KeyAccessServerRegistry struct {
kasr.UnimplementedKeyAccessServerRegistryServiceServer
dbClient *kasDb.KasRegistryDbClient
}
func NewRegistration() serviceregistry.Registration {
return serviceregistry.Registration{
Namespace: "policy",
ServiceDesc: &kasr.KeyAccessServerRegistryService_ServiceDesc,
RegisterFunc: func(srp serviceregistry.RegistrationParams) (any, serviceregistry.HandlerServer) {
return &KeyAccessServerRegistry{dbClient: kasDb.NewClient(*srp.DBClient)}, func(ctx context.Context, mux *runtime.ServeMux, s any) error {
return kasr.RegisterKeyAccessServerRegistryServiceHandlerServer(ctx, mux, s.(kasr.KeyAccessServerRegistryServiceServer))
}
},
}
}
func (s KeyAccessServerRegistry) CreateKeyAccessServer(ctx context.Context,
req *kasr.CreateKeyAccessServerRequest,
) (*kasr.CreateKeyAccessServerResponse, error) {
slog.Debug("creating key access server")
ks, err := s.dbClient.CreateKeyAccessServer(ctx, req)
if err != nil {
return nil, services.HandleError(err, services.ErrCreationFailed, slog.String("keyAccessServer", req.String()))
}
return &kasr.CreateKeyAccessServerResponse{
KeyAccessServer: ks,
}, nil
}
func (s KeyAccessServerRegistry) ListKeyAccessServers(ctx context.Context,
req *kasr.ListKeyAccessServersRequest,
) (*kasr.ListKeyAccessServersResponse, error) {
keyAccessServers, err := s.dbClient.ListKeyAccessServers(ctx)
if err != nil {
return nil, services.HandleError(err, services.ErrListRetrievalFailed)
}
return &kasr.ListKeyAccessServersResponse{
KeyAccessServers: keyAccessServers,
}, nil
}
func (s KeyAccessServerRegistry) GetKeyAccessServer(ctx context.Context,
req *kasr.GetKeyAccessServerRequest,
) (*kasr.GetKeyAccessServerResponse, error) {
keyAccessServer, err := s.dbClient.GetKeyAccessServer(ctx, req.Id)
if err != nil {
return nil, services.HandleError(err, services.ErrGetRetrievalFailed, slog.String("id", req.Id))
}
return &kasr.GetKeyAccessServerResponse{
KeyAccessServer: keyAccessServer,
}, nil
}
func (s KeyAccessServerRegistry) UpdateKeyAccessServer(ctx context.Context,
req *kasr.UpdateKeyAccessServerRequest,
) (*kasr.UpdateKeyAccessServerResponse, error) {
k, err := s.dbClient.UpdateKeyAccessServer(ctx, req.Id, req)
if err != nil {
return nil, services.HandleError(err, services.ErrUpdateFailed, slog.String("id", req.Id), slog.String("keyAccessServer", req.String()))
}
return &kasr.UpdateKeyAccessServerResponse{
KeyAccessServer: k,
}, nil
}
func (s KeyAccessServerRegistry) DeleteKeyAccessServer(ctx context.Context,
req *kasr.DeleteKeyAccessServerRequest,
) (*kasr.DeleteKeyAccessServerResponse, error) {
keyAccessServer, err := s.dbClient.DeleteKeyAccessServer(ctx, req.Id)
if err != nil {
return nil, services.HandleError(err, services.ErrDeletionFailed, slog.String("id", req.Id))
}
return &kasr.DeleteKeyAccessServerResponse{
KeyAccessServer: keyAccessServer,
}, nil
}