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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
package virtbuf
import (
"fmt"
"github.com/google/uuid"
)
// can the json protobuf output use a string and have a type handler
// to convert it back to int64?
func SetGB(gb int) int64 {
return int64(gb * 1024 * 1024 * 1024)
}
func SetMB(mb int) int64 {
return int64(mb * 1024 * 1024)
}
func (x *Hypervisor) SetMemoryGB(gb int) {
x.Memory = int64(gb * 1024 * 1024 * 1024)
}
func (x *Hypervisor) GetMemoryPrintable() string {
i := x.Memory / (1024 * 1024 * 1024)
return fmt.Sprintf("%d GB", i)
}
func (all *Droplets) FindDroplet(name string) *Droplet {
for _, d := range all.Droplets {
if d.Hostname == name {
return d
}
}
return nil
}
func (c *Cluster) FindDroplet(name string) *Droplet {
for _, d := range c.Droplets {
if d.Hostname == name {
return d
}
}
return nil
}
func (c *Cluster) FindHypervisor(name string) *Hypervisor {
for _, h := range c.Hypervisors {
if h.Hostname == name {
return h
}
}
return nil
}
func (c *Cluster) AddHypervisor(hostname string, cpus int, mem int) *Hypervisor {
h := c.FindHypervisor(hostname)
if h != nil {
return h
}
// Generate a new UUID
id := uuid.New()
h = &Hypervisor{
Uuid: id.String(),
Hostname: hostname,
Cpus: int64(cpus),
Comment: "this is a fake hypervisor",
}
if cpus < 0 {
h.Cpus = 1
}
h.SetMemoryGB(mem * 32)
c.Hypervisors = append(c.Hypervisors, h)
return h
}
func (c *Cluster) AddDroplet(uuid string, hostname string, cpus int, mem int) *Droplet {
d := c.FindDroplet(hostname)
if d != nil {
return d
}
d = &Droplet{
Uuid: uuid,
Hostname: hostname,
Cpus: int64(cpus),
}
if cpus < 0 {
d.Cpus = 1
}
d.Memory = SetGB(mem * 32)
c.Droplets = append(c.Droplets, d)
return d
}
// This isn't for the marketing department
// so this isn't going to use 'MiB' and 'GiB'
func HumanFormatBytes(b int64) string {
if b < 2000 {
return fmt.Sprintf("%d B", b)
}
kb := int(b / 1024)
if kb < 2000 {
return fmt.Sprintf("%d KB", kb)
}
mb := int(b / (1024 * 1024))
if mb < 2000 {
return fmt.Sprintf("%d MB", mb)
}
gb := int(b / (1024 * 1024 * 1024))
if gb < 2000 {
return fmt.Sprintf("%d GB", gb)
}
tb := int(b / (1024 * 1024 * 1024 * 1024))
return fmt.Sprintf("%d TB", tb)
}
func (c *Cluster) BlankFields() {
for _, d := range c.Droplets {
d.CurrentState = 0
}
}
|
