/
pvcworkloads.go
98 lines (85 loc) · 3.09 KB
/
pvcworkloads.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
/*
Copyright 2021 KubeCube Authors
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package pvc
import (
"context"
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
corev1 "k8s.io/api/core/v1"
"sigs.k8s.io/controller-runtime/pkg/client"
resourcemanage "github.com/kubecube-io/kubecube/pkg/apiserver/cubeapi/resourcemanage/handle"
"github.com/kubecube-io/kubecube/pkg/apiserver/cubeapi/resourcemanage/resources"
"github.com/kubecube-io/kubecube/pkg/apiserver/cubeapi/resourcemanage/resources/enum"
"github.com/kubecube-io/kubecube/pkg/apiserver/cubeapi/resourcemanage/resources/pod"
"github.com/kubecube-io/kubecube/pkg/clients"
mgrclient "github.com/kubecube-io/kubecube/pkg/multicluster/client"
"github.com/kubecube-io/kubecube/pkg/utils/errcode"
"github.com/kubecube-io/kubecube/pkg/utils/filter"
)
type Pvc struct {
ctx context.Context
client mgrclient.Client
namespace string
filterCondition *filter.Condition
}
func init() {
resourcemanage.SetExtendHandler(enum.PvcWorkLoadResourceType, handle)
}
func handle(param resourcemanage.ExtendContext) (interface{}, *errcode.ErrorInfo) {
access := resources.NewSimpleAccess(param.Cluster, param.Username, param.Namespace)
if allow := access.AccessAllow("", "persistentvolumeclaims", "list"); !allow {
return nil, errcode.ForbiddenErr
}
kubernetes := clients.Interface().Kubernetes(param.Cluster)
if kubernetes == nil {
return nil, errcode.ClusterNotFoundError(param.Cluster)
}
pvc := NewPvc(kubernetes, param.Namespace, param.FilterCondition)
return pvc.getPvcWorkloads(param.ResourceName)
}
func NewPvc(client mgrclient.Client, namespace string, condition *filter.Condition) Pvc {
ctx := context.Background()
return Pvc{
ctx: ctx,
client: client,
namespace: namespace,
filterCondition: condition,
}
}
// getPvcWorkloads get extend deployments
func (p *Pvc) getPvcWorkloads(pvcName string) (*unstructured.Unstructured, *errcode.ErrorInfo) {
result := make(map[string]interface{})
var pods []pod.ExtendPod
var podList corev1.PodList
err := p.client.Cache().List(p.ctx, &podList, client.InNamespace(p.namespace))
if err != nil {
return nil, errcode.BadRequest(err)
}
for _, item := range podList.Items {
for _, volume := range item.Spec.Volumes {
if volume.PersistentVolumeClaim == nil {
continue
}
claimName := volume.PersistentVolumeClaim.ClaimName
if claimName == pvcName {
pods = append(pods, pod.ExtendPod{
Reason: pod.GetPodReason(item),
Pod: item,
})
break
}
}
}
result["pods"] = pods
result["total"] = len(pods)
return &unstructured.Unstructured{Object: result}, nil
}