Skip to content

DaemonSet管理 原创

1、功能

3f629d25fe5f2084406b6d4620a0a328 MD5

2、类型转换

2.1、实现类型转换

internal/pkg/k8s/daemonset/common.go 文件中新增 DaemonSetCell,实现和 appv1.DaemonSetCell 进行类型转换,如下:

go
package daemonset

import (
	appv1 "k8s.io/api/apps/v1"
	"time"

	"github.com/joker-bai/hawkeye/internal/pkg/k8s/dataselect"
)

type DaemonSetCell appv1.DaemonSet

func (p DaemonSetCell) GetCreation() time.Time {
	return p.CreationTimestamp.Time
}

func (p DaemonSetCell) GetName() string {
	return p.Name
}

// toCells corev1.Pod 类型 转换成 DataCell 类型
// @description: Pod类型转换成DataCell
func toCells(sts []appv1.DaemonSet) []dataselect.DataCell {
	cells := make([]dataselect.DataCell, len(sts))
	for i := range sts {
		cells[i] = DaemonSetCell(sts[i])
	}
	return cells
}

// fromCells DataCell 类型转换成 corev1.Pod 类型
// @description: DataCell类型转换成Pod
func fromCells(cells []dataselect.DataCell) []appv1.DaemonSet {
	ds := make([]appv1.DaemonSet, len(cells))
	for i := range cells {
		ds[i] = appv1.DaemonSet(cells[i].(DaemonSetCell))
	}
	return ds
}

2.2、实现增删改查

(1)创建 internal/pkg/k8s/daemonset/create.go 文件,输入以下内容,用于创建 daemonset

go
package daemonset

import (
	"context"
	"github.com/joker-bai/hawkeye/global"
	"github.com/joker-bai/hawkeye/internal/app/requests"
	"github.com/joker-bai/hawkeye/internal/pkg/k8s/common"
	appv1 "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"
	"strings"
)

const (
	DescriptionAnnotationKey = "description"
)

func CreateDaemonSet(ds *requests.K8sDaemonSetCreateRequest) error {
	annotations := map[string]string{}
	if ds.Description != nil {
		annotations[DescriptionAnnotationKey] = *ds.Description
	}

	labels := common.GetLabelsMap(ds.Labels)

	metadata := metav1.ObjectMeta{
		Annotations: annotations,
		Labels:      labels,
		Name:        ds.Name,
	}

	containerSpec := corev1.Container{
		Name:  ds.Name,
		Image: ds.ContainerImage,
		SecurityContext: &corev1.SecurityContext{
			Privileged: &ds.RunAsPrivileged,
		},
		Resources: corev1.ResourceRequirements{Requests: make(map[corev1.ResourceName]resource.Quantity)},
		Env:       common.ConvertEnvVarsSpec(ds.Variables),
	}

	if ds.ContainerCommand != nil {
		containerSpec.Command = []string{*ds.ContainerCommand}
	}
	if ds.ContainerCommandArgs != nil {
		containerSpec.Args = strings.Fields(*ds.ContainerCommandArgs)
	}
	if ds.MemoryRequirement != nil {
		containerSpec.Resources.Requests[corev1.ResourceMemory] = resource.MustParse(*ds.MemoryRequirement)
	}
	if ds.CpuRequirement != nil {
		containerSpec.Resources.Requests[corev1.ResourceCPU] = resource.MustParse(*ds.CpuRequirement)
	}

	// 是否开启Readiness健康检测
	if ds.IsReadinessEnable {
		probeHandler := common.GetContainerProbe(ds.ReadinessProbe)
		containerSpec.ReadinessProbe = &corev1.Probe{
			ProbeHandler:        probeHandler,
			InitialDelaySeconds: ds.ReadinessProbe.InitialDelaySeconds,
			TimeoutSeconds:      ds.ReadinessProbe.TimeoutSeconds,
			PeriodSeconds:       ds.ReadinessProbe.PeriodSeconds,
			SuccessThreshold:    ds.ReadinessProbe.SuccessThreshold,
			FailureThreshold:    ds.ReadinessProbe.FailureThreshold,
		}
	}
	if ds.IsLivenessEnable {
		probehandler := common.GetContainerProbe(ds.ReadinessProbe)

		containerSpec.LivenessProbe = &corev1.Probe{
			ProbeHandler:        probehandler,
			InitialDelaySeconds: ds.LivenessProbe.InitialDelaySeconds,
			TimeoutSeconds:      ds.LivenessProbe.TimeoutSeconds,
			PeriodSeconds:       ds.LivenessProbe.PeriodSeconds,
			SuccessThreshold:    ds.LivenessProbe.SuccessThreshold,
			FailureThreshold:    ds.LivenessProbe.FailureThreshold,
		}
	}

	podSpec := corev1.PodSpec{Containers: []corev1.Container{containerSpec}}

	if ds.ImagePullSecret != nil {
		podSpec.ImagePullSecrets = []corev1.LocalObjectReference{{Name: *ds.ImagePullSecret}}
	}

	podTemplate := corev1.PodTemplateSpec{
		ObjectMeta: metadata,
		Spec:       podSpec,
	}

	dsment := &appv1.DaemonSet{
		ObjectMeta: metadata,
		Spec: appv1.DaemonSetSpec{
			Template: podTemplate,
			Selector: &metav1.LabelSelector{
				MatchLabels: labels,
			},
		},
	}

	if _, err := global.K8S.AppsV1().DaemonSets(ds.Namespace).Create(context.TODO(), dsment, metav1.CreateOptions{}); err != nil {
		return err
	}

	// 如果需要创建Service,执行以下代码

	if ds.IsCreateService {
		// TODO
	}

	return nil
}

(2)在 internal/pkg/k8s/daemonset/delete.go 中创建以下内容,用户删除 daemonset

go
package daemonset

import (
	"context"
	"github.com/joker-bai/hawkeye/global"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)

func DeleteDaemonSet(name, namespace string) error {
	return global.K8S.AppsV1().DaemonSets(namespace).Delete(context.TODO(), name, metav1.DeleteOptions{})
}

(3)在 internal/pkg/k8s/daemonset/list.go 中创建以下内容,用于列出 daemonset 列表

go
package daemonset

import (
	"context"
	"github.com/joker-bai/hawkeye/global"
	"github.com/joker-bai/hawkeye/internal/pkg/k8s/dataselect"
	appv1 "k8s.io/api/apps/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)

func ListDaemonSet(name, namespace string, page, limit int) ([]appv1.DaemonSet, error) {
	sts, err := global.K8S.AppsV1().DaemonSets(namespace).List(context.TODO(), metav1.ListOptions{})
	if err != nil {
		return nil, err
	}

	// 做排序
	selector := dataselect.DataSelector{
		GenericDataList: toCells(sts.Items),
		DataSelectQuery: &dataselect.DataSelectQuery{
			Filter: &dataselect.FilterQuery{
				Name: name,
			},
			Paginate: &dataselect.PaginateQuery{
				Limit: limit,
				Page:  page,
			},
		},
	}

	filted := selector.Filter()
	data := filted.Sort().Paginate()
	return fromCells(data.GenericDataList), nil
}

(4)在 internal/pkg/k8s/daemonset/update.go 中创建以下内容,用于更新 daemonset

go
package daemonset

import (
	"context"
	"encoding/json"
	"github.com/joker-bai/hawkeye/global"
	appv1 "k8s.io/api/apps/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)

func UpdateDaemonSet(namespace, content string) error {
	var sts appv1.DaemonSet
	if err := json.Unmarshal([]byte(content), &sts); err != nil {
		return err
	}

	if _, err := global.K8S.AppsV1().DaemonSets(namespace).Update(context.TODO(), &sts, metav1.UpdateOptions{}); err != nil {
		return err
	}

	return nil
}

(5)在 internal/pkg/k8s/daemonset/detail.go 中创建以下内容,用于获取 daemonset 详情

go
package daemonset

import (
	"context"
	"github.com/joker-bai/hawkeye/global"
	appv1 "k8s.io/api/apps/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)

func GetDaemonSetDetail(name, namespace string) (*appv1.DaemonSet, error) {
	sts, err := global.K8S.AppsV1().DaemonSets(namespace).Get(context.TODO(), name, metav1.GetOptions{})
	if err != nil {
		return nil, err
	}
	return sts, nil
}

(6)在 internal/pkg/k8s/daemonset/restart.go 中创建以下内容,用于重启 daemonset

go
package daemonset

import (
	"context"
	"encoding/json"
	"github.com/joker-bai/hawkeye/global"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/apimachinery/pkg/types"
	"strconv"
	"time"
)

func RestartDaemonSet(name, namespace string) error {
	// 组装数据
	patchData := map[string]interface{}{
		"spec": map[string]interface{}{
			"template": map[string]interface{}{
				"containers": []map[string]interface{}{
					{
						"name": name,
						"env": []map[string]string{
							{
								"name":  "RESTART_",
								"value": strconv.FormatInt(time.Now().Unix(), 10),
							},
						},
					},
				},
			},
		},
	}

	patchBytes, err := json.Marshal(patchData)
	if err != nil {
		return err
	}

	// 调用patch更新DaemonSet
	_, err = global.K8S.AppsV1().DaemonSets(namespace).Patch(context.TODO(), name, types.MergePatchType, patchBytes, metav1.PatchOptions{})
	if err != nil {
		return err
	}
	return nil
}

(7)在 internal/pkg/k8s/daemonset/pods.go 中新增以下内容,用户获取 daemonset 的 Pod

go
package daemonset

import (
	"context"
	"github.com/joker-bai/hawkeye/global"
	corev1 "k8s.io/api/core/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)

func GetPods(name, namespace string) ([]corev1.Pod, error) {
	d, err := global.K8S.AppsV1().DaemonSets(namespace).Get(context.TODO(), name, metav1.GetOptions{})
	if err != nil {
		return nil, err
	}
	// 先获取labelSelector
	label := d.Spec.Selector.String()

	var pods *corev1.PodList
	// 根据label获取Pod
	pods, err = global.K8S.CoreV1().Pods(namespace).List(context.TODO(), metav1.ListOptions{LabelSelector: label})
	if err != nil {
		return nil, err
	}
	return pods.Items, nil
}

3、实现 services 方法

3.1、请求参数校验

internal/app/requests 目录中新建 k8s_daemonset.go 文件,写入以下内容以完成请求参数校验:

go
package requests

import (
	"github.com/gin-gonic/gin"
	"github.com/joker-bai/hawkeye/pkg/app"
	"github.com/thedevsaddam/govalidator"
)

type K8sDaemonSetCreateRequest struct {
	Name                 string                `json:"name" valid:"name"`                                     // DaemonSet的名字
	ContainerImage       string                `json:"container_image" valid:"container_image"`               // 容器镜像名
	ImagePullSecret      *string               `json:"image_pull_secret" valid:"image_pull_secret"`           // 拉取镜像的密钥
	ContainerCommand     *string               `json:"container_command" valid:"container_command"`           // 容器启动时执行的命令
	ContainerCommandArgs *string               `json:"container_command_args" valid:"container_command_args"` // 容器启动时执行命令的参数
	PortMappings         []PortMapping         `json:"port_mappings" valid:"port_mappings"`                   // 容器端口
	Variables            []EnvironmentVariable `json:"variables" valid:"variables"`                           // 环境变量
	IsCreateService      bool                  `json:"is_create_service" valid:"is_create_service"`           // 是否创建Service
	Description          *string               `json:"description" valid:"description"`                       // 描述
	Namespace            string                `json:"namespace" valid:"namespace"`                           // 名称空间
	MemoryRequirement    *string               `json:"memory_requirement" valid:"memory_requirement"`         // 内存
	CpuRequirement       *string               `json:"cpu_requirement" valid:"cpu_requirement"`               // CPU
	Labels               []Label               `json:"labels" valid:"labels"`                                 // 标签
	RunAsPrivileged      bool                  `json:"run_as_privileged" valid:"run_as_privileged"`           // 是否特权用户运行容器
	IsReadinessEnable    bool                  `json:"is_readiness_enable" valid:"is_readiness_enable"`       // 是否开启ReadinessProbe健康检查
	ReadinessProbe       HealthCheckDetail     `json:"readiness_probe" valid:"readiness_probe"`               // ReadinessProbe配置
	IsLivenessEnable     bool                  `json:"is_liveness_enable" valid:"is_liveness_enable"`         // 是否开启LivenessProbe健康检查
	LivenessProbe        HealthCheckDetail     `json:"liveness_probe" valid:"liveness_probe"`                 // LivenessProbe 主要配置
}

func ValidK8sDaemonSetCreateRequest(data interface{}, ctx *gin.Context) map[string][]string {
	rules := govalidator.MapData{
		"name":            []string{"required"},
		"namespace":       []string{"required"},
		"container_image": []string{"required"},
		"replicas":        []string{"required"},
	}

	messages := govalidator.MapData{
		"namespace": []string{
			"required: namespace不能为空",
		},
		"name": []string{
			"required: name不能为空",
		},
		"container_image": []string{
			"required: image不能为空",
		},
		"replicas": []string{
			"required: replicas不能为空",
		},
	}

	return app.ValidateOptions(data, rules, messages)
}

type K8sDaemonSetUpdateRequest struct {
	Namespace string `json:"namespace" valid:"namespace"`
	Content   string `json:"content" valid:"content"`
}

func ValidK8sDaemonSetUpdateRequest(data interface{}, ctx *gin.Context) map[string][]string {
	rules := govalidator.MapData{
		"namespace": []string{"required"},
		"content":   []string{"required"},
	}
	messages := govalidator.MapData{
		"namespace": []string{
			"required: namespace 不能为空",
		},
		"content": []string{
			"required: content 不能为空",
		},
	}

	// 校验入参

	return app.ValidateOptions(data, rules, messages)
}

type K8sDaemonSetListRequest struct {
	K8sCommonRequest
	Page  int `json:"page" valid:"page" valid:"page"`    // 页数
	Limit int `json:"limit" valid:"limit" valid:"limit"` // 每页条数
}

func ValidK8sDaemonSetListRequest(data interface{}, ctx *gin.Context) map[string][]string {
	rules := govalidator.MapData{
		"namespace": []string{"required"},
		"page":      []string{"required"},
		"limit":     []string{"required"},
	}
	messages := govalidator.MapData{
		"namespace": []string{
			"required: namespace不能为空",
		},
		"page": []string{
			"required: page不能为空",
		},
		"limit": []string{
			"required: limit不能为空",
		},
	}

	// 校验入参

	return app.ValidateOptions(data, rules, messages)
}

3.2、实现 services 方法

internal/app/services/k8s_daemonset.go 文件中新增 DaemonSet 操作的 services 方法,如下:

go
package services

import (
	"github.com/joker-bai/hawkeye/internal/app/requests"
	"github.com/joker-bai/hawkeye/internal/pkg/k8s/daemonset"
	appv1 "k8s.io/api/apps/v1"
	corev1 "k8s.io/api/core/v1"
)

// DaemonSet Service

func (s *Services) K8sDaemonSetList(param *requests.K8sDaemonSetListRequest) ([]appv1.DaemonSet, error) {
	return daemonset.ListDaemonSet(param.Name, param.Namespace, param.Page, param.Limit)
}

func (s *Services) K8sDaemonSetDelete(param *requests.K8sCommonRequest) error {
	return daemonset.DeleteDaemonSet(param.Name, param.Namespace)
}

func (s *Services) K8sDaemonSetUpdate(param *requests.K8sDaemonSetUpdateRequest) error {
	return daemonset.UpdateDaemonSet(param.Namespace, param.Content)
}

func (s *Services) K8sDaemonSetCreate(param *requests.K8sDaemonSetCreateRequest) error {
	return daemonset.CreateDaemonSet(param)
}

func (s *Services) K8sDaemonSetRestart(param *requests.K8sCommonRequest) error {
	return daemonset.RestartDaemonSet(param.Name, param.Namespace)
}

func (s *Services) K8sDaemonSetGetPod(param *requests.K8sCommonRequest) ([]corev1.Pod, error) {
	return daemonset.GetPods(param.Name, param.Namespace)
}

func (s *Services) K8sDaemonSetDetail(param *requests.K8sCommonRequest) (*appv1.DaemonSet, error) {
	return daemonset.GetDaemonSetDetail(param.Name, param.Namespace)
}

4、新增 controllers 方法

在 internal/app/controllers/api/v1/k8s 目录中新增 daemonset.go 文件,实现如下方法:

go
package k8s

import (
	"github.com/gin-gonic/gin"
	"github.com/joker-bai/hawkeye/global"
	"github.com/joker-bai/hawkeye/internal/app/requests"
	"github.com/joker-bai/hawkeye/internal/app/services"
	"github.com/joker-bai/hawkeye/pkg/app"
	"github.com/joker-bai/hawkeye/pkg/errorcode"
	"go.uber.org/zap"
)

type DaemonSetController struct{}

// List godoc
// @Summary 列出K8s DaemonSet
// @Description 列出K8s DaemonSet
// @Tags K8s DaemonSet管理
// @Produce json
// @Param name query string false "DaemonSet名" maxlength(100)
// @Param namespace query string false "命名空间" maxlength(100)
// @Param page query int true "页码"
// @Param limit query int true "每页数量"
// @Success 200 {object} string "成功"
// @Failure 400 {object} errorcode.Error "请求错误"
// @Failure 500 {object} errorcode.Error "内部错误"
// @Router /api/v1/k8s/daemonset/list [get]
func (k *DaemonSetController) List(ctx *gin.Context) {
	param := requests.K8sDaemonSetListRequest{}
	response := app.NewResponse(ctx)

	if ok := app.Validate(ctx, &param, requests.ValidK8sDaemonSetListRequest); !ok {
		return
	}

	svc := services.New(ctx)
	daemonsets, err := svc.K8sDaemonSetList(&param)
	if err != nil {
		global.Log.Error("获取DaemonSet列表失败", zap.String("error", err.Error()))
		response.ToErrorResponse(errorcode.ErrorK8sDaemonSetListFail)
		return
	}

	response.ToResponseList(daemonsets, len(daemonsets))
}

// Update godoc
// @Summary 更新DaemonSet
// @Description 更新DaemonSet
// @Tags K8s DaemonSet管理
// @Produce json
// @Param body body requests.K8sDaemonSetUpdateRequest true "body"
// @Success 200 {object} string "成功"
// @Failure 400 {object} errorcode.Error "请求错误"
// @Failure 500 {object} errorcode.Error "内部错误"
// @Router /api/v1/k8s/daemonset/update [post]
func (k *DaemonSetController) Update(ctx *gin.Context) {
	param := requests.K8sDaemonSetUpdateRequest{}
	response := app.NewResponse(ctx)

	if ok := app.Validate(ctx, &param, requests.ValidK8sDaemonSetUpdateRequest); !ok {
		return
	}

	svc := services.New(ctx)
	err := svc.K8sDaemonSetUpdate(&param)
	if err != nil {
		global.Log.Error("更新DaemonSet失败", zap.String("error", err.Error()))
		response.ToErrorResponse(errorcode.ErrorK8sDaemonSetUpdateFail)
		return
	}

	response.ToResponse(gin.H{
		"msg": "DaemonSet更新成功",
	})
}

// Delete godoc
// @Summary 删除DaemonSet
// @Description 删除DaemonSet
// @Tags K8s DaemonSet管理
// @Produce json
// @Param body body requests.K8sCommonRequest true "body"
// @Success 200 {object} string "成功"
// @Failure 400 {object} errorcode.Error "请求错误"
// @Failure 500 {object} errorcode.Error "内部错误"
// @Router /api/v1/k8s/daemonset/delete [post]
func (k *DaemonSetController) Delete(ctx *gin.Context) {
	param := requests.K8sCommonRequest{}
	response := app.NewResponse(ctx)

	if ok := app.Validate(ctx, &param, requests.ValidK8sCommonRequest); !ok {
		return
	}

	svc := services.New(ctx)
	err := svc.K8sDaemonSetDelete(&param)
	if err != nil {
		global.Log.Error("删除DaemonSets失败", zap.String("error", err.Error()))
		response.ToErrorResponse(errorcode.ErrorK8sDaemonSetDeleteFail)
		return
	}

	response.ToResponse(gin.H{
		"msg": "DaemonSet删除成功",
	})
}

// Create godoc
// @Summary 创建DaemonSet
// @Description 创建DaemonSet
// @Tags K8s DaemonSet管理
// @Produce json
// @Param body body requests.K8sDaemonSetCreateRequest true "body"
// @Success 200 {object} string "成功"
// @Failure 400 {object} errorcode.Error "请求错误"
// @Failure 500 {object} errorcode.Error "内部错误"
// @Router /api/v1/k8s/daemonset/create [post]
func (k *DaemonSetController) Create(ctx *gin.Context) {
	param := requests.K8sDaemonSetCreateRequest{}
	response := app.NewResponse(ctx)

	if ok := app.Validate(ctx, &param, requests.ValidK8sDaemonSetCreateRequest); !ok {
		return
	}

	svc := services.New(ctx)
	err := svc.K8sDaemonSetCreate(&param)
	if err != nil {
		global.Log.Error("创建DaemonSet失败", zap.String("error", err.Error()))
		response.ToErrorResponse(errorcode.ErrorK8sDaemonSetCreateFail)
		return
	}

	response.ToResponse(gin.H{
		"msg": "DaemonSet创建成果",
	})
}

// Restart godoc
// @Summary 重启DaemonSet
// @Description 重启DaemonSet
// @Tags K8s DaemonSet管理
// @Produce json
// @Param body body requests.K8sCommonRequest true "body"
// @Success 200 {object} string "成功"
// @Failure 400 {object} errorcode.Error "请求错误"
// @Failure 500 {object} errorcode.Error "内部错误"
// @Router /api/v1/k8s/daemonset/restart [post]
func (k *DaemonSetController) Restart(ctx *gin.Context) {
	param := requests.K8sCommonRequest{}
	response := app.NewResponse(ctx)

	if ok := app.Validate(ctx, &param, requests.ValidK8sCommonRequest); !ok {
		return
	}

	svc := services.New(ctx)
	err := svc.K8sDaemonSetRestart(&param)
	if err != nil {
		global.Log.Error("重启DaemonSet失败", zap.String("error", err.Error()))
		response.ToErrorResponse(errorcode.ErrorK8sDaemonSetRestartFail)
		return
	}

	response.ToResponse(gin.H{
		"msg": "重启DaemonSet成功",
	})
}

// GetPods godoc
// @Summary 获取DaemonSet的Pod
// @Description 获取DaemonSet的Pod
// @Tags K8s DaemonSet管理
// @Produce json
// @Param name query string false "DaemonSet名" maxlength(100)
// @Param namespace query string false "命名空间" maxlength(100)
// @Param container query string false "容器" maxlength(100)
// @Success 200 {object} string "成功"
// @Failure 400 {object} errorcode.Error "请求错误"
// @Failure 500 {object} errorcode.Error "内部错误"
// @Router /api/v1/k8s/daemonset/pods [get]
func (k *DaemonSetController) GetPods(ctx *gin.Context) {
	param := requests.K8sCommonRequest{}
	response := app.NewResponse(ctx)

	if ok := app.Validate(ctx, &param, requests.ValidCommonIdRequest); !ok {
		return
	}

	svc := services.New(ctx)
	pods, err := svc.K8sDaemonSetGetPod(&param)
	if err != nil {
		global.Log.Error("获取获取DaemonSet的Pod失败", zap.String("error", err.Error()))
		response.ToErrorResponse(errorcode.ErrorK8sDaemonSetGetPodsFail)
		return
	}

	response.ToResponse(gin.H{
		"data": pods,
		"msg":  "获取DaemonSet的Pod成功",
	})
}

// Detail godoc
// @Summary 获取DaemonSet的详情
// @Description 获取DaemonSet的详情
// @Tags K8s DaemonSet管理
// @Produce json
// @Param name query string false "Job名" maxlength(100)
// @Param namespace query string false "命名空间" maxlength(100)
// @Success 200 {object} string "成功"
// @Failure 400 {object} errorcode.Error "请求错误"
// @Failure 500 {object} errorcode.Error "内部错误"
// @Router /api/v1/k8s/daemonset/detail [get]
func (k *DaemonSetController) Detail(ctx *gin.Context) {
	param := requests.K8sCommonRequest{}
	response := app.NewResponse(ctx)

	if ok := app.Validate(ctx, &param, requests.ValidK8sCommonRequest); !ok {
		return
	}

	svc := services.New(ctx)
	ds, err := svc.K8sDaemonSetDetail(&param)
	if err != nil {
		global.Log.Error("获取获取DaemonSet的详情失败", zap.String("error", err.Error()))
		response.ToErrorResponse(errorcode.ErrorK8sDaemonSetDetailFail)
		return
	}

	response.ToResponse(gin.H{
		"data": ds,
		"msg":  "获取DaemonSet的详情成功",
	})
}

再到 pkg/errorcode/k8s.go 文件中新增如下错误代码:

go
package errorcode

var (
	......
	// K8s DaemonSet 错误码

	ErrorK8sDaemonSetUpdateFail  = NewError(500041, "更新K8s DaemonSet 失败")
	ErrorK8sDaemonSetDeleteFail  = NewError(500042, "删除K8s DaemonSet 失败")
	ErrorK8sDaemonSetListFail    = NewError(500043, "获取K8s DaemonSet 列表失败")
	ErrorK8sDaemonSetDetailFail  = NewError(500044, "获取K8s DaemonSet 详情失败")
	ErrorK8sDaemonSetRestartFail = NewError(500045, "重启K8s DaemonSet 失败")
	ErrorK8sDaemonSetGetPodsFail = NewError(500046, "获取K8s DaemonSet 的Pod失败")
	ErrorK8sDaemonSetCreateFail  = NewError(500047, "创建K8s DaemonSet 失败")
)

5、新增路由

internal/app/routers/k8s.go 文件中新增 DaemonSet 操作的路由,如下:

go
package routers

import (
	"github.com/gin-gonic/gin"
	v1 "github.com/joker-bai/kubemana/internal/app/controllers/api/v1"
)

type K8sRouter struct{}

func (r *K8sRouter) Inject(router *gin.RouterGroup) {

	k8s := router.Group("/k8s")
	{
		......

		// Daemonset 管理
		dac := new(k8s.DaemonSetController)
		ks.GET("/daemonset/list", dac.List)
		ks.GET("/daemonset/detail", dac.Detail)
		ks.POST("/daemonset/create", dac.Create)
		ks.POST("/daemonset/update", dac.Update)
		ks.POST("/daemonset/delete", dac.Delete)
		ks.POST("/daemonset/restart", dac.Restart)
		ks.GET("/daemonset/pods", dac.GetPods)
	}
}

6、测试一下

PS:测试之前都需要先初始化集群,在 4.3.1 Pod 章节有介绍。

这里简单测试列出 DaemonSet 接口,如下:

96dfd7909edec4c348f0d368e02a6892 MD5

其他接口自行下去测试。

7、代码版本

本节开发完成后,记得生成 swag 和标记代码版本,如下:

go
$ swag init
$ git add .
$ git commit -m "新增k8s集群daemonset操作"
最近更新