mirror of https://github.com/databricks/cli.git
40 lines
1007 B
Go
40 lines
1007 B
Go
|
package resources
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"github.com/databricks/cli/libs/log"
|
||
|
|
"github.com/databricks/databricks-sdk-go"
|
||
|
|
"github.com/databricks/databricks-sdk-go/marshal"
|
||
|
|
"github.com/databricks/databricks-sdk-go/service/compute"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Cluster struct {
|
||
|
|
ID string `json:"id,omitempty" bundle:"readonly"`
|
||
|
|
Permissions []Permission `json:"permissions,omitempty"`
|
||
|
|
ModifiedStatus ModifiedStatus `json:"modified_status,omitempty" bundle:"internal"`
|
||
|
|
|
||
|
|
*compute.ClusterSpec
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Cluster) UnmarshalJSON(b []byte) error {
|
||
|
|
return marshal.Unmarshal(b, s)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s Cluster) MarshalJSON() ([]byte, error) {
|
||
|
|
return marshal.Marshal(s)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Cluster) Exists(ctx context.Context, w *databricks.WorkspaceClient, id string) (bool, error) {
|
||
|
|
_, err := w.Clusters.GetByClusterId(ctx, id)
|
||
|
|
if err != nil {
|
||
|
|
log.Debugf(ctx, "cluster %s does not exist", id)
|
||
|
|
return false, err
|
||
|
|
}
|
||
|
|
return true, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Cluster) TerraformResourceName() string {
|
||
|
|
return "databricks_cluster"
|
||
|
|
}
|