DKube Init
This commit is contained in:
dqzboy
161
service/dataselector.go
Normal file
161
service/dataselector.go
Normal file
@ -0,0 +1,161 @@
|
||||
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"
|
||||
)
|
||||
|
||||
//定义结构体：dataSelector，用于封装排序、过滤、分页的数据类型
|
||||
type dataSelector struct { //包内调用，首字母小写
|
||||
GenericDataList []DataCell
|
||||
DataSelect *DataSelectQuery
|
||||
}
|
||||
|
||||
// DataCell DataCell接口用于各种资源List类型的转换，转换后可以使用dataSelector的排序、过滤、分页方法
|
||||
type DataCell interface {
|
||||
GetCreation() time.Time
|
||||
GetName() string
|
||||
}
|
||||
|
||||
// DataSelectQuery 定义过滤和分页结构体,过滤使用Name过滤，分页使用Limit和page
|
||||
//Limit 是单页的数据条数
|
||||
//Page是第几页
|
||||
type DataSelectQuery struct {
|
||||
Filter *FilterQuery
|
||||
Paginate *PaginateQuery
|
||||
}
|
||||
|
||||
type FilterQuery struct {
|
||||
Name string
|
||||
}
|
||||
|
||||
type PaginateQuery struct {
|
||||
Limit int
|
||||
Page int
|
||||
}
|
||||
|
||||
// Len 实现自定义结构体的排序，需要重写Len、Swap、Less方法
|
||||
//Len方法用于获取数组的长度
|
||||
func (d *dataSelector) Len() int {
|
||||
return len(d.GenericDataList)
|
||||
}
|
||||
|
||||
// Swap 方法用于数据比较大小之后的位置变更
|
||||
func (d *dataSelector) Swap(i, j int) {
|
||||
//临时变量对调 i,j ==> j.i
|
||||
d.GenericDataList[i], d.GenericDataList[j] = d.GenericDataList[j], d.GenericDataList[i]
|
||||
}
|
||||
|
||||
// Less 方法用于比较大小
|
||||
func (d *dataSelector) Less(i, j int) bool {
|
||||
a := d.GenericDataList[i].GetCreation()
|
||||
b := d.GenericDataList[j].GetCreation()
|
||||
//比较B的时间是否在A之前，触发位置调换
|
||||
return b.Before(a)
|
||||
}
|
||||
|
||||
//重写以上三个方法，使用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
|
||||
}
|
||||
//如果不为空，则按照入参Name进行过滤
|
||||
//声明一个新的数组。若Name包含，则把数据放进数组，返回出去
|
||||
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和Page的入参，定义快捷变量
|
||||
limit := d.DataSelect.Paginate.Limit
|
||||
page := d.DataSelect.Paginate.Page
|
||||
|
||||
//检验参数的合法性
|
||||
if limit <= 0 || page <= 0 {
|
||||
return d
|
||||
}
|
||||
|
||||
//定义取数范围需要的starIndex和endIndex
|
||||
startIndex := limit * (page - 1)
|
||||
endIndex := limit*page - 1
|
||||
|
||||
//出来endIndex
|
||||
if endIndex > len(d.GenericDataList) {
|
||||
endIndex = len(d.GenericDataList)
|
||||
}
|
||||
|
||||
d.GenericDataList = d.GenericDataList[startIndex:endIndex]
|
||||
|
||||
return d
|
||||
}
|
||||
|
||||
// 定义podCell类型，重写GetCreateion和GetName方法后，可进行数据转换
|
||||
//corev1.Pod --> podCell --> DataCell
|
||||
//appsv1.Deployment --> deployCell --> DataCell
|
||||
type podCell corev1.Pod
|
||||
|
||||
// GetCreation 重写DataCell接口的两个方法
|
||||
func (p podCell) GetCreation() time.Time {
|
||||
return p.CreationTimestamp.Time
|
||||
}
|
||||
|
||||
func (p podCell) GetName() string {
|
||||
return p.Name
|
||||
}
|
||||
|
||||
type deploymentCell appsv1.Deployment
|
||||
|
||||
func (d deploymentCell) GetCreation() time.Time {
|
||||
return d.CreationTimestamp.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
|
||||
}
|
||||
307
service/deployment.go
Normal file
307
service/deployment.go
Normal file
@ -0,0 +1,307 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"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"
|
||||
"strconv"
|
||||
"time"
|
||||
)
|
||||
|
||||
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) {
|
||||
//初始化appsv1.Deployment类型的对象，并将入参的data数据放进去
|
||||
deployment := &appsv1.Deployment{
|
||||
ObjectMeta: metav1.ObjectMeta{
|
||||
Name: data.Name,
|
||||
Namespace: data.Namespace,
|
||||
Labels: data.Label,
|
||||
},
|
||||
Spec: appsv1.DeploymentSpec{
|
||||
Replicas: &data.Replicas,
|
||||
Selector: &metav1.LabelSelector{
|
||||
MatchLabels: data.Label,
|
||||
},
|
||||
Template: corev1.PodTemplateSpec{
|
||||
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{},
|
||||
}
|
||||
//判断健康检查功能是否打开，若打开，则增加健康检查功能
|
||||
if data.HealthCheck {
|
||||
deployment.Spec.Template.Spec.Containers[0].ReadinessProbe = &corev1.Probe{
|
||||
ProbeHandler: corev1.ProbeHandler{
|
||||
HTTPGet: &corev1.HTTPGetAction{
|
||||
Path: data.HealthPath,
|
||||
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(deployments []appsv1.Deployment) []DataCell {
|
||||
cells := make([]DataCell, len(deployments))
|
||||
for i := range deployments {
|
||||
cells[i] = deploymentCell(deployments[i])
|
||||
}
|
||||
return cells
|
||||
}
|
||||
|
||||
func (d *deployment) fromCells(cells []DataCell) []appsv1.Deployment {
|
||||
deployments := make([]appsv1.Deployment, len(cells))
|
||||
for i := range cells {
|
||||
//cells[i].(podCell)是将DataCell类型转成podCell
|
||||
deployments[i] = appsv1.Deployment(cells[i].(deploymentCell))
|
||||
}
|
||||
return deployments
|
||||
}
|
||||
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
|
||||
}
|
||||
32
service/init.go
Normal file
32
service/init.go
Normal file
@ -0,0 +1,32 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"dkube/config"
|
||||
"github.com/wonderivan/logger"
|
||||
"k8s.io/client-go/kubernetes"
|
||||
"k8s.io/client-go/tools/clientcmd"
|
||||
)
|
||||
|
||||
//用于初始化k8s clientset
|
||||
var K8s k8s
|
||||
|
||||
type k8s struct {
|
||||
ClientSet *kubernetes.Clientset
|
||||
}
|
||||
|
||||
//初始化方法
|
||||
func (k *k8s) Init() {
|
||||
conf, err := clientcmd.BuildConfigFromFlags("", config.Kubeconfig)
|
||||
if err != nil {
|
||||
panic("获取k8s clinet配置失败," + err.Error())
|
||||
}
|
||||
|
||||
clientset, err := kubernetes.NewForConfig(conf)
|
||||
if err != nil {
|
||||
panic("创建k8s clinet失败," + err.Error())
|
||||
} else {
|
||||
logger.Info("k8s client 初始化成功")
|
||||
}
|
||||
|
||||
k.ClientSet = clientset
|
||||
}
|
||||
190
service/pod.go
Normal file
190
service/pod.go
Normal file
@ -0,0 +1,190 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"dkube/config"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"github.com/wonderivan/logger"
|
||||
"io"
|
||||
corev1 "k8s.io/api/core/v1"
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
)
|
||||
|
||||
var Pod pod
|
||||
|
||||
type pod struct{}
|
||||
|
||||
// PodsResp 定义列表的返回内容，Items是Pod的元素列表，Total是元素数量
|
||||
type PodsResp struct {
|
||||
Total int `json:"total"`
|
||||
Items []corev1.Pod `json:"items"`
|
||||
}
|
||||
|
||||
//声明类型
|
||||
type PodsNp struct {
|
||||
Namespace string
|
||||
PodNum int
|
||||
}
|
||||
|
||||
// GetPods 获取Pod列表，支持过滤、排序、分页
|
||||
//context.TODD()用于声明一个空的context上下文，用于List方法内设置这个请求的超时(源码)，这里是常用用法
|
||||
//metav1.ListOptions{}用于过滤List数据，如使用label，field等
|
||||
func (p *pod) GetPods(filterName, namespace string, limit, page int) (podsResp *PodsResp, err error) {
|
||||
podList, err := K8s.ClientSet.CoreV1().Pods(namespace).List(context.TODO(), metav1.ListOptions{})
|
||||
if err != nil {
|
||||
//打印日志,排错
|
||||
logger.Info("获取Pod列表失败," + err.Error())
|
||||
//返回给上一层，最终返回给前端，前端打印出该error
|
||||
return nil, errors.New("获取Pod列表失败," + err.Error())
|
||||
}
|
||||
|
||||
//实例化dataSelector结构体，组装数据
|
||||
selectableData := &dataSelector{
|
||||
GenericDataList: p.toCells(podList.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类型转换成Pod
|
||||
pods := p.fromCells(data.GenericDataList)
|
||||
|
||||
return &PodsResp{
|
||||
Total: total,
|
||||
Items: pods,
|
||||
}, nil
|
||||
|
||||
}
|
||||
|
||||
//获取Pod详情
|
||||
func (p *pod) GetPodDetail(podName, namespace string) (pod *corev1.Pod, err error) {
|
||||
pod, err = K8s.ClientSet.CoreV1().Pods(namespace).Get(context.TODO(), podName, metav1.GetOptions{})
|
||||
if err != nil {
|
||||
logger.Error("获取Pod详情失败," + err.Error())
|
||||
return nil, errors.New("获取Pod详情失败," + err.Error())
|
||||
}
|
||||
return pod, nil
|
||||
}
|
||||
|
||||
//删除Pod
|
||||
func (p *pod) DeletePod(podName, namespace string) (err error) {
|
||||
err = K8s.ClientSet.CoreV1().Pods(namespace).Delete(context.TODO(), podName, metav1.DeleteOptions{})
|
||||
if err != nil {
|
||||
logger.Error("删除Pod失败," + err.Error())
|
||||
return errors.New("删除Pod失败," + err.Error())
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
//更新Pod
|
||||
func (p *pod) UpdatePod(namespace, content string) (err error) {
|
||||
pod := &corev1.Pod{}
|
||||
//将json反序列化为pod类型
|
||||
err = json.Unmarshal([]byte(content), pod)
|
||||
if err != nil {
|
||||
logger.Error("反序列化失败," + err.Error())
|
||||
return errors.New("反序列化失败," + err.Error())
|
||||
}
|
||||
_, err = K8s.ClientSet.CoreV1().Pods(namespace).Update(context.TODO(), pod, metav1.UpdateOptions{})
|
||||
if err != nil {
|
||||
logger.Error("更新Pod失败," + err.Error())
|
||||
return errors.New("更新Pod失败," + err.Error())
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
//获取Pod容器名列表
|
||||
func (p *pod) GetPodContainer(podName, namespace string) (containers []string, err error) {
|
||||
//获取Pod详情
|
||||
pod, err := p.GetPodDetail(podName, namespace)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
//从Pod对象中拿到容器名
|
||||
for _, container := range pod.Spec.Containers {
|
||||
containers = append(containers, container.Name)
|
||||
}
|
||||
return containers, nil
|
||||
}
|
||||
|
||||
//获取Pod内容器日志
|
||||
func (p *pod) GetPodLog(containerName, podName, namespace string) (log string, err error) {
|
||||
//设置日志的配置，容器名、获取内容的配置
|
||||
lineLimit := int64(config.PodLogTailLine)
|
||||
option := &corev1.PodLogOptions{
|
||||
Container: containerName,
|
||||
TailLines: &lineLimit,
|
||||
}
|
||||
//获取request实例
|
||||
req := K8s.ClientSet.CoreV1().Pods(namespace).GetLogs(podName, option)
|
||||
//发起request请求，返回一个io.ReadCloser类型（等同于response.body）
|
||||
podLogs, err := req.Stream(context.TODO())
|
||||
if err != nil {
|
||||
logger.Error(errors.New("获取PodLogs失败," + err.Error()))
|
||||
return "", errors.New("获取PodLogs失败," + err.Error())
|
||||
}
|
||||
defer podLogs.Close() //处理完关闭
|
||||
//将response body写入到缓冲区，目的是为了转成string返回
|
||||
buf := new(bytes.Buffer)
|
||||
_, err = io.Copy(buf, podLogs)
|
||||
if err != nil {
|
||||
logger.Error(errors.New("获取PodLogs失败," + err.Error()))
|
||||
return "", errors.New("获取PodLogs失败," + err.Error())
|
||||
}
|
||||
|
||||
return buf.String(), nil
|
||||
}
|
||||
|
||||
//获取每个namespace的pod数量
|
||||
func (p *pod) GetPodNumPerNp() (podsNps []*PodsNp, err error) {
|
||||
//获取namespace列表
|
||||
namespaceList, err := K8s.ClientSet.CoreV1().Namespaces().List(context.TODO(), metav1.ListOptions{})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
for _, namespace := range namespaceList.Items {
|
||||
//获取pod列表
|
||||
podList, err := K8s.ClientSet.CoreV1().Pods(namespace.Name).List(context.TODO(), metav1.ListOptions{})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
//组装数据
|
||||
podsNp := &PodsNp{
|
||||
Namespace: namespace.Name,
|
||||
PodNum: len(podList.Items),
|
||||
}
|
||||
//添加到podsNps数组中
|
||||
podsNps = append(podsNps, podsNp)
|
||||
}
|
||||
return podsNps, nil
|
||||
}
|
||||
|
||||
//类型转换的方法 corev1.Pod -> DataCell, DataCell --> corev1.Pod
|
||||
func (p *pod) toCells(pods []corev1.Pod) []DataCell {
|
||||
cells := make([]DataCell, len(pods))
|
||||
for i := range pods {
|
||||
cells[i] = podCell(pods[i])
|
||||
}
|
||||
return cells
|
||||
}
|
||||
|
||||
func (p *pod) fromCells(cells []DataCell) []corev1.Pod {
|
||||
pods := make([]corev1.Pod, len(cells))
|
||||
for i := range cells {
|
||||
//cells[i].(podCell)是将DataCell类型转成podCell类型
|
||||
pods[i] = corev1.Pod(cells[i].(podCell))
|
||||
}
|
||||
return pods
|
||||
}
|
||||
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