78 lines
2.8 KiB
Go
78 lines
2.8 KiB
Go
package tenant
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/influxdata/influxdb/v2"
|
|
"github.com/influxdata/influxdb/v2/kit/metric"
|
|
"github.com/influxdata/influxdb/v2/kit/platform"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
)
|
|
|
|
type BucketMetrics struct {
|
|
// RED metrics
|
|
rec *metric.REDClient
|
|
|
|
bucketService influxdb.BucketService
|
|
}
|
|
|
|
var _ influxdb.BucketService = (*BucketMetrics)(nil)
|
|
|
|
// NewBucketMetrics returns a metrics service middleware for the Bucket Service.
|
|
func NewBucketMetrics(reg prometheus.Registerer, s influxdb.BucketService, opts ...metric.ClientOptFn) *BucketMetrics {
|
|
o := metric.ApplyMetricOpts(opts...)
|
|
return &BucketMetrics{
|
|
rec: metric.New(reg, o.ApplySuffix("bucket")),
|
|
bucketService: s,
|
|
}
|
|
}
|
|
|
|
// Returns a single bucket by ID.
|
|
func (m *BucketMetrics) FindBucketByID(ctx context.Context, id platform.ID) (*influxdb.Bucket, error) {
|
|
rec := m.rec.Record("find_bucket_by_id")
|
|
bucket, err := m.bucketService.FindBucketByID(ctx, id)
|
|
return bucket, rec(err)
|
|
}
|
|
|
|
// Returns the first bucket that matches filter.
|
|
func (m *BucketMetrics) FindBucket(ctx context.Context, filter influxdb.BucketFilter) (*influxdb.Bucket, error) {
|
|
rec := m.rec.Record("find_bucket")
|
|
bucket, err := m.bucketService.FindBucket(ctx, filter)
|
|
return bucket, rec(err)
|
|
}
|
|
|
|
// FindBuckets returns a list of buckets that match filter and the total count of matching buckets.
|
|
func (m *BucketMetrics) FindBuckets(ctx context.Context, filter influxdb.BucketFilter, opt ...influxdb.FindOptions) ([]*influxdb.Bucket, int, error) {
|
|
rec := m.rec.Record("find_buckets")
|
|
buckets, n, err := m.bucketService.FindBuckets(ctx, filter, opt...)
|
|
return buckets, n, rec(err)
|
|
}
|
|
|
|
// Creates a new bucket and sets b.ID with the new identifier.
|
|
func (m *BucketMetrics) CreateBucket(ctx context.Context, b *influxdb.Bucket) error {
|
|
rec := m.rec.Record("create_bucket")
|
|
err := m.bucketService.CreateBucket(ctx, b)
|
|
return rec(err)
|
|
}
|
|
|
|
// Updates a single bucket with changeset and returns the new bucket state after update.
|
|
func (m *BucketMetrics) UpdateBucket(ctx context.Context, id platform.ID, upd influxdb.BucketUpdate) (*influxdb.Bucket, error) {
|
|
rec := m.rec.Record("update_bucket")
|
|
updatedBucket, err := m.bucketService.UpdateBucket(ctx, id, upd)
|
|
return updatedBucket, rec(err)
|
|
}
|
|
|
|
// Removes a bucket by ID.
|
|
func (m *BucketMetrics) DeleteBucket(ctx context.Context, id platform.ID) error {
|
|
rec := m.rec.Record("delete_bucket")
|
|
err := m.bucketService.DeleteBucket(ctx, id)
|
|
return rec(err)
|
|
}
|
|
|
|
// FindBucketByName finds a Bucket given its name and Organization ID
|
|
func (m *BucketMetrics) FindBucketByName(ctx context.Context, orgID platform.ID, name string) (*influxdb.Bucket, error) {
|
|
rec := m.rec.Record("find_bucket_by_name")
|
|
bucket, err := m.bucketService.FindBucketByName(ctx, orgID, name)
|
|
return bucket, rec(err)
|
|
}
|