gary/dospin
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Added support for resizing Droplets after creation.

Browse Source
This commit is contained in:
Gary Talent 2016-02-28 02:45:41 -06:00
parent 92a31f2244
commit c4dc4c2de1
4 changed files with 72 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
dospin.json
View File

@ -3,6 +3,7 @@
"Servers": {
"minecraft": {
"Ports": [25565],
"InitialSize": "512mb",
"Size": "4gb",
"Region": "nyc3"
}

84
droplethandler.go
View File

@ -57,11 +57,21 @@ func (me *DropletHandler) Spinup(name string) (string, error) {
if err != nil {
return "", err
}
vd := me.settings.Servers[name]
// determine droplet size
var size string
if vd.InitialSize != "" {
size = vd.InitialSize
} else {
size = vd.Size
}
createRequest := &godo.DropletCreateRequest{
Name: DROPLET_NS + name,
Region: vd.Region,
Size: vd.Size,
Size: size,
PrivateNetworking: true,
Image: godo.DropletCreateImage{
ID: image.ID,
@ -89,6 +99,28 @@ func (me *DropletHandler) Spinup(name string) (string, error) {
}
log.Println("Spinup: Created " + name)
// resize if necessary
if vd.InitialSize != "" && vd.InitialSize != vd.Size {
// power off
me.poweroff(name)
// resize
log.Println("Spinup: Resizing " + name)
action, _, err := me.client.DropletActions.Resize(droplet.ID, vd.Size, false)
if err != nil || !me.actionWait(action.ID) {
return "", err
}
log.Println("Spinup: Resized " + name)
// power back on
log.Println("Spinup: Powering on " + name)
action, _, err = me.client.DropletActions.PowerOn(droplet.ID)
if err != nil || !me.actionWait(action.ID) {
return "", err
}
log.Println("Spinup: Powered on " + name)
}
// delete the image
log.Println("Spinup: Deleting image " + name)
_, err = me.client.Images.Delete(image.ID)
@ -113,23 +145,9 @@ func (me *DropletHandler) Spindown(name string) error {
}
// power off
if droplet.Status != "off" {
log.Println("Spindown: Powering down " + name)
// wait until machine is off
for {
droplet, err = me.getDroplet(name)
err = me.poweroff(name)
if err != nil {
log.Println(err)
} else if droplet.Status == "off" {
break
}
time.Sleep(100 * time.Millisecond)
_, _, err = me.client.DropletActions.Shutdown(droplet.ID)
if err != nil {
log.Println("Spindown: Power down of ", name, " failed: ", err)
}
}
log.Println("Spindown: Powered down " + name)
return err
}
// snapshot existing droplet
@ -151,6 +169,35 @@ func (me *DropletHandler) Spindown(name string) error {
return err
}
func (me *DropletHandler) poweroff(name string) error {
droplet, err := me.getDroplet(name)
if err != nil {
return err
}
if droplet.Status != "off" {
log.Println("Powering down " + name)
// wait until machine is off
for {
droplet, err = me.getDroplet(name)
if err != nil {
log.Println("Power down of", name, "failed:", err)
if droplet.ID < 1 {
return err
}
} else if droplet.Status == "off" {
break
}
time.Sleep(100 * time.Millisecond)
_, _, err = me.client.DropletActions.Shutdown(droplet.ID)
if err != nil {
log.Println("Power down of", name, "failed:", err)
}
}
log.Println("Powered down", name)
}
return err
}
func (me *DropletHandler) getDroplet(name string) (godo.Droplet, error) {
name = DROPLET_NS + name
page := 0
@ -220,8 +267,7 @@ func (me *DropletHandler) actionWait(actionId int) bool {
for {
a, _, err := me.client.Actions.Get(actionId)
if err != nil {
log.Println("Action failed: ", err)
return false
log.Println("Action retrieval failed: ", err)
} else if a.Status == "completed" {
return true
} else if a.Status == "errored" {

2
servermanager.go
View File

@ -92,7 +92,7 @@ func (me *ServerManager) serveAction(action int) bool {
case SERVERMANAGER_SPINUP:
ip, err := me.server.Spinup(me.name)
if err == nil {
log.Println("ServerManager: Got IP for", me.name, ":", ip)
log.Println("ServerManager: Got IP for", me.name+":", ip)
me.addPortForwards(ip)
} else {
log.Println("ServerManager: Could not spin up "+me.name+":", err)

1
settings.go
View File

@ -19,6 +19,7 @@ type Settings struct {
type Server struct {
Ports []int
InitialSize string
Size string
Region string
}