70 lines
2.5 KiB
Go
70 lines
2.5 KiB
Go
package transport
|
|
|
|
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"
|
|
)
|
|
|
|
func newMetricCollectingService(reg prometheus.Registerer, underlying ReplicationService, opts ...metric.ClientOptFn) *metricsService {
|
|
o := metric.ApplyMetricOpts(opts...)
|
|
return &metricsService{
|
|
rec: metric.New(reg, o.ApplySuffix("replication")),
|
|
underlying: underlying,
|
|
}
|
|
}
|
|
|
|
type metricsService struct {
|
|
rec *metric.REDClient
|
|
underlying ReplicationService
|
|
}
|
|
|
|
var _ ReplicationService = (*metricsService)(nil)
|
|
|
|
func (m metricsService) ListReplications(ctx context.Context, filter influxdb.ReplicationListFilter) (*influxdb.Replications, error) {
|
|
rec := m.rec.Record("find_replications")
|
|
rcs, err := m.underlying.ListReplications(ctx, filter)
|
|
return rcs, rec(err)
|
|
}
|
|
|
|
func (m metricsService) CreateReplication(ctx context.Context, request influxdb.CreateReplicationRequest) (*influxdb.Replication, error) {
|
|
rec := m.rec.Record("create_replication")
|
|
r, err := m.underlying.CreateReplication(ctx, request)
|
|
return r, rec(err)
|
|
}
|
|
|
|
func (m metricsService) ValidateNewReplication(ctx context.Context, request influxdb.CreateReplicationRequest) error {
|
|
rec := m.rec.Record("validate_create_replication")
|
|
return rec(m.underlying.ValidateNewReplication(ctx, request))
|
|
}
|
|
|
|
func (m metricsService) GetReplication(ctx context.Context, id platform.ID) (*influxdb.Replication, error) {
|
|
rec := m.rec.Record("find_replication_by_id")
|
|
r, err := m.underlying.GetReplication(ctx, id)
|
|
return r, rec(err)
|
|
}
|
|
|
|
func (m metricsService) UpdateReplication(ctx context.Context, id platform.ID, request influxdb.UpdateReplicationRequest) (*influxdb.Replication, error) {
|
|
rec := m.rec.Record("update_replication")
|
|
r, err := m.underlying.UpdateReplication(ctx, id, request)
|
|
return r, rec(err)
|
|
}
|
|
|
|
func (m metricsService) ValidateUpdatedReplication(ctx context.Context, id platform.ID, request influxdb.UpdateReplicationRequest) error {
|
|
rec := m.rec.Record("validate_update_replication")
|
|
return rec(m.underlying.ValidateUpdatedReplication(ctx, id, request))
|
|
}
|
|
|
|
func (m metricsService) DeleteReplication(ctx context.Context, id platform.ID) error {
|
|
rec := m.rec.Record("delete_replication")
|
|
return rec(m.underlying.DeleteReplication(ctx, id))
|
|
}
|
|
|
|
func (m metricsService) ValidateReplication(ctx context.Context, id platform.ID) error {
|
|
rec := m.rec.Record("validate_replication")
|
|
return rec(m.underlying.ValidateReplication(ctx, id))
|
|
}
|