diff --git a/kv/auth.go b/kv/auth.go index 5edee12a83..8f22ecc1c2 100644 --- a/kv/auth.go +++ b/kv/auth.go @@ -303,6 +303,11 @@ func (s *Service) CreateAuthorization(ctx context.Context, a *influxdb.Authoriza }) } +// CreateAuthorizationTx is used when importing kv as a library +func (s *Service) CreateAuthorizationTx(ctx context.Context, tx Tx, a *influxdb.Authorization) error { + return s.createAuthorization(ctx, tx, a) +} + func (s *Service) createAuthorization(ctx context.Context, tx Tx, a *influxdb.Authorization) error { if err := a.Valid(); err != nil { return &influxdb.Error{ diff --git a/kv/bucket.go b/kv/bucket.go index 085f1f052a..4617b13875 100644 --- a/kv/bucket.go +++ b/kv/bucket.go @@ -455,6 +455,11 @@ func (s *Service) CreateBucket(ctx context.Context, b *influxdb.Bucket) error { }) } +// CreateBucketTx is used when importing kv as a library +func (s *Service) CreateBucketTx(ctx context.Context, tx Tx, b *influxdb.Bucket) (err error) { + return s.createBucket(ctx, tx, b) +} + func (s *Service) createBucket(ctx context.Context, tx Tx, b *influxdb.Bucket) (err error) { if b.OrgID.Valid() { span, ctx := tracing.StartSpanFromContext(ctx) diff --git a/kv/org.go b/kv/org.go index 36e3ab2919..4ec84c6403 100644 --- a/kv/org.go +++ b/kv/org.go @@ -279,6 +279,11 @@ func (s *Service) addOrgOwner(ctx context.Context, tx Tx, orgID influxdb.ID) err return s.addResourceOwner(ctx, tx, influxdb.OrgsResourceType, orgID) } +// CreateOrganizationTx is used when importing kv as a library +func (s *Service) CreateOrganizationTx(ctx context.Context, tx Tx, o *influxdb.Organization) (err error) { + return s.createOrganization(ctx, tx, o) +} + func (s *Service) createOrganization(ctx context.Context, tx Tx, o *influxdb.Organization) (err error) { if err := s.validOrganizationName(ctx, tx, o); err != nil { return err diff --git a/kv/urm.go b/kv/urm.go index 0ed5dddf13..a3d353e966 100644 --- a/kv/urm.go +++ b/kv/urm.go @@ -180,6 +180,11 @@ func (s *Service) CreateUserResourceMapping(ctx context.Context, m *influxdb.Use }) } +// CreateUserResourceMappingTx is used when importing kv as a library +func (s *Service) CreateUserResourceMappingTx(ctx context.Context, tx Tx, m *influxdb.UserResourceMapping) error { + return s.createUserResourceMapping(ctx, tx, m) +} + func (s *Service) createUserResourceMapping(ctx context.Context, tx Tx, m *influxdb.UserResourceMapping) error { span, ctx := tracing.StartSpanFromContext(ctx) defer span.Finish() diff --git a/kv/user.go b/kv/user.go index 1ce9c1a93d..d889a8d22c 100644 --- a/kv/user.go +++ b/kv/user.go @@ -230,6 +230,11 @@ func (s *Service) CreateUser(ctx context.Context, u *influxdb.User) error { }) } +// CreateUserTx is used when importing kv as a library +func (s *Service) CreateUserTx(ctx context.Context, tx Tx, u *influxdb.User) error { + return s.createUser(ctx, tx, u) +} + func (s *Service) createUser(ctx context.Context, tx Tx, u *influxdb.User) error { if err := s.uniqueUserName(ctx, tx, u); err != nil { return err