steve/databricks-cli
steve
/
databricks-cli
mirror of https://github.com/databricks/cli.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Force send local_ssd_count field even if it's set to zero

This commit is contained in:
Andrew Nester 2024-04-16 12:13:10 +02:00
parent b71f853649
commit a51e789599
No known key found for this signature in database
GPG Key ID: 12BC628A44B7DA57
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
bundle/config/resources/job.go
View File

@ -5,6 +5,7 @@ import (
"strconv" "strconv"
"github.com/databricks/cli/bundle/config/paths" "github.com/databricks/cli/bundle/config/paths"
"github.com/databricks/cli/libs/dyn"
"github.com/databricks/cli/libs/log" "github.com/databricks/cli/libs/log"
"github.com/databricks/databricks-sdk-go" "github.com/databricks/databricks-sdk-go"
"github.com/databricks/databricks-sdk-go/marshal" "github.com/databricks/databricks-sdk-go/marshal"
@ -26,6 +27,20 @@ func (s *Job) UnmarshalJSON(b []byte) error {
} }
func (s Job) MarshalJSON() ([]byte, error) { func (s Job) MarshalJSON() ([]byte, error) {
// If local_ssd_count is set, we need to force send it
p := dyn.NewPattern(
dyn.Key("job_clusters"),
dyn.AnyIndex(),
dyn.Key("new_cluster"),
dyn.Key("gcp_attributes"),
dyn.Key("local_ssd_count"),
)
dyn.MapByPattern(s.DynamicValue, p, func(p dyn.Path, v dyn.Value) (dyn.Value, error) {
s.JobSettings.JobClusters[p[1].Index()].NewCluster.GcpAttributes.ForceSendFields = []string{"LocalSsdCount"}
return v, nil
})
return marshal.Marshal(s) return marshal.Marshal(s)
} }