refactor: remove urms from new label service (#18301)

pull/18317/head
Alirie Gray 2020-06-01 09:28:03 -07:00 committed by GitHub
parent d304f920dc
commit 03ba8f250a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 5 additions and 23 deletions

View File

@ -876,7 +876,7 @@ func (m *Launcher) run(ctx context.Context) (err error) {
m.log.Error("Failed creating new labels store", zap.Error(err)) m.log.Error("Failed creating new labels store", zap.Error(err))
return err return err
} }
ls := label.NewService(labelsStore, m.kvService) ls := label.NewService(labelsStore)
labelSvc = label.NewLabelController(flagger, m.kvService, ls) labelSvc = label.NewLabelController(flagger, m.kvService, ls)
} }

View File

@ -2,7 +2,6 @@ package label
import ( import (
"context" "context"
"fmt"
"strings" "strings"
"github.com/influxdata/influxdb/v2" "github.com/influxdata/influxdb/v2"
@ -10,18 +9,12 @@ import (
) )
type Service struct { type Service struct {
store *Store store *Store
urmCreator UserResourceMappingCreator
} }
type UserResourceMappingCreator interface { func NewService(st *Store) influxdb.LabelService {
CreateUserResourceMappingForOrg(ctx context.Context, tx kv.Tx, orgID influxdb.ID, resID influxdb.ID, resType influxdb.ResourceType) error
}
func NewService(st *Store, urmCreator UserResourceMappingCreator) influxdb.LabelService {
return &Service{ return &Service{
store: st, store: st,
urmCreator: urmCreator, // todo (al) this can be removed once URMs are removed from the Label service
} }
} }
@ -45,10 +38,6 @@ func (s *Service) CreateLabel(ctx context.Context, l *influxdb.Label) error {
return err return err
} }
if err := s.urmCreator.CreateUserResourceMappingForOrg(ctx, tx, l.OrgID, l.ID, influxdb.LabelsResourceType); err != nil {
return err
}
return nil return nil
}) })
@ -145,17 +134,14 @@ func (s *Service) DeleteLabel(ctx context.Context, id influxdb.ID) error {
// CreateLabelMapping creates a new mapping between a resource and a label. // CreateLabelMapping creates a new mapping between a resource and a label.
func (s *Service) CreateLabelMapping(ctx context.Context, m *influxdb.LabelMapping) error { func (s *Service) CreateLabelMapping(ctx context.Context, m *influxdb.LabelMapping) error {
fmt.Println("creating label mapping")
err := s.store.View(ctx, func(tx kv.Tx) error { err := s.store.View(ctx, func(tx kv.Tx) error {
if _, err := s.store.GetLabel(ctx, tx, m.LabelID); err != nil { if _, err := s.store.GetLabel(ctx, tx, m.LabelID); err != nil {
fmt.Println("could not get label")
return err return err
} }
ls := []*influxdb.Label{} ls := []*influxdb.Label{}
err := s.store.FindResourceLabels(ctx, tx, influxdb.LabelMappingFilter{ResourceID: m.ResourceID, ResourceType: m.ResourceType}, &ls) err := s.store.FindResourceLabels(ctx, tx, influxdb.LabelMappingFilter{ResourceID: m.ResourceID, ResourceType: m.ResourceType}, &ls)
if err != nil { if err != nil {
fmt.Println("could not get resource labels")
return err return err
} }
for i := 0; i < len(ls); i++ { for i := 0; i < len(ls); i++ {
@ -167,12 +153,10 @@ func (s *Service) CreateLabelMapping(ctx context.Context, m *influxdb.LabelMappi
return nil return nil
}) })
if err != nil { if err != nil {
fmt.Println("could not .....?")
return err return err
} }
return s.store.Update(ctx, func(tx kv.Tx) error { return s.store.Update(ctx, func(tx kv.Tx) error {
fmt.Println("calling to store ")
return s.store.CreateLabelMapping(ctx, tx, m) return s.store.CreateLabelMapping(ctx, tx, m)
}) })
} }

View File

@ -59,9 +59,7 @@ func initLabelService(s kv.Store, f influxdbtesting.LabelFields, t *testing.T) (
t.Fatalf("failed to create label store: %v", err) t.Fatalf("failed to create label store: %v", err)
} }
kvSvc := kv.NewService(zaptest.NewLogger(t), s) svc := label.NewService(st)
svc := label.NewService(st, kvSvc)
ctx := context.Background() ctx := context.Background()
for _, l := range f.Labels { for _, l := range f.Labels {