databricks-cli/bundle/deploy/lock/acquire.go

65 lines
1.5 KiB
Go
Raw Normal View History

package lock
import (
"context"
"errors"
"fmt"
"github.com/databricks/cli/bundle"
"github.com/databricks/cli/libs/filer"
"github.com/databricks/cli/libs/locker"
"github.com/databricks/cli/libs/log"
)
type acquire struct{}
func Acquire() bundle.Mutator {
return &acquire{}
}
func (m *acquire) Name() string {
return "lock:acquire"
}
func (m *acquire) init(b *bundle.Bundle) error {
user := b.Config.Workspace.CurrentUser.UserName
dir := b.Config.Workspace.StatePath
l, err := locker.CreateLocker(user, dir, b.WorkspaceClient())
if err != nil {
return err
}
b.Locker = l
return nil
}
func (m *acquire) Apply(ctx context.Context, b *bundle.Bundle) error {
// Return early if locking is disabled.
if !b.Config.Bundle.Deployment.Lock.IsEnabled() {
log.Infof(ctx, "Skipping; locking is disabled")
return nil
}
err := m.init(b)
if err != nil {
return err
}
force := b.Config.Bundle.Deployment.Lock.Force
log.Infof(ctx, "Acquiring deployment lock (force: %v)", force)
err = b.Locker.Lock(ctx, force)
if err != nil {
log.Errorf(ctx, "Failed to acquire deployment lock: %v", err)
notExistsError := filer.NoSuchDirectoryError{}
if errors.As(err, &notExistsError) {
// If we get a "doesn't exist" error from the API this indicates
// we either don't have permissions or the path is invalid.
return fmt.Errorf("cannot write to deployment root (this can indicate a previous deploy was done with a different identity): %s", b.Config.Workspace.RootPath)
}
return err
}
return nil
}