deploy_crud.go

package k8s

import (
	"context"
	"errors"
	"time"

	"github.com/datewu/gtea/jsonlog"
	apps_v1 "k8s.io/api/apps/v1"
	v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)

func newDBio(d *apps_v1.Deployment) *Bio {
	bio := &Bio{
		Name: d.Name,
	}
	containes := d.Spec.Template.Spec.Containers
	cs := make([]*ConBio, len(containes))
	for i, c := range containes {
		cs[i] = &ConBio{
			Name:  c.Name,
			Image: c.Image,
			Pull:  string(c.ImagePullPolicy),
		}
	}
	bio.Containers = cs
	return bio
}

// GetDBio get deployment Bio
func GetDBio(ns, name string) (*Bio, error) {
	opts := v1.GetOptions{}
	ctx := context.Background()
	d, err := classicalClientSet.AppsV1().Deployments(ns).Get(ctx, name, opts)
	if err != nil {
		return nil, err
	}
	return newDBio(d), nil
}

// ListDeployWithLabels list deployment by label
func ListDeployWithLabels(ns, label string) ([]apps_v1.Deployment, error) {
	ctx := context.Background()
	opts := v1.ListOptions{LabelSelector: label}
	deploys, err := classicalClientSet.AppsV1().Deployments(ns).List(ctx, opts)
	if err != nil {
		return nil, err
	}
	return deploys.Items, nil
}

// ListBios list deployment bios
func ListBios(ns string) ([]*Bio, error) {
	ctx := context.Background()
	opts := v1.ListOptions{}
	deploys, err := classicalClientSet.AppsV1().Deployments(ns).List(ctx, opts)
	if err != nil {
		return nil, err
	}
	its := deploys.Items
	res := make([]*Bio, len(its))
	for i, d := range its {
		res[i] = newDBio(&d)
	}
	return res, nil
}

// SetDeployImgWithLabel ...
func SetDeployImgWithLabel(id *ContainerPath, label ...string) error {
	return setDeployImg(id, label...)
}

// SetDeployImg ...
func SetDeployImg(id *ContainerPath) error {
	return setDeployImg(id)
}

func setDeployImg(id *ContainerPath, labels ...string) error {
	ctx := context.Background()
	opts := v1.GetOptions{}
	d, err := classicalClientSet.AppsV1().Deployments(id.Ns).Get(ctx, id.Name, opts)
	if err != nil {
		jsonlog.Err(err, map[string]any{"name": id.Name, "msg": "get deploy failed"})
		return err
	}
	if labels != nil {
		ls := d.GetLabels()
		if err = checkLabels(ls, labels); err != nil {
			return err
		}
	}
	cpy := d.DeepCopy()
	found := false
	for i, c := range cpy.Spec.Template.Spec.Containers {
		if c.Name == id.CName {
			jsonlog.Info("got new image tag", map[string]any{"deploy": id.Name, "image": id.Img})
			cpy.Spec.Template.Spec.Containers[i].Image = id.Img
			found = true
			break
		}
	}
	if !found {
		fErr := errors.New("cannot find container")
		jsonlog.Err(fErr, map[string]any{"deploy": id.Name, "image": id.Img, "container": id.CName})
		return fErr
	}
	uOpts := v1.UpdateOptions{}
	if id.Name != "set-img" {
		zero := int32(0)
		cpy.Spec.Replicas = &zero
	}
	_, err = classicalClientSet.AppsV1().Deployments(id.Ns).Update(ctx, cpy, uOpts)
	if err != nil {
		jsonlog.Err(err, map[string]any{"deploy": id.Name, "image": id.Img, "msg": "update deploy failed"})
		return err
	}
	go func() {
		if id.Name == "set-img" {
			jsonlog.Debug("no scale deploy back replics for set-img")
			return
		}
		time.Sleep(3 * time.Second)
		a, rerr := classicalClientSet.AppsV1().Deployments(id.Ns).Get(ctx, id.Name, opts)
		if rerr != nil {
			jsonlog.Err(err, map[string]any{"name": id.Name, "msg": "get deploy failed"})
			return
		}
		acpy := a.DeepCopy()
		acpy.Spec.Replicas = d.Spec.Replicas
		jsonlog.Debug("going to scale deploy back replics",
			map[string]any{"*replicas": *d.Spec.Replicas, "replicas": d.Spec.Replicas})

		_, rerr = classicalClientSet.AppsV1().Deployments(id.Ns).Update(ctx, acpy, uOpts)
		if rerr != nil {
			jsonlog.Err(err, map[string]any{"name": id.Name, "msg": "scale deploy failed"})
			return
		}
	}()
	return nil
}