diff --git a/add.go b/add.go index 1bfb480..e644fed 100644 --- a/add.go +++ b/add.go @@ -77,7 +77,15 @@ func (c *NewCluster) AddHypervisor(hostname string, cpus int, mem int) *Hypervis return h } -func (c *NewCluster) AddDroplet(uuid string, hostname string, cpus int, mem int) *Droplet { +func (c *NewCluster) AddEvent(e *Event) { + c.e.Events = append(c.e.Events, e) +} + +func (c *NewCluster) AddDroplet(d *Droplet) { + c.d.Droplets = append(c.d.Droplets, d) +} + +func (c *NewCluster) AddDropletSimple(uuid string, hostname string, cpus int, mem int) *Droplet { d := c.FindDropletByName(hostname) if d != nil { return d @@ -93,7 +101,7 @@ func (c *NewCluster) AddDroplet(uuid string, hostname string, cpus int, mem int) d.Cpus = 1 } d.Memory = SetGB(mem * 32) - c.d.Droplets = append(c.d.Droplets, d) + c.AddDroplet(d) return d }