mirror of https://github.com/databricks/cli.git
23 lines
621 B
Go
23 lines
621 B
Go
|
package command
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"github.com/databricks/databricks-sdk-go"
|
||
|
|
)
|
||
|
|
|
||
|
|
func SetWorkspaceClient(ctx context.Context, w *databricks.WorkspaceClient) context.Context {
|
||
|
|
if v := ctx.Value(workspaceClientKey); v != nil {
|
||
|
|
panic("command.SetWorkspaceClient called twice on the same context.")
|
||
|
|
}
|
||
|
|
return context.WithValue(ctx, workspaceClientKey, w)
|
||
|
|
}
|
||
|
|
|
||
|
|
func WorkspaceClient(ctx context.Context) *databricks.WorkspaceClient {
|
||
|
|
v := ctx.Value(workspaceClientKey)
|
||
|
|
if v == nil {
|
||
|
|
panic("command.WorkspaceClient called without calling command.SetWorkspaceClient first.")
|
||
|
|
}
|
||
|
|
return v.(*databricks.WorkspaceClient)
|
||
|
|
}
|