mirror of https://github.com/databricks/cli.git
35 lines
1.2 KiB
Go
35 lines
1.2 KiB
Go
package clusters
|
|
|
|
import (
|
|
"github.com/databricks/cli/libs/cmdio"
|
|
"github.com/databricks/databricks-sdk-go/service/compute"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func listOverride(listCmd *cobra.Command, _ *compute.ListClustersRequest) {
|
|
listCmd.Annotations["headerTemplate"] = cmdio.Heredoc(`
|
|
{{header "ID"}} {{header "Name"}} {{header "State"}}`)
|
|
listCmd.Annotations["template"] = cmdio.Heredoc(`
|
|
{{range .}}{{.ClusterId | green}} {{.ClusterName | cyan}} {{if eq .State "RUNNING"}}{{green "%s" .State}}{{else if eq .State "TERMINATED"}}{{red "%s" .State}}{{else}}{{blue "%s" .State}}{{end}}
|
|
{{end}}`)
|
|
}
|
|
|
|
func listNodeTypesOverride(listNodeTypesCmd *cobra.Command) {
|
|
listNodeTypesCmd.Annotations["template"] = cmdio.Heredoc(`
|
|
{{range .NodeTypes}}{{.NodeTypeId | green}} {{.NumCores}} {{.MemoryMb}} {{.Category | blue}}
|
|
{{end}}`)
|
|
}
|
|
|
|
func sparkVersionsOverride(sparkVersionsCmd *cobra.Command) {
|
|
sparkVersionsCmd.Annotations["template"] = cmdio.Heredoc(`
|
|
{{range .Versions}}{{.Key | green}} {{.Name}}
|
|
{{end}}
|
|
`)
|
|
}
|
|
|
|
func init() {
|
|
listOverrides = append(listOverrides, listOverride)
|
|
listNodeTypesOverrides = append(listNodeTypesOverrides, listNodeTypesOverride)
|
|
sparkVersionsOverrides = append(sparkVersionsOverrides, sparkVersionsOverride)
|
|
}
|