forked from TrueCloudLab/distribution
e0281dc609
gofumpt (https://github.com/mvdan/gofumpt) provides a supserset of `gofmt` / `go fmt`, and addresses various formatting issues that linters may be checking for. We can consider enabling the `gofumpt` linter to verify the formatting in CI, although not every developer may have it installed, so for now this runs it once to get formatting in shape. Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
56 lines
1.7 KiB
Go
56 lines
1.7 KiB
Go
package middleware
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/distribution/distribution/v3"
|
|
"github.com/distribution/distribution/v3/registry/storage"
|
|
)
|
|
|
|
// InitFunc is the type of a RegistryMiddleware factory function and is
|
|
// used to register the constructor for different RegistryMiddleware backends.
|
|
type InitFunc func(ctx context.Context, registry distribution.Namespace, options map[string]interface{}) (distribution.Namespace, error)
|
|
|
|
var (
|
|
middlewares map[string]InitFunc
|
|
registryoptions []storage.RegistryOption
|
|
)
|
|
|
|
// Register is used to register an InitFunc for
|
|
// a RegistryMiddleware backend with the given name.
|
|
func Register(name string, initFunc InitFunc) error {
|
|
if middlewares == nil {
|
|
middlewares = make(map[string]InitFunc)
|
|
}
|
|
if _, exists := middlewares[name]; exists {
|
|
return fmt.Errorf("name already registered: %s", name)
|
|
}
|
|
|
|
middlewares[name] = initFunc
|
|
|
|
return nil
|
|
}
|
|
|
|
// Get constructs a RegistryMiddleware with the given options using the named backend.
|
|
func Get(ctx context.Context, name string, options map[string]interface{}, registry distribution.Namespace) (distribution.Namespace, error) {
|
|
if middlewares != nil {
|
|
if initFunc, exists := middlewares[name]; exists {
|
|
return initFunc(ctx, registry, options)
|
|
}
|
|
}
|
|
|
|
return nil, fmt.Errorf("no registry middleware registered with name: %s", name)
|
|
}
|
|
|
|
// RegisterOptions adds more options to RegistryOption list. Options get applied before
|
|
// any other configuration-based options.
|
|
func RegisterOptions(options ...storage.RegistryOption) error {
|
|
registryoptions = append(registryoptions, options...)
|
|
return nil
|
|
}
|
|
|
|
// GetRegistryOptions returns list of RegistryOption.
|
|
func GetRegistryOptions() []storage.RegistryOption {
|
|
return registryoptions
|
|
}
|