140 lines
3.7 KiB
Go
140 lines
3.7 KiB
Go
package service
|
||
|
||
import (
|
||
appsv1 "k8s.io/api/apps/v1" //引入K8s的APPSv1的包.定义别名为appsv1
|
||
corev1 "k8s.io/api/core/v1" //引入K8s的corev1的包.定义别名为corev1
|
||
"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
|
||
}
|