【GO】 K8s 管理系统项目[API部分--Deployment]

K8s 管理系统项目[API部分–Deployment]

1. 实现接口

service/dataselector.go

package service

import (
	"sort"
	"strings"
	"time"

	appsv1 "k8s.io/api/apps/v1"
	corev1 "k8s.io/api/core/v1"
)

// dataselector 用于排序,过滤,分页的数据类型
type dataSelector struct {
    
    
	GenericDataList []DataCell
	DataSelect      *DataSelectQuery
}

// DataCell 接口,用于各种资源List的类型转换,转换后可以使用dataselector的排序,过滤,分页方法
type DataCell interface {
    
    
	GetCreation() time.Time
	GetName() string
}

// DataSelectQuery 定义过滤和分页的结构体,过滤:Name 分页:Limit和Page
type DataSelectQuery struct {
    
    
	Filter   *FilterQuery
	Paginate *PaginateQuery
}

// FilterQuery 用于查询 过滤:Name
type FilterQuery struct {
    
    
	Name string
}

// 分页:Limit和Page Limit是单页的数据条数,Page是第几页
type PaginateQuery struct {
    
    
	Page  int
	Limit int
}

// 实现自定义的排序方法,需要重写Len,Swap,Less方法
// Len用于获取数组的长度
func (d *dataSelector) Len() int {
    
    
	return len(d.GenericDataList)
}

// Swap用于数据比较大小后的位置变更
func (d *dataSelector) Swap(i, j int) {
    
    
	d.GenericDataList[i], d.GenericDataList[j] = d.GenericDataList[j], d.GenericDataList[i]
}

// Less用于比较大小
func (d *dataSelector) Less(i, j int) bool {
    
    
	return d.GenericDataList[i].GetCreation().Before(d.GenericDataList[j].GetCreation())
}

// 重写以上三个方法,用sort.Sort 方法触发排序
func (d *dataSelector) Sort() *dataSelector {
    
    
	sort.Sort(d)
	return d
}

// Filter方法用于过滤,比较数据Name属性,若包含则返回
func (d *dataSelector) Filter() *dataSelector {
    
    
	if d.DataSelect.Filter.Name == "" {
    
    
		return d
	}
	filtered := []DataCell{
    
    }
	for _, value := range d.GenericDataList {
    
    
		// 定义是否匹配的标签变量,默认是匹配的
		matches := true
		objName := value.GetName()
		if !strings.Contains(objName, d.DataSelect.Filter.Name) {
    
    
			matches = false
			continue
		}
		if matches {
    
    
			filtered = append(filtered, value)
		}
	}
	d.GenericDataList = filtered
	return d
}

// Paginate 分页,根据Limit和Page的传参,取一定范围内的数据返回
func (d *dataSelector) Paginate() *dataSelector {
    
    
	limit := d.DataSelect.Paginate.Limit
	page := d.DataSelect.Paginate.Page
	//验证参数合法,若参数不合法,则返回所有数据
	if limit <= 0 || page <= 0 {
    
    
		return d
	}
	//举例:25个元素的数组,limit是10,page是3,startIndex是20,endIndex是30(实际上endIndex是25)
	startIndex := limit * (page - 1)
	endIndex := limit * page

	//处理最后一页,这时候就把endIndex由30改为25了
	if len(d.GenericDataList) < endIndex {
    
    
		endIndex = len(d.GenericDataList)
	}
	d.GenericDataList = d.GenericDataList[startIndex:endIndex]
	return d
}

// 定义podCell, 重写GetCreation和GetName 方法后,可以进行数据转换
// covev1.Pod --> podCell  --> DataCell
// appsv1.Deployment --> deployCell --> DataCell
type podCell corev1.Pod

// 重写DataCell接口的两个方法
func (p podCell) GetCreation() time.Time {
    
    
	return p.CreationTimestamp.Time
}
func (p podCell) GetName() string {
    
    
	return p.Name
}
// deployCell
type deploymentCell appsv1.Deployment

func (d deploymentCell) GetCreation() time.Time {
    
    
	return d.CreationTimestamp.Time
}
func (d deploymentCell) GetName() string {
    
    
	return d.Name
}

2. Deployment功能

service/deployment.go

  1. 列表
  2. 获取Deployment详情
  3. 修改Deployment副本数
  4. 创建Deployment
  5. 删除Deployment
  6. 重启Deployment
  7. 更新Deployment
  8. 获取每个namespace的Deployment数量

2.1 from和to方法

func (d *deployment) toCells(std []appsv1.Deployment) []DataCell {
    
    
	cells := make([]DataCell, len(std))
	for i := range std {
    
    
		cells[i] = deploymentCell(std[i])
	}
	return cells
}

func (d *deployment) fromCells(cells []DataCell) []appsv1.Deployment {
    
    
	deployments := make([]appsv1.Deployment, len(cells))
	for i := range cells {
    
    
		deployments[i] = appsv1.Deployment(cells[i].(deploymentCell))
	}

	return deployments
}

2.2 获取deployment副本数

// 设置deployment副本数
func (d *deployment) ScaleDeployment(deploymentName, namespace string, scaleNum int) (replica int32, err error) {
    
    
	//获取autoscalingv1.Scale类型的对象,能点出当前的副本数
	scale, err := K8s.ClientSet.AppsV1().Deployments(namespace).GetScale(context.TODO(), deploymentName, metav1.GetOptions{
    
    })
	if err != nil {
    
    
		logger.Error(errors.New("获取Deployment副本数信息失败, " + err.Error()))
		return 0, errors.New("获取Deployment副本数信息失败, " + err.Error())
	}
	//修改副本数
	scale.Spec.Replicas = int32(scaleNum)
	//更新副本数,传入scale对象
	newScale, err := K8s.ClientSet.AppsV1().Deployments(namespace).UpdateScale(context.TODO(), deploymentName, scale, metav1.UpdateOptions{
    
    })
	if err != nil {
    
    
		logger.Error(errors.New("更新Deployment副本数信息失败, " + err.Error()))
		return 0, errors.New("更新Deployment副本数信息失败, " + err.Error())
	}

	return newScale.Spec.Replicas, nil
}

2.3 创建deployment

// 创建deployment,接收DeployCreate对象
func (d *deployment) CreateDeployment(data *DeployCreate) (err error) {
    
    
	//将data中的属性组装成appsv1.Deployment对象,并将入参数据放入
	deployment := &appsv1.Deployment{
    
    
		//ObjectMeta中定义资源名、命名空间以及标签
		ObjectMeta: metav1.ObjectMeta{
    
    
			Name:      data.Name,
			Namespace: data.Namespace,
			Labels:    data.Label,
		},
		//Spec中定义副本数、选择器、以及pod属性
		Spec: appsv1.DeploymentSpec{
    
    
			Replicas: &data.Replicas,
			Selector: &metav1.LabelSelector{
    
    
				MatchLabels: data.Label,
			},
			Template: corev1.PodTemplateSpec{
    
    
				//定义pod名和标签
				ObjectMeta: metav1.ObjectMeta{
    
    
					Name:   data.Name,
					Labels: data.Label,
				},
				//定义容器名、镜像和端口
				Spec: corev1.PodSpec{
    
    
					Containers: []corev1.Container{
    
    
						{
    
    
							Name:  data.Name,
							Image: data.Image,
							Ports: []corev1.ContainerPort{
    
    
								{
    
    
									Name:          "http",
									Protocol:      corev1.ProtocolTCP,
									ContainerPort: data.ContainerPort,
								},
							},
						},
					},
				},
			},
		},
		//Status定义资源的运行状态,这里由于是新建,传入空的appsv1.DeploymentStatus{}对象即可
		Status: appsv1.DeploymentStatus{
    
    },
	}
	//判断是否打开健康检查功能,若打开,则定义ReadinessProbe和LivenessProbe
	if data.HealthCheck {
    
    
		//设置第一个容器的ReadinessProbe,因为我们pod中只有一个容器,所以直接使用index 0即可
		//若pod中有多个容器,则这里需要使用for循环去定义了
		deployment.Spec.Template.Spec.Containers[0].ReadinessProbe = &corev1.Probe{
    
    
			ProbeHandler: corev1.ProbeHandler{
    
    
				HTTPGet: &corev1.HTTPGetAction{
    
    
					Path: data.HealthPath,
					//intstr.IntOrString的作用是端口可以定义为整型,也可以定义为字符串
					//Type=0则表示表示该结构体实例内的数据为整型,转json时只使用IntVal的数据
					//Type=1则表示表示该结构体实例内的数据为字符串,转json时只使用StrVal的数据
					Port: intstr.IntOrString{
    
    
						Type:   0,
						IntVal: data.ContainerPort,
					},
				},
			},
			//初始化等待时间
			InitialDelaySeconds: 5,
			//超时时间
			TimeoutSeconds: 5,
			//执行间隔
			PeriodSeconds: 5,
		}
		deployment.Spec.Template.Spec.Containers[0].LivenessProbe = &corev1.Probe{
    
    
			ProbeHandler: corev1.ProbeHandler{
    
    
				HTTPGet: &corev1.HTTPGetAction{
    
    
					Path: data.HealthPath,
					Port: intstr.IntOrString{
    
    
						Type:   0,
						IntVal: data.ContainerPort,
					},
				},
			},
			InitialDelaySeconds: 15,
			TimeoutSeconds:      5,
			PeriodSeconds:       5,
		}
		//定义容器的limit和request资源
		deployment.Spec.Template.Spec.Containers[0].Resources.Limits = map[corev1.ResourceName]resource.Quantity{
    
    
			corev1.ResourceCPU:    resource.MustParse(data.Cpu),
			corev1.ResourceMemory: resource.MustParse(data.Memory),
		}
		deployment.Spec.Template.Spec.Containers[0].Resources.Requests = map[corev1.ResourceName]resource.Quantity{
    
    
			corev1.ResourceCPU:    resource.MustParse(data.Cpu),
			corev1.ResourceMemory: resource.MustParse(data.Memory),
		}
	}
	//调用sdk创建deployment
	_, err = K8s.ClientSet.AppsV1().Deployments(data.Namespace).Create(context.TODO(), deployment, metav1.CreateOptions{
    
    })
	if err != nil {
    
    
		logger.Error(errors.New("创建Deployment失败, " + err.Error()))
		return errors.New("创建Deployment失败, " + err.Error())
	}
	return nil
}

2.4 删除Deployment

// 删除deployment
func (d *deployment) DeleteDeployment(deploymentName, namespace string) (err error) {
    
    
	err = K8s.ClientSet.AppsV1().Deployments(namespace).Delete(context.TODO(), deploymentName, metav1.DeleteOptions{
    
    })
	if err != nil {
    
    
		logger.Error(errors.New("删除Deployment失败, " + err.Error()))
		return errors.New("删除Deployment失败, " + err.Error())
	}

	return nil
}

2.5 重启deployment

// 重启deployment
func (d *deployment) RestartDeployment(deploymentName, namespace string) (err error) {
    
    
	//此功能等同于一下kubectl命令
	//kubectl deployment ${service} -p \
	//'{"spec":{"template":{"spec":{"containers":[{"name":"'"${service}"'","env":[{"name":"RESTART_","value":"'$(date +%s)'"}]}]}}}}'

	//使用patchData Map组装数据
	patchData := map[string]interface{
    
    }{
    
    
		"spec": map[string]interface{
    
    }{
    
    
			"template": map[string]interface{
    
    }{
    
    
				"spec": map[string]interface{
    
    }{
    
    
					"containers": []map[string]interface{
    
    }{
    
    
						{
    
    "name": deploymentName,
							"env": []map[string]string{
    
    {
    
    
								"name":  "RESTART_",
								"value": strconv.FormatInt(time.Now().Unix(), 10),
							}},
						},
					},
				},
			},
		},
	}
	//序列化为字节,因为patch方法只接收字节类型参数
	patchByte, err := json.Marshal(patchData)
	if err != nil {
    
    
		logger.Error(errors.New("json序列化失败, " + err.Error()))
		return errors.New("json序列化失败, " + err.Error())
	}
	//调用patch方法更新deployment
	_, err = K8s.ClientSet.AppsV1().Deployments(namespace).Patch(context.TODO(), deploymentName, "application/strategic-merge-patch+json", patchByte, metav1.PatchOptions{
    
    })
	if err != nil {
    
    
		logger.Error(errors.New("重启Deployment失败, " + err.Error()))
		return errors.New("重启Deployment失败, " + err.Error())
	}

	return nil
}

2.6 更新deployment

// 更新deployment
func (d *deployment) UpdateDeployment(namespace, content string) (err error) {
    
    
	var deploy = &appsv1.Deployment{
    
    }

	err = json.Unmarshal([]byte(content), deploy)
	if err != nil {
    
    
		logger.Error(errors.New("反序列化失败, " + err.Error()))
		return errors.New("反序列化失败, " + err.Error())
	}

	_, err = K8s.ClientSet.AppsV1().Deployments(namespace).Update(context.TODO(), deploy, metav1.UpdateOptions{
    
    })
	if err != nil {
    
    
		logger.Error(errors.New("更新Deployment失败, " + err.Error()))
		return errors.New("更新Deployment失败, " + err.Error())
	}
	return nil
}

2.7 获取所有Namespace中Deployment数量

// 获取每个namespace的deployment数量
func (d *deployment) GetDeployNumPerNp() (deploysNps []*DeploysNp, err error) {
    
    
	namespaceList, err := K8s.ClientSet.CoreV1().Namespaces().List(context.TODO(), metav1.ListOptions{
    
    })
	if err != nil {
    
    
		return nil, err
	}
	for _, namespace := range namespaceList.Items {
    
    
		deploymentList, err := K8s.ClientSet.AppsV1().Deployments(namespace.Name).List(context.TODO(), metav1.ListOptions{
    
    })
		if err != nil {
    
    
			return nil, err
		}

		deploysNp := &DeploysNp{
    
    
			Namespace: namespace.Name,
			DeployNum: len(deploymentList.Items),
		}

		deploysNps = append(deploysNps, deploysNp)
	}
	return deploysNps, nil
}

2.8 整个deployment

service/deployment.go

package service

import (
	"context"
	"encoding/json"
	"errors"
	"strconv"
	"time"
	"github.com/wonderivan/logger"
	appsv1 "k8s.io/api/apps/v1"
	corev1 "k8s.io/api/core/v1"
	"k8s.io/apimachinery/pkg/api/resource"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/apimachinery/pkg/util/intstr"
)

var Deployment deployment

type deployment struct{
    
    }

// 定义列表的返回内容,Items是deployment元素列表,Total为deployment元素数量
type DeploymentsResp struct {
    
    
	Items []appsv1.Deployment `json:"items"`
	Total int                 `json:"total"`
}
// 定义DeployCreate结构体,用于创建deployment需要的参数属性的定义
type DeployCreate struct {
    
    
	Name          string            `json:"name"`
	Namespace     string            `json:"namespace"`
	Replicas      int32             `json:"replicas"`
	Image         string            `json:"image"`
	Label         map[string]string `json:"label"`
	Cpu           string            `json:"cpu"`
	Memory        string            `json:"memory"`
	ContainerPort int32             `json:"container_port"`
	HealthCheck   bool              `json:"health_check"`
	HealthPath    string            `json:"health_path"`
}
// 定义DeploysNp类型,用于返回namespace中deployment的数量
type DeploysNp struct {
    
    
	Namespace string `json:"namespace"`
	DeployNum int    `json:"deployment_num"`
}
// 获取deployment列表,支持过滤、排序、分页
func (d *deployment) GetDeployments(filterName, namespace string, limit, page int) (deploymentsResp *DeploymentsResp, err error) {
    
    
	//获取deploymentList类型的deployment列表
	deploymentList, err := K8s.ClientSet.AppsV1().Deployments(namespace).List(context.TODO(), metav1.ListOptions{
    
    })
	if err != nil {
    
    
		logger.Error(errors.New("获取Deployment列表失败, " + err.Error()))
		return nil, errors.New("获取Deployment列表失败, " + err.Error())
	}
	//将deploymentList中的deployment列表(Items),放进dataselector对象中,进行排序
	selectableData := &dataSelector{
    
    
		GenericDataList: d.toCells(deploymentList.Items),
		DataSelect: &DataSelectQuery{
    
    
			Filter: &FilterQuery{
    
    Name: filterName},
			Paginate: &PaginateQuery{
    
    
				Limit: limit,
				Page:  page,
			},
		},
	}
	filtered := selectableData.Filter()
	total := len(filtered.GenericDataList)
	data := filtered.Sort().Paginate()
	//将[]DataCell类型的deployment列表转为appsv1.deployment列表
	deployments := d.fromCells(data.GenericDataList)
	return &DeploymentsResp{
    
    
		Items: deployments,
		Total: total,
	}, nil
}
// 获取deployment详情
func (d *deployment) GetDeploymentDetail(deploymentName, namespace string) (deployment *appsv1.Deployment, err error) {
    
    
	deployment, err = K8s.ClientSet.AppsV1().Deployments(namespace).Get(context.TODO(), deploymentName, metav1.GetOptions{
    
    })
	if err != nil {
    
    
		logger.Error(errors.New("获取Deployment详情失败, " + err.Error()))
		return nil, errors.New("获取Deployment详情失败, " + err.Error())
	}
	return deployment, nil
}
// 设置deployment副本数
func (d *deployment) ScaleDeployment(deploymentName, namespace string, scaleNum int) (replica int32, err error) {
    
    
	//获取autoscalingv1.Scale类型的对象,能点出当前的副本数
	scale, err := K8s.ClientSet.AppsV1().Deployments(namespace).GetScale(context.TODO(), deploymentName, metav1.GetOptions{
    
    })
	if err != nil {
    
    
		logger.Error(errors.New("获取Deployment副本数信息失败, " + err.Error()))
		return 0, errors.New("获取Deployment副本数信息失败, " + err.Error())
	}
	//修改副本数
	scale.Spec.Replicas = int32(scaleNum)
	//更新副本数,传入scale对象
	newScale, err := K8s.ClientSet.AppsV1().Deployments(namespace).UpdateScale(context.TODO(), deploymentName, scale, metav1.UpdateOptions{
    
    })
	if err != nil {
    
    
		logger.Error(errors.New("更新Deployment副本数信息失败, " + err.Error()))
		return 0, errors.New("更新Deployment副本数信息失败, " + err.Error())
	}
	return newScale.Spec.Replicas, nil
}
// 创建deployment,接收DeployCreate对象
func (d *deployment) CreateDeployment(data *DeployCreate) (err error) {
    
    
	//将data中的属性组装成appsv1.Deployment对象,并将入参数据放入
	deployment := &appsv1.Deployment{
    
    
		//ObjectMeta中定义资源名、命名空间以及标签
		ObjectMeta: metav1.ObjectMeta{
    
    
			Name:      data.Name,
			Namespace: data.Namespace,
			Labels:    data.Label,
		},
		//Spec中定义副本数、选择器、以及pod属性
		Spec: appsv1.DeploymentSpec{
    
    
			Replicas: &data.Replicas,
			Selector: &metav1.LabelSelector{
    
    
				MatchLabels: data.Label,
			},
			Template: corev1.PodTemplateSpec{
    
    
				//定义pod名和标签
				ObjectMeta: metav1.ObjectMeta{
    
    
					Name:   data.Name,
					Labels: data.Label,
				},
				//定义容器名、镜像和端口
				Spec: corev1.PodSpec{
    
    
					Containers: []corev1.Container{
    
    
						{
    
    
							Name:  data.Name,
							Image: data.Image,
							Ports: []corev1.ContainerPort{
    
    
								{
    
    
									Name:          "http",
									Protocol:      corev1.ProtocolTCP,
									ContainerPort: data.ContainerPort,
								},
							},
						},
					},
				},
			},
		},
		//Status定义资源的运行状态,这里由于是新建,传入空的appsv1.DeploymentStatus{}对象即可
		Status: appsv1.DeploymentStatus{
    
    },
	}
	//判断是否打开健康检查功能,若打开,则定义ReadinessProbe和LivenessProbe
	if data.HealthCheck {
    
    
		//设置第一个容器的ReadinessProbe,因为我们pod中只有一个容器,所以直接使用index 0即可
		//若pod中有多个容器,则这里需要使用for循环去定义了
		deployment.Spec.Template.Spec.Containers[0].ReadinessProbe = &corev1.Probe{
    
    
			ProbeHandler: corev1.ProbeHandler{
    
    
				HTTPGet: &corev1.HTTPGetAction{
    
    
					Path: data.HealthPath,
					//intstr.IntOrString的作用是端口可以定义为整型,也可以定义为字符串
					//Type=0则表示表示该结构体实例内的数据为整型,转json时只使用IntVal的数据
					//Type=1则表示表示该结构体实例内的数据为字符串,转json时只使用StrVal的数据
					Port: intstr.IntOrString{
    
    
						Type:   0,
						IntVal: data.ContainerPort,
					},
				},
			},
			//初始化等待时间
			InitialDelaySeconds: 5,
			//超时时间
			TimeoutSeconds: 5,
			//执行间隔
			PeriodSeconds: 5,
		}
		deployment.Spec.Template.Spec.Containers[0].LivenessProbe = &corev1.Probe{
    
    
			ProbeHandler: corev1.ProbeHandler{
    
    
				HTTPGet: &corev1.HTTPGetAction{
    
    
					Path: data.HealthPath,
					Port: intstr.IntOrString{
    
    
						Type:   0,
						IntVal: data.ContainerPort,
					},
				},
			},
			InitialDelaySeconds: 15,
			TimeoutSeconds:      5,
			PeriodSeconds:       5,
		}
		//定义容器的limit和request资源
		deployment.Spec.Template.Spec.Containers[0].Resources.Limits = map[corev1.ResourceName]resource.Quantity{
    
    
			corev1.ResourceCPU:    resource.MustParse(data.Cpu),
			corev1.ResourceMemory: resource.MustParse(data.Memory),
		}
		deployment.Spec.Template.Spec.Containers[0].Resources.Requests = map[corev1.ResourceName]resource.Quantity{
    
    
			corev1.ResourceCPU:    resource.MustParse(data.Cpu),
			corev1.ResourceMemory: resource.MustParse(data.Memory),
		}
	}
	//调用sdk创建deployment
	_, err = K8s.ClientSet.AppsV1().Deployments(data.Namespace).Create(context.TODO(), deployment, metav1.CreateOptions{
    
    })
	if err != nil {
    
    
		logger.Error(errors.New("创建Deployment失败, " + err.Error()))
		return errors.New("创建Deployment失败, " + err.Error())
	}
	return nil
}

// 删除deployment
func (d *deployment) DeleteDeployment(deploymentName, namespace string) (err error) {
    
    
	err = K8s.ClientSet.AppsV1().Deployments(namespace).Delete(context.TODO(), deploymentName, metav1.DeleteOptions{
    
    })
	if err != nil {
    
    
		logger.Error(errors.New("删除Deployment失败, " + err.Error()))
		return errors.New("删除Deployment失败, " + err.Error())
	}
	return nil
}

// 重启deployment
func (d *deployment) RestartDeployment(deploymentName, namespace string) (err error) {
    
    
	//此功能等同于一下kubectl命令
	//kubectl deployment ${service} -p \
	//'{"spec":{"template":{"spec":{"containers":[{"name":"'"${service}"'","env":[{"name":"RESTART_","value":"'$(date +%s)'"}]}]}}}}'

	//使用patchData Map组装数据
	patchData := map[string]interface{
    
    }{
    
    
		"spec": map[string]interface{
    
    }{
    
    
			"template": map[string]interface{
    
    }{
    
    
				"spec": map[string]interface{
    
    }{
    
    
					"containers": []map[string]interface{
    
    }{
    
    
						{
    
    "name": deploymentName,
							"env": []map[string]string{
    
    {
    
    
								"name":  "RESTART_",
								"value": strconv.FormatInt(time.Now().Unix(), 10),
							}},
						},
					},
				},
			},
		},
	}
	//序列化为字节,因为patch方法只接收字节类型参数
	patchByte, err := json.Marshal(patchData)
	if err != nil {
    
    
		logger.Error(errors.New("json序列化失败, " + err.Error()))
		return errors.New("json序列化失败, " + err.Error())
	}
	//调用patch方法更新deployment
	_, err = K8s.ClientSet.AppsV1().Deployments(namespace).Patch(context.TODO(), deploymentName, "application/strategic-merge-patch+json", patchByte, metav1.PatchOptions{
    
    })
	if err != nil {
    
    
		logger.Error(errors.New("重启Deployment失败, " + err.Error()))
		return errors.New("重启Deployment失败, " + err.Error())
	}
	return nil
}

// 更新deployment
func (d *deployment) UpdateDeployment(namespace, content string) (err error) {
    
    
	var deploy = &appsv1.Deployment{
    
    }

	err = json.Unmarshal([]byte(content), deploy)
	if err != nil {
    
    
		logger.Error(errors.New("反序列化失败, " + err.Error()))
		return errors.New("反序列化失败, " + err.Error())
	}
	_, err = K8s.ClientSet.AppsV1().Deployments(namespace).Update(context.TODO(), deploy, metav1.UpdateOptions{
    
    })
	if err != nil {
    
    
		logger.Error(errors.New("更新Deployment失败, " + err.Error()))
		return errors.New("更新Deployment失败, " + err.Error())
	}
	return nil
}

// 获取每个namespace的deployment数量
func (d *deployment) GetDeployNumPerNp() (deploysNps []*DeploysNp, err error) {
    
    
	namespaceList, err := K8s.ClientSet.CoreV1().Namespaces().List(context.TODO(), metav1.ListOptions{
    
    })
	if err != nil {
    
    
		return nil, err
	}
	for _, namespace := range namespaceList.Items {
    
    
		deploymentList, err := K8s.ClientSet.AppsV1().Deployments(namespace.Name).List(context.TODO(), metav1.ListOptions{
    
    })
		if err != nil {
    
    
			return nil, err
		}

		deploysNp := &DeploysNp{
    
    
			Namespace: namespace.Name,
			DeployNum: len(deploymentList.Items),
		}
		deploysNps = append(deploysNps, deploysNp)
	}
	return deploysNps, nil
}

func (d *deployment) toCells(std []appsv1.Deployment) []DataCell {
    
    
	cells := make([]DataCell, len(std))
	for i := range std {
    
    
		cells[i] = deploymentCell(std[i])
	}
	return cells
}

func (d *deployment) fromCells(cells []DataCell) []appsv1.Deployment {
    
    
	deployments := make([]appsv1.Deployment, len(cells))
	for i := range cells {
    
    
		deployments[i] = appsv1.Deployment(cells[i].(deploymentCell))
	}
	return deployments
}

3. 获取Deployment列表

controller/deployment.go

package controller

import (
	"fmt"
	"k8s-plantform/service"
	"net/http"

	"github.com/gin-gonic/gin"
	"github.com/wonderivan/logger"
)

var Deployment deployment

type deployment struct{
    
    }

// 获取deployment列表,支持过滤、排序、分页
func (d *deployment) GetDeployments(ctx *gin.Context) {
    
    
	params := new(struct {
    
    
		FilterName string `form:"filter_name"`
		Namespace  string `form:"namespace"`
		Page       int    `form:"page"`
		Limit      int    `form:"limit"`
	})
	if err := ctx.Bind(params); err != nil {
    
    
		logger.Error("Bind请求参数失败, " + err.Error())
		ctx.JSON(http.StatusInternalServerError, gin.H{
    
    
			"msg":  err.Error(),
			"data": nil,
		})
		return
	}

	data, err := service.Deployment.GetDeployments(params.FilterName, params.Namespace, params.Limit, params.Page)
	if err != nil {
    
    
		ctx.JSON(http.StatusInternalServerError, gin.H{
    
    
			"msg":  err.Error(),
			"data": nil,
		})
		return
	}

	ctx.JSON(http.StatusOK, gin.H{
    
    
		"msg":  "获取Deployment列表成功",
		"data": data,
	})
}

// 获取deployment详情
func (d *deployment) GetDeploymentDetail(ctx *gin.Context) {
    
    
	params := new(struct {
    
    
		DeploymentName string `form:"deployment_name"`
		Namespace      string `form:"namespace"`
	})
	if err := ctx.Bind(params); err != nil {
    
    
		logger.Error("Bind请求参数失败, " + err.Error())
		ctx.JSON(http.StatusInternalServerError, gin.H{
    
    
			"msg":  err.Error(),
			"data": nil,
		})
		return
	}
	data, err := service.Deployment.GetDeploymentDetail(params.DeploymentName, params.Namespace)
	if err != nil {
    
    
		ctx.JSON(http.StatusInternalServerError, gin.H{
    
    
			"msg":  err.Error(),
			"data": nil,
		})
		return
	}
	ctx.JSON(http.StatusOK, gin.H{
    
    
		"msg":  "获取Deployment详情成功",
		"data": data,
	})
}

// 创建deployment
func (d *deployment) CreateDeployment(ctx *gin.Context) {
    
    
	var (
		deployCreate = new(service.DeployCreate)
		err          error
	)

	if err = ctx.ShouldBindJSON(deployCreate); err != nil {
    
    
		logger.Error("Bind请求参数失败, " + err.Error())
		ctx.JSON(http.StatusInternalServerError, gin.H{
    
    
			"msg":  err.Error(),
			"data": nil,
		})
		return
	}

	if err = service.Deployment.CreateDeployment(deployCreate); err != nil {
    
    
		ctx.JSON(http.StatusInternalServerError, gin.H{
    
    
			"msg":  err.Error(),
			"data": nil,
		})
	}

	ctx.JSON(http.StatusOK, gin.H{
    
    
		"msg":  "创建Deployment成功",
		"data": nil,
	})
}

// 设置deployment副本数
func (d *deployment) ScaleDeployment(ctx *gin.Context) {
    
    
	params := new(struct {
    
    
		DeploymentName string `json:"deployment_name"`
		Namespace      string `json:"namespace"`
		ScaleNum       int    `json:"scale_num"`
	})
	//PUT请求,绑定参数方法改为ctx.ShouldBindJSON
	if err := ctx.ShouldBindJSON(params); err != nil {
    
    
		logger.Error("Bind请求参数失败, " + err.Error())
		ctx.JSON(http.StatusInternalServerError, gin.H{
    
    
			"msg":  err.Error(),
			"data": nil,
		})
		return
	}

	data, err := service.Deployment.ScaleDeployment(params.DeploymentName, params.Namespace, params.ScaleNum)
	if err != nil {
    
    
		ctx.JSON(http.StatusInternalServerError, gin.H{
    
    
			"msg":  err.Error(),
			"data": nil,
		})
		return
	}
	ctx.JSON(http.StatusOK, gin.H{
    
    
		"msg":  "设置Deployment副本数成功",
		"data": fmt.Sprintf("最新副本数: %d", data),
	})
}

// 删除deployment
func (d *deployment) DeleteDeployment(ctx *gin.Context) {
    
    
	params := new(struct {
    
    
		DeploymentName string `json:"deployment_name"`
		Namespace      string `json:"namespace"`
	})
	//DELETE请求,绑定参数方法改为ctx.ShouldBindJSON
	if err := ctx.ShouldBindJSON(params); err != nil {
    
    
		logger.Error("Bind请求参数失败, " + err.Error())
		ctx.JSON(http.StatusInternalServerError, gin.H{
    
    
			"msg":  err.Error(),
			"data": nil,
		})
		return
	}

	err := service.Deployment.DeleteDeployment(params.DeploymentName, params.Namespace)
	if err != nil {
    
    
		ctx.JSON(http.StatusInternalServerError, gin.H{
    
    
			"msg":  err.Error(),
			"data": nil,
		})
		return
	}
	ctx.JSON(http.StatusOK, gin.H{
    
    
		"msg":  "删除Deployment成功",
		"data": nil,
	})
}

// 重启deployment
func (d *deployment) RestartDeployment(ctx *gin.Context) {
    
    
	params := new(struct {
    
    
		DeploymentName string `json:"deployment_name"`
		Namespace      string `json:"namespace"`
	})
	//PUT请求,绑定参数方法改为ctx.ShouldBindJSON
	if err := ctx.ShouldBindJSON(params); err != nil {
    
    
		logger.Error("Bind请求参数失败, " + err.Error())
		ctx.JSON(http.StatusInternalServerError, gin.H{
    
    
			"msg":  err.Error(),
			"data": nil,
		})
		return
	}

	err := service.Deployment.RestartDeployment(params.DeploymentName, params.Namespace)
	if err != nil {
    
    
		ctx.JSON(http.StatusInternalServerError, gin.H{
    
    
			"msg":  err.Error(),
			"data": nil,
		})
		return
	}
	ctx.JSON(http.StatusOK, gin.H{
    
    
		"msg":  "重启Deployment成功",
		"data": nil,
	})
}

// 更新deployment
func (d *deployment) UpdateDeployment(ctx *gin.Context) {
    
    
	params := new(struct {
    
    
		Namespace string `json:"namespace"`
		Content   string `json:"content"`
	})
	//PUT请求,绑定参数方法改为ctx.ShouldBindJSON
	if err := ctx.ShouldBindJSON(params); err != nil {
    
    
		logger.Error("Bind请求参数失败, " + err.Error())
		ctx.JSON(http.StatusInternalServerError, gin.H{
    
    
			"msg":  err.Error(),
			"data": nil,
		})
		return
	}

	err := service.Deployment.UpdateDeployment(params.Namespace, params.Content)
	if err != nil {
    
    
		ctx.JSON(http.StatusInternalServerError, gin.H{
    
    
			"msg":  err.Error(),
			"data": nil,
		})
		return
	}
	ctx.JSON(http.StatusOK, gin.H{
    
    
		"msg":  "更新Deployment成功",
		"data": nil,
	})
}

// 获取每个namespace的pod数量
func (d *deployment) GetDeployNumPerNp(ctx *gin.Context) {
    
    
	data, err := service.Deployment.GetDeployNumPerNp()
	if err != nil {
    
    
		ctx.JSON(http.StatusInternalServerError, gin.H{
    
    
			"msg":  err.Error(),
			"data": nil,
		})
		return
	}

	ctx.JSON(http.StatusOK, gin.H{
    
    
		"msg":  "获取每个namespace的deployment数量成功",
		"data": data,
	})
}

4. 定义路由

controller/router.go

package controller

import (
	"github.com/gin-gonic/gin"
)

// 初始化router类型的对象,首字母大写,用于跨包调用
var Router router

// 声明一个router的结构体
type router struct{
    
    }

func (r *router) InitApiRouter(router *gin.Engine) {
    
    
	router.
		// Pods
		GET("/api/k8s/pods", Pod.GetPods).
		GET("/api/k8s/pods/detail", Pod.GetPodDetail).
		POST("/api/k8s/pods", Pod.DeletePod).
		DELETE("/api/k8s/pod/del", Pod.DeletePod).
		PUT("/api/k8s/pod/update", Pod.UpdatePod).
		GET("/api/k8s/pod/container", Pod.GetPodContainer).
		GET("/api/k8s/pod/log", Pod.GetPodLog).
		GET("/api/k8s/pod/numnp", Pod.GetPodNumPerNp).
	//deployment操作
		GET("/api/k8s/deployments", Deployment.GetDeployments).
		GET("/api/k8s/deployment/detail", Deployment.GetDeploymentDetail).
		PUT("/api/k8s/deployment/scale", Deployment.ScaleDeployment).
		DELETE("/api/k8s/deployment/del", Deployment.DeleteDeployment).
		PUT("/api/k8s/deployment/restart", Deployment.RestartDeployment).
		PUT("/api/k8s/deployment/update", Deployment.UpdateDeployment).
		GET("/api/k8s/deployment/numnp", Deployment.GetDeployNumPerNp).
		POST("/api/k8s/deployment/create", Deployment.CreateDeployment)
}

5. 测试Deployment方法

5.1 获取Deployment清单

/api/k8s/deployments

请添加图片描述

5.2 获取Deployment清单详情

/api/k8s/deployment/detail

请添加图片描述

5.3 修改Deployment副本数

/api/k8s/deployment/scale
root@master-01:~/exam-CKA# kubectl get deployments.apps -A
NAMESPACE     NAME                            READY   UP-TO-DATE   AVAILABLE   AGE
default       nginx-deployment                3/3     3            3           2d20h
kube-system   coredns                         0/2     2            0           2d23h
wework        wework-tomcat-app1-deployment   1/1     1            1           2d21h
wework        wework-tomcat-app2-deployment   1/1     1            1           2d21h

请添加图片描述

root@master-01:~/exam-CKA# kubectl get deployments.apps 
NAME               READY   UP-TO-DATE   AVAILABLE   AGE
nginx-deployment   4/4     4            4           2d20h
root@master-01:~/exam-CKA# kubectl get pods
NAME                                READY   STATUS    RESTARTS   AGE
nginx-deployment-66b6c48dd5-24grz   1/1     Running   0          2d20h
nginx-deployment-66b6c48dd5-cz6ql   1/1     Running   0          31s
nginx-deployment-66b6c48dd5-gttz6   1/1     Running   0          2d20h
nginx-deployment-66b6c48dd5-gxksp   1/1     Running   0          2d20h
test-pod1                           1/1     Running   1          2d23h
test-pod2                           1/1     Running   1          2d23h
test-pod3                           1/1     Running   1          2d23h
test-pod4                           1/1     Running   1          2d23h

5.4 创建Deployment

/api/k8s/deployment/create
root@master-01:~/exam-CKA# kubectl get deployments.apps 
NAME               READY   UP-TO-DATE   AVAILABLE   AGE
nginx-deployment   4/4     4            4           2d20h

这个放到前端一起测

5.5 删除Deployment

/api/k8s/deployment/del
root@master-01:~/exam-CKA/yaml# kubectl get deployments.apps -n k8s
NAME        READY   UP-TO-DATE   AVAILABLE   AGE
front-end   1/1     1            1           11s

请添加图片描述

front-end这个deployment就被删除了

# kubectl get deployments.apps -n k8s
No resources found in k8s namespace.

5.6 重启Deployment

/api/k8s/deployment/restart
root@master-01:~/exam-CKA/yaml# kubectl get deployments.apps,pods
NAME                    READY   UP-TO-DATE   AVAILABLE   AGE
deployment.apps/nginx   3/3     3            3           115s

NAME                         READY   STATUS    RESTARTS   AGE
pod/nginx-64dd8946f7-7c6nj   1/1     Running   0          63s
pod/nginx-64dd8946f7-8wp7f   1/1     Running   0          67s
pod/nginx-64dd8946f7-9f9bn   1/1     Running   0          65s
pod/test-pod1                1/1     Running   1          3d
pod/test-pod2                1/1     Running   1          3d
pod/test-pod3                1/1     Running   1          3d
pod/test-pod4                1/1     Running   1          3d

请添加图片描述

这里可以看到pod已经被重启过了

root@master-01:~/exam-CKA/yaml# kubectl get deployments.apps,pods
NAME                    READY   UP-TO-DATE   AVAILABLE   AGE
deployment.apps/nginx   3/3     3            3           3m15s

NAME                         READY   STATUS    RESTARTS   AGE
pod/nginx-64f8cb6df4-5mtjg   1/1     Running   0          17s
pod/nginx-64f8cb6df4-jkgtq   1/1     Running   0          22s
pod/nginx-64f8cb6df4-qj9dj   1/1     Running   0          19s
pod/test-pod1                1/1     Running   1          3d
pod/test-pod2                1/1     Running   1          3d
pod/test-pod3                1/1     Running   1          3d
pod/test-pod4                1/1     Running   1          3d

5.7 更新Deployment

/api/k8s/deployment/update

这个放到前端一起测.其实之前的restart已经实现了部分功能

5.8 获取每个namespace的Deployment数量

/api/k8s/deployment/numnp

请添加图片描述

猜你喜欢

转载自blog.csdn.net/qq_29974229/article/details/128330912