This repository has been archived by the owner on Mar 16, 2024. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 102
/
volume.go
77 lines (66 loc) · 2.23 KB
/
volume.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
package client
import (
"context"
"sort"
apiv1 "github.com/acorn-io/runtime/pkg/apis/api.acorn.io/v1"
apierror "k8s.io/apimachinery/pkg/api/errors"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
kclient "sigs.k8s.io/controller-runtime/pkg/client"
)
func (c *DefaultClient) VolumeList(ctx context.Context) (result []apiv1.Volume, _ error) {
vols := &apiv1.VolumeList{}
err := c.Client.List(ctx, vols, &kclient.ListOptions{
Namespace: c.Namespace,
})
if err != nil {
return nil, err
}
sort.Slice(vols.Items, func(i, j int) bool {
if vols.Items[i].CreationTimestamp.Time == vols.Items[j].CreationTimestamp.Time {
return vols.Items[i].Name < vols.Items[j].Name
}
return vols.Items[i].CreationTimestamp.After(vols.Items[j].CreationTimestamp.Time)
})
return vols.Items, nil
}
func (c *DefaultClient) VolumeGet(ctx context.Context, name string) (*apiv1.Volume, error) {
vol := &apiv1.Volume{}
return vol, c.Client.Get(ctx, kclient.ObjectKey{
Name: name,
Namespace: c.Namespace,
}, vol)
}
func (c *DefaultClient) VolumeDelete(ctx context.Context, name string) (*apiv1.Volume, error) {
// get first to ensure the namespace matches
v, err := c.VolumeGet(ctx, name)
if apierror.IsNotFound(err) {
return nil, nil
}
return v, c.Client.Delete(ctx, &apiv1.Volume{
ObjectMeta: metav1.ObjectMeta{
Name: name,
Namespace: c.Namespace,
},
})
}
func (c *DefaultClient) VolumeClassList(ctx context.Context) ([]apiv1.VolumeClass, error) {
volumeClasses := new(apiv1.VolumeClassList)
err := c.Client.List(ctx, volumeClasses, &kclient.ListOptions{Namespace: c.Namespace})
if err != nil {
return nil, err
}
sort.Slice(volumeClasses.Items, func(i, j int) bool {
if volumeClasses.Items[i].CreationTimestamp.Time == volumeClasses.Items[j].CreationTimestamp.Time {
return volumeClasses.Items[i].Name < volumeClasses.Items[j].Name
}
return volumeClasses.Items[i].CreationTimestamp.After(volumeClasses.Items[j].CreationTimestamp.Time)
})
return volumeClasses.Items, nil
}
func (c *DefaultClient) VolumeClassGet(ctx context.Context, name string) (*apiv1.VolumeClass, error) {
storage := new(apiv1.VolumeClass)
return storage, c.Client.Get(ctx, kclient.ObjectKey{
Namespace: c.Namespace,
Name: name,
}, storage)
}