Cloudpods/backend/pkg/mcclient/modules/k8s/getter.go

// Copyright 2019 Yunion
//
// 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 k8s

import (
	"fmt"
	"strings"
	"time"

	"github.com/hako/durafmt"

	"yunion.io/x/jsonutils"
	"yunion.io/x/log"
)

var (
	getName      nameGetter
	getStatus    statusGetter
	getAge       ageGetter
	getNamespace namespaceGetter
	getLabel     labelGetter
)

type nameGetter struct{}

func (g nameGetter) Get_Id(obj jsonutils.JSONObject) interface{} {
	id, _ := obj.GetString("id")
	return id
}

func (g nameGetter) Get_Name(obj jsonutils.JSONObject) interface{} {
	name, _ := obj.GetString("name")
	return name
}

type statusGetter struct{}

func (g statusGetter) Get_Status(obj jsonutils.JSONObject) interface{} {
	status, _ := obj.GetString("status")
	return status
}

type ageGetter struct{}

func (g ageGetter) Get_Age(obj jsonutils.JSONObject) interface{} {
	creationTimestamp, err := obj.GetString("creationTimestamp")
	if err != nil {
		creationTimestamp, err = obj.GetString("created_at")
		if err != nil {
			log.Errorf("Get creationTimestamp and created_at error: %v", err)
			return nil
		}
	}
	t, err := time.Parse(time.RFC3339, creationTimestamp)
	if err != nil {
		log.Errorf("parse creationTimestamp %v: %v", creationTimestamp, err)
		return nil
	}
	dur := time.Since(t)
	return durafmt.ParseShort(dur).String()
}

type namespaceGetter struct{}

func (g namespaceGetter) Get_Namespace(obj jsonutils.JSONObject) interface{} {
	ns, _ := obj.GetString("namespace")
	return ns
}

type labelGetter struct{}

func (g labelGetter) Get_Labels(obj jsonutils.JSONObject) interface{} {
	labels, _ := obj.GetMap("labels")
	str := ""
	ls := []string{}
	for k, v := range labels {
		vs, _ := v.GetString()
		ls = append(ls, fmt.Sprintf("%s=%s", k, vs))
	}
	if len(ls) != 0 {
		str = strings.Join(ls, ",")
	}
	return str
}