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>
40 lines
1.2 KiB
Go
40 lines
1.2 KiB
Go
package middleware
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/distribution/distribution/v3"
|
|
)
|
|
|
|
// InitFunc is the type of a RepositoryMiddleware factory function and is
|
|
// used to register the constructor for different RepositoryMiddleware backends.
|
|
type InitFunc func(ctx context.Context, repository distribution.Repository, options map[string]interface{}) (distribution.Repository, error)
|
|
|
|
var middlewares map[string]InitFunc
|
|
|
|
// Register is used to register an InitFunc for
|
|
// a RepositoryMiddleware 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 RepositoryMiddleware with the given options using the named backend.
|
|
func Get(ctx context.Context, name string, options map[string]interface{}, repository distribution.Repository) (distribution.Repository, error) {
|
|
if middlewares != nil {
|
|
if initFunc, exists := middlewares[name]; exists {
|
|
return initFunc(ctx, repository, options)
|
|
}
|
|
}
|
|
|
|
return nil, fmt.Errorf("no repository middleware registered with name: %s", name)
|
|
}
|