mirror of https://github.com/databricks/cli.git
35 lines
808 B
Go
35 lines
808 B
Go
package bundle
|
|
|
|
import (
|
|
"context"
|
|
)
|
|
|
|
// Placeholder to use as unique key in context.Context.
|
|
var bundleKey int
|
|
|
|
// Context stores the specified bundle on a new context.
|
|
// The bundle is available through the `Get()` function.
|
|
func Context(ctx context.Context, b *Bundle) context.Context {
|
|
return context.WithValue(ctx, &bundleKey, b)
|
|
}
|
|
|
|
// GetOrNil returns the bundle as configured on the context.
|
|
// It returns nil if it isn't configured.
|
|
func GetOrNil(ctx context.Context) *Bundle {
|
|
bundle, ok := ctx.Value(&bundleKey).(*Bundle)
|
|
if !ok {
|
|
return nil
|
|
}
|
|
return bundle
|
|
}
|
|
|
|
// Get returns the bundle as configured on the context.
|
|
// It panics if it isn't configured.
|
|
func Get(ctx context.Context) *Bundle {
|
|
b := GetOrNil(ctx)
|
|
if b == nil {
|
|
panic("context not configured with bundle")
|
|
}
|
|
return b
|
|
}
|