DKube Add

This commit is contained in:
dqzboy
2022-09-25 20:02:35 +08:00
parent 84c4f484e2
commit be038dd5e7
12 changed files with 505 additions and 189 deletions

View File

@ -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
View 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
}

View File

@ -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"
)

View File

@ -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
View 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
}