127 lines
2.3 KiB
Go
127 lines
2.3 KiB
Go
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
|
|
}
|
|
}
|