github.com/influxdata/influxdb/v2@v2.7.6/remotes/transport/middleware_metrics.go (about)

     1  package transport
     2  
     3  import (
     4  	"context"
     5  
     6  	"github.com/influxdata/influxdb/v2"
     7  	"github.com/influxdata/influxdb/v2/kit/metric"
     8  	"github.com/influxdata/influxdb/v2/kit/platform"
     9  	"github.com/prometheus/client_golang/prometheus"
    10  )
    11  
    12  func newMetricCollectingService(reg prometheus.Registerer, underlying RemoteConnectionService, opts ...metric.ClientOptFn) *metricsService {
    13  	o := metric.ApplyMetricOpts(opts...)
    14  	return &metricsService{
    15  		rec:        metric.New(reg, o.ApplySuffix("remote")),
    16  		underlying: underlying,
    17  	}
    18  }
    19  
    20  type metricsService struct {
    21  	// RED metrics
    22  	rec        *metric.REDClient
    23  	underlying RemoteConnectionService
    24  }
    25  
    26  var _ RemoteConnectionService = (*metricsService)(nil)
    27  
    28  func (m metricsService) ListRemoteConnections(ctx context.Context, filter influxdb.RemoteConnectionListFilter) (*influxdb.RemoteConnections, error) {
    29  	rec := m.rec.Record("find_remotes")
    30  	rcs, err := m.underlying.ListRemoteConnections(ctx, filter)
    31  	return rcs, rec(err)
    32  }
    33  
    34  func (m metricsService) CreateRemoteConnection(ctx context.Context, request influxdb.CreateRemoteConnectionRequest) (*influxdb.RemoteConnection, error) {
    35  	rec := m.rec.Record("create_remote")
    36  	rc, err := m.underlying.CreateRemoteConnection(ctx, request)
    37  	return rc, rec(err)
    38  }
    39  
    40  func (m metricsService) GetRemoteConnection(ctx context.Context, id platform.ID) (*influxdb.RemoteConnection, error) {
    41  	rec := m.rec.Record("find_remote_by_id")
    42  	rc, err := m.underlying.GetRemoteConnection(ctx, id)
    43  	return rc, rec(err)
    44  }
    45  
    46  func (m metricsService) UpdateRemoteConnection(ctx context.Context, id platform.ID, request influxdb.UpdateRemoteConnectionRequest) (*influxdb.RemoteConnection, error) {
    47  	rec := m.rec.Record("update_remote")
    48  	rc, err := m.underlying.UpdateRemoteConnection(ctx, id, request)
    49  	return rc, rec(err)
    50  }
    51  
    52  func (m metricsService) DeleteRemoteConnection(ctx context.Context, id platform.ID) error {
    53  	rec := m.rec.Record("delete_remote")
    54  	return rec(m.underlying.DeleteRemoteConnection(ctx, id))
    55  }