forked from TrueCloudLab/distribution
d1cb12fa3d
with a new `proxy` section in the configuration file. Create a new registry type which delegates storage to a proxyBlobStore and proxyManifestStore. These stores will pull through data if not present locally. proxyBlobStore takes care not to write duplicate data to disk. Add a scheduler to cleanup expired content. The scheduler runs as a background goroutine. When a blob or manifest is pulled through from the remote registry, an entry is added to the scheduler with a TTL. When the TTL expires the scheduler calls a pre-specified function to remove the fetched resource. Add token authentication to the registry middleware. Get a token at startup and preload the credential store with the username and password supplied in the config file. Allow resumable digest functionality to be disabled at runtime and disable it when the registry is a pull through cache. Signed-off-by: Richard Scothern <richard.scothern@gmail.com>
74 lines
2 KiB
Go
74 lines
2 KiB
Go
package proxy
|
|
|
|
import (
|
|
"expvar"
|
|
"sync/atomic"
|
|
)
|
|
|
|
// Metrics is used to hold metric counters
|
|
// related to the proxy
|
|
type Metrics struct {
|
|
Requests uint64
|
|
Hits uint64
|
|
Misses uint64
|
|
BytesPulled uint64
|
|
BytesPushed uint64
|
|
}
|
|
|
|
type proxyMetricsCollector struct {
|
|
blobMetrics Metrics
|
|
manifestMetrics Metrics
|
|
}
|
|
|
|
// BlobPull tracks metrics about blobs pulled into the cache
|
|
func (pmc *proxyMetricsCollector) BlobPull(bytesPulled uint64) {
|
|
atomic.AddUint64(&pmc.blobMetrics.Misses, 1)
|
|
atomic.AddUint64(&pmc.blobMetrics.BytesPulled, bytesPulled)
|
|
}
|
|
|
|
// BlobPush tracks metrics about blobs pushed to clients
|
|
func (pmc *proxyMetricsCollector) BlobPush(bytesPushed uint64) {
|
|
atomic.AddUint64(&pmc.blobMetrics.Requests, 1)
|
|
atomic.AddUint64(&pmc.blobMetrics.Hits, 1)
|
|
atomic.AddUint64(&pmc.blobMetrics.BytesPushed, bytesPushed)
|
|
}
|
|
|
|
// ManifestPull tracks metrics related to Manifests pulled into the cache
|
|
func (pmc *proxyMetricsCollector) ManifestPull(bytesPulled uint64) {
|
|
atomic.AddUint64(&pmc.manifestMetrics.Misses, 1)
|
|
atomic.AddUint64(&pmc.manifestMetrics.BytesPulled, bytesPulled)
|
|
}
|
|
|
|
// ManifestPush tracks metrics about manifests pushed to clients
|
|
func (pmc *proxyMetricsCollector) ManifestPush(bytesPushed uint64) {
|
|
atomic.AddUint64(&pmc.manifestMetrics.Requests, 1)
|
|
atomic.AddUint64(&pmc.manifestMetrics.Hits, 1)
|
|
atomic.AddUint64(&pmc.manifestMetrics.BytesPushed, bytesPushed)
|
|
}
|
|
|
|
// proxyMetrics tracks metrics about the proxy cache. This is
|
|
// kept globally and made available via expvar.
|
|
var proxyMetrics = &proxyMetricsCollector{}
|
|
|
|
func init() {
|
|
registry := expvar.Get("registry")
|
|
if registry == nil {
|
|
registry = expvar.NewMap("registry")
|
|
}
|
|
|
|
pm := registry.(*expvar.Map).Get("proxy")
|
|
if pm == nil {
|
|
pm = &expvar.Map{}
|
|
pm.(*expvar.Map).Init()
|
|
registry.(*expvar.Map).Set("proxy", pm)
|
|
}
|
|
|
|
pm.(*expvar.Map).Set("blobs", expvar.Func(func() interface{} {
|
|
return proxyMetrics.blobMetrics
|
|
}))
|
|
|
|
pm.(*expvar.Map).Set("manifests", expvar.Func(func() interface{} {
|
|
return proxyMetrics.manifestMetrics
|
|
}))
|
|
|
|
}
|