databricks-cli/bundle/config/variable/resolve_cluster_policy.go

25 lines
489 B
Go
Raw Normal View History

2024-11-20 21:40:33 +00:00
package variable
import (
"context"
"fmt"
"github.com/databricks/databricks-sdk-go"
)
2024-11-20 22:12:16 +00:00
type resolveClusterPolicy struct {
2024-11-20 21:40:33 +00:00
name string
}
2024-11-20 22:12:16 +00:00
func (l resolveClusterPolicy) Resolve(ctx context.Context, w *databricks.WorkspaceClient) (string, error) {
2024-11-20 21:40:33 +00:00
entity, err := w.ClusterPolicies.GetByName(ctx, l.name)
if err != nil {
return "", err
}
return fmt.Sprint(entity.PolicyId), nil
}
2024-11-20 22:12:16 +00:00
func (l resolveClusterPolicy) String() string {
2024-11-20 21:40:33 +00:00
return fmt.Sprintf("cluster-policy: %s", l.name)
}