-
Notifications
You must be signed in to change notification settings - Fork 2
/
persistentVolumeClaim.go
84 lines (70 loc) · 2.67 KB
/
persistentVolumeClaim.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
package v1
import (
"fmt"
"reflect"
"github.com/wwmoraes/kubegraph/internal/adapter"
coreV1 "k8s.io/api/core/v1"
"k8s.io/apimachinery/pkg/runtime"
)
type persistentVolumeClaimAdapter struct {
adapter.ResourceData
}
func init() {
adapter.Register(&persistentVolumeClaimAdapter{
adapter.ResourceData{
ResourceType: reflect.TypeOf(&coreV1.PersistentVolumeClaim{}),
},
})
}
func (thisAdapter *persistentVolumeClaimAdapter) tryCastObject(obj runtime.Object) (*coreV1.PersistentVolumeClaim, error) {
casted, ok := obj.(*coreV1.PersistentVolumeClaim)
if !ok {
return nil, fmt.Errorf("unable to cast object %s to %s", reflect.TypeOf(obj), thisAdapter.GetType().String())
}
return casted, nil
}
// GetType returns the reflected type of the k8s kind managed by this instance
func (thisAdapter *persistentVolumeClaimAdapter) GetType() reflect.Type {
return thisAdapter.ResourceType
}
// Create add a graph node for the given object and stores it for further actions
func (thisAdapter *persistentVolumeClaimAdapter) Create(statefulGraph adapter.StatefulGraph, obj runtime.Object) (adapter.Node, error) {
resource, err := thisAdapter.tryCastObject(obj)
if err != nil {
return nil, err
}
name := fmt.Sprintf("%s.%s~%s", resource.APIVersion, resource.Kind, resource.Name)
return statefulGraph.AddStyledNode(thisAdapter.GetType(), obj, name, resource.Name, "icons/pvc.svg")
}
// Connect creates and edge between the given node and an object on this adapter
func (thisAdapter *persistentVolumeClaimAdapter) Connect(statefulGraph adapter.StatefulGraph, source adapter.Node, targetName string) (adapter.Edge, error) {
return statefulGraph.LinkNode(source, thisAdapter.GetType(), targetName)
}
// Configure connects the resources on this adapter with its dependencies
func (thisAdapter *persistentVolumeClaimAdapter) Configure(statefulGraph adapter.StatefulGraph) error {
persistentVolumeAdapter, err := adapter.Get(reflect.TypeOf(&coreV1.PersistentVolume{}))
if err != nil {
return fmt.Errorf("warning[%s configure]: %v", thisAdapter.GetType().String(), err)
}
objects, err := statefulGraph.GetObjects(thisAdapter.GetType())
if err != nil {
return err
}
for resourceName, resourceObject := range objects {
resource, err := thisAdapter.tryCastObject(resourceObject)
if err != nil {
return err
}
resourceNode, err := statefulGraph.GetNode(thisAdapter.GetType(), resourceName)
if err != nil {
return err
}
if resource.Spec.VolumeName != "" {
_, err := persistentVolumeAdapter.Connect(statefulGraph, resourceNode, resource.Spec.VolumeName)
if err != nil {
fmt.Println(fmt.Errorf("%s configure error: %w", thisAdapter.GetType().String(), err))
}
}
}
return nil
}