/
resources.go
72 lines (65 loc) · 1.75 KB
/
resources.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
package containers
import (
v1 "k8s.io/api/core/v1"
"k8s.io/apimachinery/pkg/api/resource"
)
// Resources represents Kubernetes ResourceRequirements
type Resources struct {
Limit Limit
Request Request
}
// toK8S converts Resources to Kuberntes client object
func (r *Resources) toK8S() v1.ResourceRequirements {
return v1.ResourceRequirements{
Limits: r.Limit.toK8S(),
Requests: r.Request.toK8S(),
}
}
// Limit represents Kubernetes ResourceList with Limits
type Limit struct {
CPU string
Memory string
Storage string
EphemeralStorage string
}
// toK8S converts Limit to Kuberntes client object
func (l *Limit) toK8S() v1.ResourceList {
m := map[v1.ResourceName]resource.Quantity{}
if len(l.CPU) > 0 {
m[v1.ResourceCPU] = resource.MustParse(l.CPU)
}
if len(l.Memory) > 0 {
m[v1.ResourceMemory] = resource.MustParse(l.Memory)
}
if len(l.Storage) > 0 {
m[v1.ResourceStorage] = resource.MustParse(l.Storage)
}
if len(l.EphemeralStorage) > 0 {
m[v1.ResourceEphemeralStorage] = resource.MustParse(l.EphemeralStorage)
}
return m
}
// Request represents Kubernetes ResourceList with Requests
type Request struct {
CPU string
Memory string
Storage string
EphemeralStorage string
}
// toK8S converts Request to Kuberntes client object
func (r *Request) toK8S() v1.ResourceList {
m := map[v1.ResourceName]resource.Quantity{}
if len(r.CPU) > 0 {
m[v1.ResourceCPU] = resource.MustParse(r.CPU)
}
if len(r.Memory) > 0 {
m[v1.ResourceMemory] = resource.MustParse(r.Memory)
}
if len(r.Storage) > 0 {
m[v1.ResourceStorage] = resource.MustParse(r.Storage)
}
if len(r.EphemeralStorage) > 0 {
m[v1.ResourceEphemeralStorage] = resource.MustParse(r.EphemeralStorage)
}
return m
}