Files
kubevela/pkg/cmd/status.go
2020-08-13 18:05:49 +08:00

99 lines
2.9 KiB
Go

package cmd
import (
"context"
"os"
corev1alpha2 "github.com/crossplane/oam-kubernetes-runtime/apis/core/v1alpha2"
"github.com/cloud-native-application/rudrx/api/types"
cmdutil "github.com/cloud-native-application/rudrx/pkg/cmd/util"
"github.com/ghodss/yaml"
"github.com/spf13/cobra"
"sigs.k8s.io/controller-runtime/pkg/client"
)
type ApplicationStatusMeta struct {
Status string `json:"Status,omitempty"`
Workload corev1alpha2.ComponentSpec `json:"Workload,omitempty"`
Traits []corev1alpha2.ComponentTrait `json:"Traits,omitempty"`
}
func NewAppStatusCommand(c types.Args, ioStreams cmdutil.IOStreams) *cobra.Command {
ctx := context.Background()
cmd := &cobra.Command{
Use: "app:status <APPLICATION-NAME>",
Aliases: []string{"status"},
Short: "get status of an application",
Long: "get status of an application, including its workload and trait",
Example: `vela app:status <APPLICATION-NAME>`,
RunE: func(cmd *cobra.Command, args []string) error {
argsLength := len(args)
if argsLength == 0 {
ioStreams.Errorf("Hint: please specify an application")
os.Exit(1)
}
appName := args[0]
env, err := GetEnv(cmd)
if err != nil {
ioStreams.Errorf("Error: failed to get Env: %s", err)
return err
}
namespace := env.Namespace
newClient, err := client.New(c.Config, client.Options{Scheme: c.Schema})
if err != nil {
return err
}
return printApplicationStatus(ctx, newClient, ioStreams, appName, namespace)
},
Annotations: map[string]string{
types.TagCommandType: types.TypeApp,
},
}
cmd.SetOut(ioStreams.Out)
return cmd
}
func printApplicationStatus(ctx context.Context, c client.Client, ioStreams cmdutil.IOStreams, appName string, namespace string) error {
application, err := RetrieveApplicationStatusByName(ctx, c, appName, namespace)
if err != nil {
return err
}
out, err := yaml.Marshal(application)
if err != nil {
return err
}
ioStreams.Info(string(out))
return nil
}
func RetrieveApplicationStatusByName(ctx context.Context, c client.Client, applicationName string, namespace string) (ApplicationStatusMeta, error) {
var applicationStatusMeta ApplicationStatusMeta
var appConfig corev1alpha2.ApplicationConfiguration
if err := c.Get(ctx, client.ObjectKey{Name: applicationName, Namespace: namespace}, &appConfig); err != nil {
return applicationStatusMeta, err
}
for _, com := range appConfig.Spec.Components {
// Just get the one component from appConfig
if com.ComponentName != applicationName {
continue
}
component, err := cmdutil.GetComponent(ctx, c, com.ComponentName, namespace)
if err != nil {
return applicationStatusMeta, err
}
var status = "UNKNOWN"
if len(appConfig.Status.Conditions) != 0 {
status = string(appConfig.Status.Conditions[0].Status)
}
applicationStatusMeta = ApplicationStatusMeta{
Status: status,
Workload: component.Spec,
Traits: com.Traits,
}
}
return applicationStatusMeta, nil
}