DKube Add
This commit is contained in:
		@ -3,6 +3,7 @@ package service
 | 
			
		||||
import (
 | 
			
		||||
	appsv1 "k8s.io/api/apps/v1" //引入K8s的APPSv1的包.定义别名为appsv1
 | 
			
		||||
	corev1 "k8s.io/api/core/v1" //引入K8s的corev1的包.定义别名为corev1
 | 
			
		||||
	nwv1 "k8s.io/api/networking/v1"
 | 
			
		||||
	"sort"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"time"
 | 
			
		||||
@ -137,3 +138,24 @@ func (d deploymentCell) GetCreation() time.Time {
 | 
			
		||||
func (d deploymentCell) GetName() string {
 | 
			
		||||
	return d.Name
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type serviceCell corev1.Service
 | 
			
		||||
 | 
			
		||||
func (s serviceCell) GetCreation() time.Time {
 | 
			
		||||
	return s.CreationTimestamp.Time
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (s serviceCell) GetName() string {
 | 
			
		||||
	return s.Name
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
//追加ingress DataCell接口相关代码
 | 
			
		||||
type ingressCell nwv1.Ingress
 | 
			
		||||
 | 
			
		||||
func (i ingressCell) GetCreation() time.Time {
 | 
			
		||||
	return i.CreationTimestamp.Time
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (i ingressCell) GetName() string {
 | 
			
		||||
	return i.Name
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										183
									
								
								service/ingress.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										183
									
								
								service/ingress.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,183 @@
 | 
			
		||||
package service
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"errors"
 | 
			
		||||
	"github.com/wonderivan/logger"
 | 
			
		||||
	nwv1 "k8s.io/api/networking/v1"
 | 
			
		||||
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var Ingress ingress
 | 
			
		||||
 | 
			
		||||
type ingress struct{}
 | 
			
		||||
 | 
			
		||||
type IngressesResp struct {
 | 
			
		||||
	Items []nwv1.Ingress `json:"items"`
 | 
			
		||||
	Total int            `json:"total"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
//定义ServiceCreate结构体,用于创建service需要的参数属性的定义
 | 
			
		||||
type IngressCreate struct {
 | 
			
		||||
	Name      string                 `json:"name"`
 | 
			
		||||
	Namespace string                 `json:"namespace"`
 | 
			
		||||
	Label     map[string]string      `json:"label"`
 | 
			
		||||
	Hosts     map[string][]*HttpPath `json:"hosts"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
//定义ingress的path结构体
 | 
			
		||||
type HttpPath struct {
 | 
			
		||||
	Path        string        `json:"path"`
 | 
			
		||||
	PathType    nwv1.PathType `json:"path_type"`
 | 
			
		||||
	ServiceName string        `json:"service_name"`
 | 
			
		||||
	ServicePort int32         `json:"service_port"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
//获取ingress列表,支持过滤、排序、分页
 | 
			
		||||
func (i *ingress) GetIngresses(filterName, namespace string, limit, page int) (ingressesResp *IngressesResp, err error) {
 | 
			
		||||
	//获取ingressList类型的ingress列表
 | 
			
		||||
	ingressList, err := K8s.ClientSet.NetworkingV1().Ingresses(namespace).List(context.TODO(), metav1.ListOptions{})
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logger.Error(errors.New("获取Ingress列表失败, " + err.Error()))
 | 
			
		||||
		return nil, errors.New("获取Ingress列表失败, " + err.Error())
 | 
			
		||||
	}
 | 
			
		||||
	//将ingressList中的ingress列表(Items),放进dataselector对象中,进行排序
 | 
			
		||||
	selectableData := &dataSelector{
 | 
			
		||||
		GenericDataList: i.toCells(ingressList.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类型的ingress列表转为v1.ingress列表
 | 
			
		||||
	ingresss := i.fromCells(data.GenericDataList)
 | 
			
		||||
 | 
			
		||||
	return &IngressesResp{
 | 
			
		||||
		Items: ingresss,
 | 
			
		||||
		Total: total,
 | 
			
		||||
	}, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
//获取ingress详情
 | 
			
		||||
func (i *ingress) GetIngresstDetail(ingressName, namespace string) (ingress *nwv1.Ingress, err error) {
 | 
			
		||||
	ingress, err = K8s.ClientSet.NetworkingV1().Ingresses(namespace).Get(context.TODO(), ingressName, metav1.GetOptions{})
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logger.Error(errors.New("获取Ingress详情失败, " + err.Error()))
 | 
			
		||||
		return nil, errors.New("获取Ingress详情失败, " + err.Error())
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return ingress, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
//创建ingress
 | 
			
		||||
func (i *ingress) CreateIngress(data *IngressCreate) (err error) {
 | 
			
		||||
	//声明nwv1.IngressRule和nwv1.HTTPIngressPath变量,后面组装数据于鏊用到
 | 
			
		||||
	var ingressRules []nwv1.IngressRule
 | 
			
		||||
	var httpIngressPATHs []nwv1.HTTPIngressPath
 | 
			
		||||
	//将data中的数据组装成nwv1.Ingress对象
 | 
			
		||||
	ingress := &nwv1.Ingress{
 | 
			
		||||
		ObjectMeta: metav1.ObjectMeta{
 | 
			
		||||
			Name:      data.Name,
 | 
			
		||||
			Namespace: data.Namespace,
 | 
			
		||||
			Labels:    data.Label,
 | 
			
		||||
		},
 | 
			
		||||
		Status: nwv1.IngressStatus{},
 | 
			
		||||
	}
 | 
			
		||||
	//第一层for循环是将host组装成nwv1.IngressRule类型的对象
 | 
			
		||||
	// 一个host对应一个ingressrule,每个ingressrule中包含一个host和多个path
 | 
			
		||||
	for key, value := range data.Hosts {
 | 
			
		||||
		ir := nwv1.IngressRule{
 | 
			
		||||
			Host: key,
 | 
			
		||||
			//这里现将nwv1.HTTPIngressRuleValue类型中的Paths置为空,后面组装好数据再赋值
 | 
			
		||||
			IngressRuleValue: nwv1.IngressRuleValue{
 | 
			
		||||
				HTTP: &nwv1.HTTPIngressRuleValue{Paths: nil},
 | 
			
		||||
			},
 | 
			
		||||
		}
 | 
			
		||||
		//第二层for循环是将path组装成nwv1.HTTPIngressPath类型的对象
 | 
			
		||||
		for _, httpPath := range value {
 | 
			
		||||
			hip := nwv1.HTTPIngressPath{
 | 
			
		||||
				Path:     httpPath.Path,
 | 
			
		||||
				PathType: &httpPath.PathType,
 | 
			
		||||
				Backend: nwv1.IngressBackend{
 | 
			
		||||
					Service: &nwv1.IngressServiceBackend{
 | 
			
		||||
						Name: httpPath.ServiceName,
 | 
			
		||||
						Port: nwv1.ServiceBackendPort{
 | 
			
		||||
							Number: httpPath.ServicePort,
 | 
			
		||||
						},
 | 
			
		||||
					},
 | 
			
		||||
				},
 | 
			
		||||
			}
 | 
			
		||||
			//将每个hip对象组装成数组
 | 
			
		||||
			httpIngressPATHs = append(httpIngressPATHs, hip)
 | 
			
		||||
		}
 | 
			
		||||
		//给Paths赋值,前面置为空了
 | 
			
		||||
		ir.IngressRuleValue.HTTP.Paths = httpIngressPATHs
 | 
			
		||||
		//将每个ir对象组装成数组,这个ir对象就是IngressRule,每个元素是一个host和多个path
 | 
			
		||||
		ingressRules = append(ingressRules, ir)
 | 
			
		||||
	}
 | 
			
		||||
	//将ingressRules对象加入到ingress的规则中
 | 
			
		||||
	ingress.Spec.Rules = ingressRules
 | 
			
		||||
	//创建ingress
 | 
			
		||||
	_, err = K8s.ClientSet.NetworkingV1().Ingresses(data.Namespace).Create(context.TODO(), ingress, metav1.CreateOptions{})
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logger.Error(errors.New("创建Ingress失败, " + err.Error()))
 | 
			
		||||
		return errors.New("创建Ingress失败, " + err.Error())
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
//删除ingress
 | 
			
		||||
func (i *ingress) DeleteIngress(ingressName, namespace string) (err error) {
 | 
			
		||||
	err = K8s.ClientSet.NetworkingV1().Ingresses(namespace).Delete(context.TODO(), ingressName, metav1.DeleteOptions{})
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logger.Error(errors.New("删除Ingress失败, " + err.Error()))
 | 
			
		||||
		return errors.New("删除Ingress失败, " + err.Error())
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
//更新ingress
 | 
			
		||||
func (i *ingress) UpdateIngress(namespace, content string) (err error) {
 | 
			
		||||
	var ingress = &nwv1.Ingress{}
 | 
			
		||||
 | 
			
		||||
	err = json.Unmarshal([]byte(content), ingress)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logger.Error(errors.New("反序列化失败, " + err.Error()))
 | 
			
		||||
		return errors.New("反序列化失败, " + err.Error())
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	_, err = K8s.ClientSet.NetworkingV1().Ingresses(namespace).Update(context.TODO(), ingress, metav1.UpdateOptions{})
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logger.Error(errors.New("更新ingress失败, " + err.Error()))
 | 
			
		||||
		return errors.New("更新ingress失败, " + err.Error())
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (i *ingress) toCells(std []nwv1.Ingress) []DataCell {
 | 
			
		||||
	cells := make([]DataCell, len(std))
 | 
			
		||||
	for i := range std {
 | 
			
		||||
		cells[i] = ingressCell(std[i])
 | 
			
		||||
	}
 | 
			
		||||
	return cells
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (i *ingress) fromCells(cells []DataCell) []nwv1.Ingress {
 | 
			
		||||
	ingresss := make([]nwv1.Ingress, len(cells))
 | 
			
		||||
	for i := range cells {
 | 
			
		||||
		ingresss[i] = nwv1.Ingress(cells[i].(ingressCell))
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return ingresss
 | 
			
		||||
}
 | 
			
		||||
@ -1,8 +1,8 @@
 | 
			
		||||
package service
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"github.com/wonderivan/logger"
 | 
			
		||||
	"dkube/config"
 | 
			
		||||
	"github.com/wonderivan/logger"
 | 
			
		||||
	"k8s.io/client-go/kubernetes"
 | 
			
		||||
	"k8s.io/client-go/tools/clientcmd"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
@ -3,11 +3,11 @@ package service
 | 
			
		||||
import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"context"
 | 
			
		||||
	"dkube/config"
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"errors"
 | 
			
		||||
	"github.com/wonderivan/logger"
 | 
			
		||||
	"io"
 | 
			
		||||
	"dkube/config"
 | 
			
		||||
	corev1 "k8s.io/api/core/v1"
 | 
			
		||||
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										161
									
								
								service/service.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										161
									
								
								service/service.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,161 @@
 | 
			
		||||
package service
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"errors"
 | 
			
		||||
	"github.com/wonderivan/logger"
 | 
			
		||||
	corev1 "k8s.io/api/core/v1"
 | 
			
		||||
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 | 
			
		||||
	"k8s.io/apimachinery/pkg/util/intstr"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var Servicev1 servicev1
 | 
			
		||||
 | 
			
		||||
type servicev1 struct{}
 | 
			
		||||
 | 
			
		||||
type ServicesResp struct {
 | 
			
		||||
	Items []corev1.Service `json:"items"`
 | 
			
		||||
	Total int              `json:"total"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type ServiceCreate struct {
 | 
			
		||||
	Name          string            `json:"name"`
 | 
			
		||||
	Namespace     string            `json:"namespace"`
 | 
			
		||||
	Type          string            `json:"type"`
 | 
			
		||||
	ContainerPort int32             `json:"container_port"`
 | 
			
		||||
	Port          int32             `json:"port"`
 | 
			
		||||
	NodePort      int32             `json:"node_port"`
 | 
			
		||||
	Label         map[string]string `json:"label"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
//获取service列表,支持过滤、排序、分页
 | 
			
		||||
func (s *servicev1) GetServices(filterName, namespace string, limit, page int) (servicesResp *ServicesResp, err error) {
 | 
			
		||||
	//获取serviceList类型的service列表
 | 
			
		||||
	serviceList, err := K8s.ClientSet.CoreV1().Services(namespace).List(context.TODO(), metav1.ListOptions{})
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logger.Error(errors.New("获取Service列表失败, " + err.Error()))
 | 
			
		||||
		return nil, errors.New("获取Service列表失败, " + err.Error())
 | 
			
		||||
	}
 | 
			
		||||
	//将serviceList中的service列表(Items),放进dataselector对象中,进行排序
 | 
			
		||||
	selectableData := &dataSelector{
 | 
			
		||||
		GenericDataList: s.toCells(serviceList.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类型的service列表转为v1.service列表
 | 
			
		||||
	services := s.fromCells(data.GenericDataList)
 | 
			
		||||
 | 
			
		||||
	return &ServicesResp{
 | 
			
		||||
		Items: services,
 | 
			
		||||
		Total: total,
 | 
			
		||||
	}, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
//获取service详情
 | 
			
		||||
func (s *servicev1) GetServicetDetail(serviceName, namespace string) (service *corev1.Service, err error) {
 | 
			
		||||
	service, err = K8s.ClientSet.CoreV1().Services(namespace).Get(context.TODO(), serviceName, metav1.GetOptions{})
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logger.Error(errors.New("获取Service详情失败, " + err.Error()))
 | 
			
		||||
		return nil, errors.New("获取Service详情失败, " + err.Error())
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return service, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
//创建service,,接收ServiceCreate对象
 | 
			
		||||
func (s *servicev1) CreateService(data *ServiceCreate) (err error) {
 | 
			
		||||
	//将data中的数据组装成corev1.Service对象
 | 
			
		||||
	service := &corev1.Service{
 | 
			
		||||
		//ObjectMeta中定义资源名、命名空间以及标签
 | 
			
		||||
		ObjectMeta: metav1.ObjectMeta{
 | 
			
		||||
			Name:      data.Name,
 | 
			
		||||
			Namespace: data.Namespace,
 | 
			
		||||
			Labels:    data.Label,
 | 
			
		||||
		},
 | 
			
		||||
		//Spec中定义类型,端口,选择器
 | 
			
		||||
		Spec: corev1.ServiceSpec{
 | 
			
		||||
			Type: corev1.ServiceType(data.Type),
 | 
			
		||||
			Ports: []corev1.ServicePort{
 | 
			
		||||
				{
 | 
			
		||||
					Name:     "http",
 | 
			
		||||
					Port:     data.Port,
 | 
			
		||||
					Protocol: "TCP",
 | 
			
		||||
					TargetPort: intstr.IntOrString{
 | 
			
		||||
						Type:   0,
 | 
			
		||||
						IntVal: data.ContainerPort,
 | 
			
		||||
					},
 | 
			
		||||
				},
 | 
			
		||||
			},
 | 
			
		||||
			Selector: data.Label,
 | 
			
		||||
		},
 | 
			
		||||
	}
 | 
			
		||||
	//默认ClusterIP,这里是判断NodePort,添加配置
 | 
			
		||||
	if data.NodePort != 0 && data.Type == "NodePort" {
 | 
			
		||||
		service.Spec.Ports[0].NodePort = data.NodePort
 | 
			
		||||
	}
 | 
			
		||||
	//创建Service
 | 
			
		||||
	_, err = K8s.ClientSet.CoreV1().Services(data.Namespace).Create(context.TODO(), service, metav1.CreateOptions{})
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logger.Error(errors.New("创建Service失败, " + err.Error()))
 | 
			
		||||
		return errors.New("创建Service失败, " + err.Error())
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
//删除service
 | 
			
		||||
func (s *servicev1) DeleteService(serviceName, namespace string) (err error) {
 | 
			
		||||
	err = K8s.ClientSet.CoreV1().Services(namespace).Delete(context.TODO(), serviceName, metav1.DeleteOptions{})
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logger.Error(errors.New("删除Service失败, " + err.Error()))
 | 
			
		||||
		return errors.New("删除Service失败, " + err.Error())
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
//更新service
 | 
			
		||||
func (s *servicev1) UpdateService(namespace, content string) (err error) {
 | 
			
		||||
	var service = &corev1.Service{}
 | 
			
		||||
 | 
			
		||||
	err = json.Unmarshal([]byte(content), service)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logger.Error(errors.New("反序列化失败, " + err.Error()))
 | 
			
		||||
		return errors.New("反序列化失败, " + err.Error())
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	_, err = K8s.ClientSet.CoreV1().Services(namespace).Update(context.TODO(), service, metav1.UpdateOptions{})
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logger.Error(errors.New("更新service失败, " + err.Error()))
 | 
			
		||||
		return errors.New("更新service失败, " + err.Error())
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (s *servicev1) toCells(std []corev1.Service) []DataCell {
 | 
			
		||||
	cells := make([]DataCell, len(std))
 | 
			
		||||
	for i := range std {
 | 
			
		||||
		cells[i] = serviceCell(std[i])
 | 
			
		||||
	}
 | 
			
		||||
	return cells
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (s *servicev1) fromCells(cells []DataCell) []corev1.Service {
 | 
			
		||||
	services := make([]corev1.Service, len(cells))
 | 
			
		||||
	for i := range cells {
 | 
			
		||||
		services[i] = corev1.Service(cells[i].(serviceCell))
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return services
 | 
			
		||||
}
 | 
			
		||||
		Reference in New Issue
	
	Block a user