forked from TrueCloudLab/distribution
1d33874951
Go 1.13 and up enforce import paths to be versioned if a project contains a go.mod and has released v2 or up. The current v2.x branches (and releases) do not yet have a go.mod, and therefore are still allowed to be imported with a non-versioned import path (go modules add a `+incompatible` annotation in that case). However, now that this project has a `go.mod` file, incompatible import paths will not be accepted by go modules, and attempting to use code from this repository will fail. This patch uses `v3` for the import-paths (not `v2`), because changing import paths itself is a breaking change, which means that the next release should increment the "major" version to comply with SemVer (as go modules dictate). Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
69 lines
2.2 KiB
Go
69 lines
2.2 KiB
Go
package metrics
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/distribution/distribution/v3"
|
|
prometheus "github.com/distribution/distribution/v3/metrics"
|
|
"github.com/distribution/distribution/v3/registry/storage/cache"
|
|
"github.com/docker/go-metrics"
|
|
"github.com/opencontainers/go-digest"
|
|
)
|
|
|
|
type prometheusCacheProvider struct {
|
|
cache.BlobDescriptorCacheProvider
|
|
latencyTimer metrics.LabeledTimer
|
|
}
|
|
|
|
func NewPrometheusCacheProvider(wrap cache.BlobDescriptorCacheProvider, name, help string) cache.BlobDescriptorCacheProvider {
|
|
return &prometheusCacheProvider{
|
|
wrap,
|
|
// TODO: May want to have fine grained buckets since redis calls are generally <1ms and the default minimum bucket is 5ms.
|
|
prometheus.StorageNamespace.NewLabeledTimer(name, help, "operation"),
|
|
}
|
|
}
|
|
|
|
func (p *prometheusCacheProvider) Stat(ctx context.Context, dgst digest.Digest) (distribution.Descriptor, error) {
|
|
start := time.Now()
|
|
d, e := p.BlobDescriptorCacheProvider.Stat(ctx, dgst)
|
|
p.latencyTimer.WithValues("Stat").UpdateSince(start)
|
|
return d, e
|
|
}
|
|
|
|
func (p *prometheusCacheProvider) SetDescriptor(ctx context.Context, dgst digest.Digest, desc distribution.Descriptor) error {
|
|
start := time.Now()
|
|
e := p.BlobDescriptorCacheProvider.SetDescriptor(ctx, dgst, desc)
|
|
p.latencyTimer.WithValues("SetDescriptor").UpdateSince(start)
|
|
return e
|
|
}
|
|
|
|
type prometheusRepoCacheProvider struct {
|
|
distribution.BlobDescriptorService
|
|
latencyTimer metrics.LabeledTimer
|
|
}
|
|
|
|
func (p *prometheusRepoCacheProvider) Stat(ctx context.Context, dgst digest.Digest) (distribution.Descriptor, error) {
|
|
start := time.Now()
|
|
d, e := p.BlobDescriptorService.Stat(ctx, dgst)
|
|
p.latencyTimer.WithValues("RepoStat").UpdateSince(start)
|
|
return d, e
|
|
}
|
|
|
|
func (p *prometheusRepoCacheProvider) SetDescriptor(ctx context.Context, dgst digest.Digest, desc distribution.Descriptor) error {
|
|
start := time.Now()
|
|
e := p.BlobDescriptorService.SetDescriptor(ctx, dgst, desc)
|
|
p.latencyTimer.WithValues("RepoSetDescriptor").UpdateSince(start)
|
|
return e
|
|
}
|
|
|
|
func (p *prometheusCacheProvider) RepositoryScoped(repo string) (distribution.BlobDescriptorService, error) {
|
|
s, err := p.BlobDescriptorCacheProvider.RepositoryScoped(repo)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &prometheusRepoCacheProvider{
|
|
s,
|
|
p.latencyTimer,
|
|
}, nil
|
|
}
|