mirror of https://github.com/databricks/cli.git
106 lines
2.2 KiB
Go
106 lines
2.2 KiB
Go
package deploy
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"io"
|
|
"io/fs"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/databricks/cli/bundle"
|
|
"github.com/databricks/cli/internal/build"
|
|
"github.com/databricks/cli/libs/diag"
|
|
"github.com/databricks/cli/libs/log"
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
type stateUpdate struct {
|
|
}
|
|
|
|
func (s *stateUpdate) Name() string {
|
|
return "deploy:state-update"
|
|
}
|
|
|
|
func (s *stateUpdate) Apply(ctx context.Context, b *bundle.Bundle) diag.Diagnostics {
|
|
state, err := load(ctx, b)
|
|
if err != nil {
|
|
return diag.FromErr(err)
|
|
}
|
|
|
|
// Increment the state sequence.
|
|
state.Seq = state.Seq + 1
|
|
|
|
// Update timestamp.
|
|
state.Timestamp = time.Now().UTC()
|
|
|
|
// Update the CLI version and deployment state version.
|
|
state.CliVersion = build.GetInfo().Version
|
|
state.Version = DeploymentStateVersion
|
|
|
|
// Update the state with the current list of synced files.
|
|
fl, err := FromSlice(b.Files)
|
|
if err != nil {
|
|
return diag.FromErr(err)
|
|
}
|
|
state.Files = fl
|
|
|
|
// Generate a UUID for the deployment, if one does not already exist
|
|
if state.ID == uuid.Nil {
|
|
state.ID = uuid.New()
|
|
}
|
|
|
|
statePath, err := getPathToStateFile(ctx, b)
|
|
if err != nil {
|
|
return diag.FromErr(err)
|
|
}
|
|
// Write the state back to the file.
|
|
f, err := os.OpenFile(statePath, os.O_CREATE|os.O_RDWR|os.O_TRUNC, 0600)
|
|
if err != nil {
|
|
log.Infof(ctx, "Unable to open deployment state file: %s", err)
|
|
return diag.FromErr(err)
|
|
}
|
|
defer f.Close()
|
|
|
|
data, err := json.Marshal(state)
|
|
if err != nil {
|
|
return diag.FromErr(err)
|
|
}
|
|
|
|
_, err = io.Copy(f, bytes.NewReader(data))
|
|
if err != nil {
|
|
return diag.FromErr(err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func StateUpdate() bundle.Mutator {
|
|
return &stateUpdate{}
|
|
}
|
|
|
|
func load(ctx context.Context, b *bundle.Bundle) (*DeploymentState, error) {
|
|
// If the file does not exist, return a new DeploymentState.
|
|
statePath, err := getPathToStateFile(ctx, b)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
log.Infof(ctx, "Loading deployment state from %s", statePath)
|
|
f, err := os.Open(statePath)
|
|
if err != nil {
|
|
if errors.Is(err, fs.ErrNotExist) {
|
|
log.Infof(ctx, "No deployment state file found")
|
|
return &DeploymentState{
|
|
Version: DeploymentStateVersion,
|
|
CliVersion: build.GetInfo().Version,
|
|
}, nil
|
|
}
|
|
return nil, err
|
|
}
|
|
defer f.Close()
|
|
return loadState(f)
|
|
}
|