Update pod.go

This commit is contained in:
dqzboy 2022-09-25 21:12:09 +08:00 committed by GitHub
parent 8ba971e7f2
commit 0cbfa83182
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -16,31 +16,24 @@ 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数据如使用labelfield等
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{
@ -52,13 +45,11 @@ func (p *pod) GetPods(filterName, namespace string, limit, page int) (podsResp *
},
}
//先过滤
filtered := selectableData.Filter()
total := len(filtered.GenericDataList)
//排序和分页
data := filtered.Sort().Paginate()
//将DataCell类型转换成Pod
pods := p.fromCells(data.GenericDataList)
return &PodsResp{
@ -91,7 +82,6 @@ func (p *pod) DeletePod(podName, namespace string) (err error) {
//更新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())
@ -112,7 +102,6 @@ func (p *pod) GetPodContainer(podName, namespace string) (containers []string, e
if err != nil {
return nil, err
}
//从Pod对象中拿到容器名
for _, container := range pod.Spec.Containers {
containers = append(containers, container.Name)
}
@ -121,7 +110,6 @@ func (p *pod) GetPodContainer(podName, namespace string) (containers []string, e
//获取Pod内容器日志
func (p *pod) GetPodLog(containerName, podName, namespace string) (log string, err error) {
//设置日志的配置,容器名、获取内容的配置
lineLimit := int64(config.PodLogTailLine)
option := &corev1.PodLogOptions{
Container: containerName,
@ -129,14 +117,12 @@ func (p *pod) GetPodLog(containerName, podName, namespace string) (log string, e
}
//获取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 {
@ -160,18 +146,15 @@ func (p *pod) GetPodNumPerNp() (podsNps []*PodsNp, err error) {
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 {
@ -183,7 +166,6 @@ func (p *pod) toCells(pods []corev1.Pod) []DataCell {
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