works, it's neat, but do I really want this?

Signed-off-by: Jeff Carr <[email protected]>

1 files changed, 4 insertions, 3 deletions

diff --git a/addDroplet.go b/addDroplet.go
index dd625fb..1bff71e 100644
--- a/addDroplet.go
+++ b/addDroplet.go
@@ -173,38 +173,39 @@ func updateMemory(d *DropletT, domcfg *libvirtxml.Domain) bool {
 	if domcfg.Memory == nil {
 		return true
 	}
+	var m int64 = 0
 
 	// check memory
 	if domcfg.Memory.Unit == "KiB" {
-		var m int64
 		m = int64(domcfg.Memory.Value * 1024)
 		if d.pb.Memory != m {
 			d.pb.Memory = m
 			me.changed = true
 			fmt.Printf("Memory changed %s to %d %s\n", pb.HumanFormatBytes(d.pb.Memory), domcfg.Memory.Value, domcfg.Memory.Unit)
 		}
+		d.SetMemory(m)
 		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 %s to %d %s\n", pb.HumanFormatBytes(d.pb.Memory), domcfg.Memory.Value, domcfg.Memory.Unit)
 		}
+		d.SetMemory(m)
 		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 %s, %d %s\n", pb.HumanFormatBytes(d.pb.Memory), domcfg.Memory.Value, domcfg.Memory.Unit)
 		}
+		d.SetMemory(m)
 		return true
 	}
 	fmt.Println("Unknown Memory Unit", domcfg.Memory.Unit)