...
1 package resolver
2
3
4
5
6
7 import (
8 "context"
9
10 "edge-infra.dev/pkg/edge/api/graph/model"
11 )
12
13
14 func (r *mutationResolver) CreateCapability(ctx context.Context, capInput model.CapabilityInput) (*model.Capability, error) {
15 return r.CapabilityService.CreateCapability(ctx, capInput)
16 }
17
18
19 func (r *mutationResolver) UpdateCapability(ctx context.Context, capUpdateInput model.CapabilityUpdateInput) (bool, error) {
20 if err := r.CapabilityService.UpdateCapability(ctx, &capUpdateInput); err != nil {
21 return false, err
22 }
23 return true, nil
24 }
25
26
27 func (r *mutationResolver) AddCapability(ctx context.Context, capMapping model.CapabilityBannerMappingInput) (bool, error) {
28 if err := r.CapabilityService.AddCapability(ctx, &capMapping); err != nil {
29 return false, err
30 }
31 return true, nil
32 }
33
34
35 func (r *mutationResolver) RemoveCapability(ctx context.Context, capMapping model.CapabilityBannerMappingInput) (bool, error) {
36 if err := r.CapabilityService.RemoveCapability(ctx, &capMapping); err != nil {
37 return false, err
38 }
39 return true, nil
40 }
41
42
43 func (r *queryResolver) Capabilities(ctx context.Context, bannerEdgeID *string) ([]*model.Capability, error) {
44 if bannerEdgeID != nil {
45 return r.CapabilityService.ListCapabilitiesByBanner(ctx, bannerEdgeID)
46 }
47 return r.CapabilityService.ListCapabilities(ctx)
48 }
49
View as plain text