This endpoint is relatively expensive, as it returns a large amount of data for each stack (essentially the stack's entire checkpoint). We currently call this endpoint in a few places that really just need information about a single stack. In these places we should simply interact with that stack directly. After these changes, there is only one call to the `stacks` endpoint on the startup path: `upgradeConfigurationFiles` loops over all of the stacks in each backend and attempts to upgrade each stack's configuration. Once we no longer need to do this we will not be hitting the `stacks` endpoint on each CLI invocation.
58 lines
1.4 KiB
Go
58 lines
1.4 KiB
Go
// Copyright 2016-2018, Pulumi Corporation. All rights reserved.
|
|
|
|
package state
|
|
|
|
import (
|
|
"github.com/pulumi/pulumi/pkg/backend"
|
|
"github.com/pulumi/pulumi/pkg/tokens"
|
|
"github.com/pulumi/pulumi/pkg/workspace"
|
|
)
|
|
|
|
// Stack returns an stack name after ensuring the stack exists. When an empty stack name is passed, the
|
|
// "current" ambient stack is returned.
|
|
func Stack(name tokens.QName, backends []backend.Backend) (backend.Stack, error) {
|
|
if name == "" {
|
|
return CurrentStack(backends)
|
|
}
|
|
|
|
// If not using the current stack, check all of the known backends to see if they know about it.
|
|
for _, be := range backends {
|
|
stack, err := be.GetStack(name)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if stack != nil {
|
|
return stack, nil
|
|
}
|
|
}
|
|
|
|
return nil, nil
|
|
}
|
|
|
|
// CurrentStack reads the current stack and returns an instance connected to its backend provider.
|
|
func CurrentStack(backends []backend.Backend) (backend.Stack, error) {
|
|
w, err := workspace.New()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
stackName := w.Settings().Stack
|
|
if stackName == "" {
|
|
return nil, nil
|
|
}
|
|
|
|
return Stack(stackName, backends)
|
|
}
|
|
|
|
// SetCurrentStack changes the current stack to the given stack name.
|
|
func SetCurrentStack(name tokens.QName) error {
|
|
// Switch the current workspace to that stack.
|
|
w, err := workspace.New()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
w.Settings().Stack = name
|
|
return w.Save()
|
|
}
|