kubevela/references/cli/top/model/pod.go

/*
Copyright 2022 The KubeVela Authors.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

	http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

package model

import (
	"context"
	"fmt"
	"strconv"
	"time"

	v1 "k8s.io/api/core/v1"
	"k8s.io/apimachinery/pkg/runtime"
	"k8s.io/client-go/rest"
	"sigs.k8s.io/controller-runtime/pkg/client"

	"github.com/oam-dev/kubevela/pkg/workflow/providers/legacy/query"
	"github.com/oam-dev/kubevela/references/cli/top/utils"
	clicommon "github.com/oam-dev/kubevela/references/common"
)

// Pod represent the k8s pod resource instance
type Pod struct {
	Name      string
	Namespace string
	Cluster   string
	Ready     string
	Status    string
	CPU       string
	Mem       string
	CPUR      string
	CPUL      string
	MemR      string
	MemL      string
	IP        string
	NodeName  string
	Age       string
}

// PodList is pod list
type PodList []Pod

// ListPods return pod list of component
func ListPods(ctx context.Context, cfg *rest.Config, c client.Client) (PodList, error) {
	appName := ctx.Value(&CtxKeyAppName).(string)
	appNamespace := ctx.Value(&CtxKeyNamespace).(string)
	compCluster := ctx.Value(&CtxKeyCluster).(string)
	compNamespace := ctx.Value(&CtxKeyClusterNamespace).(string)
	compName := ctx.Value(&CtxKeyComponentName).(string)

	opt := query.Option{
		Name:      appName,
		Namespace: appNamespace,
		Filter: query.FilterOption{
			Cluster:          compCluster,
			ClusterNamespace: compNamespace,
			Components:       []string{compName},
			APIVersion:       "v1",
			Kind:             "Pod",
		},
		WithTree: true,
	}
	resource, err := clicommon.CollectApplicationResource(ctx, c, opt)

	if err != nil {
		return PodList{}, err
	}
	list := make(PodList, len(resource))
	for index, object := range resource {
		pod := &v1.Pod{}
		err = runtime.DefaultUnstructuredConverter.FromUnstructured(object.UnstructuredContent(), pod)
		if err != nil {
			continue
		}
		list[index] = LoadPodDetail(c, cfg, pod, compCluster)
	}
	return list, nil
}

// LoadPodDetail gather the pod detail info
func LoadPodDetail(c client.Client, cfg *rest.Config, pod *v1.Pod, componentCluster string) Pod {
	podInfo := Pod{
		Name:      pod.Name,
		Namespace: pod.Namespace,
		Cluster:   componentCluster,
		Ready:     readyContainerNum(pod),
		Status:    string(pod.Status.Phase),
		CPU:       clicommon.MetricsNA,
		Mem:       clicommon.MetricsNA,
		CPUR:      clicommon.MetricsNA,
		MemR:      clicommon.MetricsNA,
		CPUL:      clicommon.MetricsNA,
		MemL:      clicommon.MetricsNA,
		IP:        pod.Status.PodIP,
		NodeName:  pod.Spec.NodeName,
		Age:       utils.TimeFormat(time.Since(pod.CreationTimestamp.Time)),
	}
	metric, err := clicommon.GetPodMetrics(cfg, pod.Name, pod.Namespace, componentCluster)
	if err == nil {
		spec, usage := clicommon.GetPodResourceSpecAndUsage(c, pod, metric)
		podInfo.CPU, podInfo.Mem = strconv.FormatInt(usage.CPU, 10), strconv.FormatInt(usage.Mem/(1024*1024), 10)
		podInfo.CPUL, podInfo.MemL = clicommon.ToPercentageStr(usage.CPU, spec.Lcpu), clicommon.ToPercentageStr(usage.Mem, spec.Lmem)
		podInfo.CPUR, podInfo.MemR = clicommon.ToPercentageStr(usage.CPU, spec.Rcpu), clicommon.ToPercentageStr(usage.Mem, spec.Rmem)
	}
	return podInfo
}

func readyContainerNum(pod *v1.Pod) string {
	total := len(pod.Status.ContainerStatuses)
	ready := 0
	for _, c := range pod.Status.ContainerStatuses {
		if c.Ready {
			ready++
		}
	}
	return fmt.Sprintf("%d/%d", ready, total)
}

// ToTableBody generate body of table in pod view
func (l PodList) ToTableBody() [][]string {
	data := make([][]string, len(l))
	for index, pod := range l {
		data[index] = []string{pod.Name, pod.Namespace, pod.Cluster, pod.Ready, pod.Status, pod.CPU, pod.Mem, pod.CPUR, pod.MemR, pod.CPUL, pod.MemL, pod.IP, pod.NodeName, pod.Age}
	}
	return data
}