Skip to content

Commit

Permalink
refactor(kv): augment bucket and org json errors
Browse files Browse the repository at this point in the history
  • Loading branch information
gavincabbage committed Jan 21, 2020
1 parent 523070f commit f9214ac
Show file tree
Hide file tree
Showing 5 changed files with 32 additions and 19 deletions.
1 change: 1 addition & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@
1. [16523](https://github.com/influxdata/influxdb/pull/16523): Change influx packages to be CRD compliant
1. [16547](https://github.com/influxdata/influxdb/pull/16547): Allow trailing newline in credentials file and CLI integration
1. [16545](https://github.com/influxdata/influxdb/pull/16545): Add support for prefixed cursor search to ForwardCursor types
1. [16522](https://github.com/influxdata/influxdb/pull/16522): Introduce resource logger to tasks, buckets and organizations

### UI Improvements

Expand Down
11 changes: 11 additions & 0 deletions bucket.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ package influxdb

import (
"context"
"fmt"
"strings"
"time"
)
Expand Down Expand Up @@ -68,6 +69,7 @@ var (
OpFindBucket = "FindBucket"
OpFindBuckets = "FindBuckets"
OpCreateBucket = "CreateBucket"
OpPutBucket = "PutBucket"
OpUpdateBucket = "UpdateBucket"
OpDeleteBucket = "DeleteBucket"
)
Expand Down Expand Up @@ -153,3 +155,12 @@ func (f BucketFilter) String() string {
}
return "[" + strings.Join(parts, ", ") + "]"
}

func ErrInternalBucketServiceError(op string, err error) *Error {
return &Error{
Code: EInternal,
Msg: fmt.Sprintf("unexpected error in buckets; Err: %v", err),
Op: op,
Err: err,
}
}
12 changes: 3 additions & 9 deletions kv/bucket.go
Original file line number Diff line number Diff line change
Expand Up @@ -481,9 +481,7 @@ func (s *Service) createBucket(ctx context.Context, tx Tx, b *influxdb.Bucket) (

v, err := json.Marshal(b)
if err != nil {
return &influxdb.Error{
Err: err,
}
return influxdb.ErrInternalBucketServiceError(influxdb.OpCreateBucket, err)
}
if err := s.putBucket(ctx, tx, b, v); err != nil {
return err
Expand Down Expand Up @@ -514,9 +512,7 @@ func (s *Service) PutBucket(ctx context.Context, b *influxdb.Bucket) error {
return s.kv.Update(ctx, func(tx Tx) error {
v, err := json.Marshal(b)
if err != nil {
return &influxdb.Error{
Err: err,
}
return influxdb.ErrInternalBucketServiceError(influxdb.OpPutBucket, err)
}

if err := s.putBucket(ctx, tx, b, v); err != nil {
Expand Down Expand Up @@ -722,9 +718,7 @@ func (s *Service) updateBucket(ctx context.Context, tx Tx, id influxdb.ID, upd i

v, err := json.Marshal(b)
if err != nil {
return nil, &influxdb.Error{
Err: err,
}
return nil, influxdb.ErrInternalBucketServiceError(influxdb.OpUpdateBucket, err)
}

if err := s.putBucket(ctx, tx, b, v); err != nil {
Expand Down
12 changes: 3 additions & 9 deletions kv/org.go
Original file line number Diff line number Diff line change
Expand Up @@ -297,9 +297,7 @@ func (s *Service) createOrganization(ctx context.Context, tx Tx, o *influxdb.Org

v, err := json.Marshal(o)
if err != nil {
return &influxdb.Error{
Err: err,
}
return influxdb.ErrInternalOrgServiceError(influxdb.OpCreateOrganization, err)
}
if err := s.putOrganization(ctx, tx, o, v); err != nil {
return &influxdb.Error{
Expand Down Expand Up @@ -328,9 +326,7 @@ func (s *Service) PutOrganization(ctx context.Context, o *influxdb.Organization)
return s.kv.Update(ctx, func(tx Tx) error {
v, err := json.Marshal(o)
if err != nil {
return &influxdb.Error{
Err: err,
}
return influxdb.ErrInternalOrgServiceError(influxdb.OpPutOrganization, err)
}

if err := s.putOrganization(ctx, tx, o, v); err != nil {
Expand Down Expand Up @@ -482,9 +478,7 @@ func (s *Service) updateOrganization(ctx context.Context, tx Tx, id influxdb.ID,

v, err := json.Marshal(o)
if err != nil {
return nil, &influxdb.Error{
Err: err,
}
return nil, influxdb.ErrInternalOrgServiceError(influxdb.OpUpdateOrganization, err)
}
if pe := s.putOrganization(ctx, tx, o, v); pe != nil {
return nil, pe
Expand Down
15 changes: 14 additions & 1 deletion organization.go
Original file line number Diff line number Diff line change
@@ -1,6 +1,9 @@
package influxdb

import "context"
import (
"context"
"fmt"
)

// Organization is an organization. 🎉
type Organization struct {
Expand All @@ -25,6 +28,7 @@ const (
OpFindOrganization = "FindOrganization"
OpFindOrganizations = "FindOrganizations"
OpCreateOrganization = "CreateOrganization"
OpPutOrganization = "PutOrganization"
OpUpdateOrganization = "UpdateOrganization"
OpDeleteOrganization = "DeleteOrganization"
)
Expand Down Expand Up @@ -70,3 +74,12 @@ type OrganizationFilter struct {
Name *string
ID *ID
}

func ErrInternalOrgServiceError(op string, err error) *Error {
return &Error{
Code: EInternal,
Msg: fmt.Sprintf("unexpected error in organizations; Err: %v", err),
Op: op,
Err: err,
}
}

0 comments on commit f9214ac

Please sign in to comment.