compose/context/context.go

package context

import (
	gocontext "context"

	"golang.org/x/net/context"
)

// Key is the key where the current docker context is stored in the metadata
// of a gRPC request
const Key = "context_key"

type currentContextKey struct{}

// WithCurrentContext sets the name of the current docker context
func WithCurrentContext(ctx gocontext.Context, contextName string) context.Context {
	return context.WithValue(ctx, currentContextKey{}, contextName)
}

// CurrentContext returns the current context name
func CurrentContext(ctx context.Context) string {
	cc, _ := ctx.Value(currentContextKey{}).(string)
	return cc
}