save drive filename

Signed-off-by: Jeff Carr <jcarr@wit.com>
This commit is contained in:
Jeff Carr 2024-10-23 05:32:49 -05:00
parent bdb72e5682
commit 583fa15779
1 changed files with 63 additions and 3 deletions

View File

@ -6,9 +6,9 @@ import (
"errors"
"fmt"
pb "go.wit.com/lib/protobuf/virtbuf"
"go.wit.com/log"
"libvirt.org/go/libvirtxml"
pb "go.wit.com/lib/protobuf/virtbuf"
)
func addDomainDroplet(domcfg *libvirtxml.Domain) (*DropletT, error) {
@ -67,7 +67,7 @@ func updateDroplet(d *DropletT, domcfg *libvirtxml.Domain) bool {
return false
}
if ! updateMemory(d, domcfg) {
if !updateMemory(d, domcfg) {
log.Info("updateMemory() failed")
ok = false
}
@ -85,10 +85,15 @@ func updateDroplet(d *DropletT, domcfg *libvirtxml.Domain) bool {
ok = false
}
if ! updateNetwork(d, domcfg) {
if !updateNetwork(d, domcfg) {
log.Info("updateNetwork() failed")
ok = false
}
if !updateDisk(d, domcfg) {
log.Info("updateDisk() failed")
ok = false
}
return ok
}
@ -109,6 +114,28 @@ func updateMemory(d *DropletT, domcfg *libvirtxml.Domain) bool {
}
return true
}
if domcfg.Memory.Unit == "MiB" {
var m int64
m = int64(domcfg.Memory.Value * 1024 * 1024)
if d.pb.Memory != m {
d.pb.Memory = m
me.changed = true
fmt.Printf("Memory changed %d, %d %s\n", d.pb.Memory, domcfg.Memory.Value, domcfg.Memory.Unit)
}
return true
}
if domcfg.Memory.Unit == "GiB" {
var m int64
m = int64(domcfg.Memory.Value * 1024 * 1024 * 1024)
if d.pb.Memory != m {
d.pb.Memory = m
me.changed = true
fmt.Printf("Memory changed %d, %d %s\n", d.pb.Memory, domcfg.Memory.Value, domcfg.Memory.Unit)
}
return true
}
fmt.Println("Unknown Memory Unit", domcfg.Memory.Unit)
return false
}
@ -154,3 +181,36 @@ func updateNetwork(d *DropletT, domcfg *libvirtxml.Domain) bool {
log.Info("mac addrs:", macs)
return true
}
// returns false if something went wrong
func updateDisk(d *DropletT, domcfg *libvirtxml.Domain) bool {
if (d == nil) || (domcfg == nil) {
return false
}
for _, disk := range domcfg.Devices.Disks {
var t *libvirtxml.DomainDiskSourceFile
t = disk.Source.File
filename := t.File
if filename == "" {
fmt.Println("No disk source file found.")
continue
}
var found bool = false
for _, disk := range d.pb.Disks {
if disk.Filename == filename {
log.Info("OKAY. FOUND filename", filename)
found = true
}
}
if !found {
var disk *pb.Disk
disk = new(pb.Disk)
disk.Filename = filename
d.pb.Disks = append(d.pb.Disks, disk)
log.Info("New filename", filename)
me.changed = true
}
}
return true
}