b3aeba7062
Let the test runners choose the namespaces and wrap the contexts. This allows the test suite to create multiple contexts without worrying about namespacing or leasing in the contexts. Signed-off-by: Derek McGowan <derek@mcgstyle.net>
64 lines
1.6 KiB
Go
64 lines
1.6 KiB
Go
package namespaces
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
|
|
"github.com/containerd/containerd/errdefs"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
const (
|
|
// NamespaceEnvVar is the environment variable key name
|
|
NamespaceEnvVar = "CONTAINERD_NAMESPACE"
|
|
// Default is the name of the default namespace
|
|
Default = "default"
|
|
)
|
|
|
|
type namespaceKey struct{}
|
|
|
|
// WithNamespace sets a given namespace on the context
|
|
func WithNamespace(ctx context.Context, namespace string) context.Context {
|
|
ctx = context.WithValue(ctx, namespaceKey{}, namespace) // set our key for namespace
|
|
|
|
// also store on the grpc headers so it gets picked up by any clients that
|
|
// are using this.
|
|
return withGRPCNamespaceHeader(ctx, namespace)
|
|
}
|
|
|
|
// NamespaceFromEnv uses the namespace defined in CONTAINERD_NAMESPACE or
|
|
// default
|
|
func NamespaceFromEnv(ctx context.Context) context.Context {
|
|
namespace := os.Getenv(NamespaceEnvVar)
|
|
if namespace == "" {
|
|
namespace = Default
|
|
}
|
|
return WithNamespace(ctx, namespace)
|
|
}
|
|
|
|
// Namespace returns the namespace from the context.
|
|
//
|
|
// The namespace is not guaranteed to be valid.
|
|
func Namespace(ctx context.Context) (string, bool) {
|
|
namespace, ok := ctx.Value(namespaceKey{}).(string)
|
|
if !ok {
|
|
return fromGRPCHeader(ctx)
|
|
}
|
|
|
|
return namespace, ok
|
|
}
|
|
|
|
// NamespaceRequired returns the valid namepace from the context or an error.
|
|
func NamespaceRequired(ctx context.Context) (string, error) {
|
|
namespace, ok := Namespace(ctx)
|
|
if !ok || namespace == "" {
|
|
return "", errors.Wrapf(errdefs.ErrFailedPrecondition, "namespace is required")
|
|
}
|
|
|
|
if err := Validate(namespace); err != nil {
|
|
return "", errors.Wrap(err, "namespace validation")
|
|
}
|
|
|
|
return namespace, nil
|
|
}
|