7c98be0e52
The state_timeout config allows you to determine the timeout for "waiting for droplet to become [active, off, etc.]". This still defaults to 3 minutes.
57 lines
1.1 KiB
Go
57 lines
1.1 KiB
Go
package digitalocean
|
|
|
|
import (
|
|
"errors"
|
|
"log"
|
|
"time"
|
|
)
|
|
|
|
// waitForState simply blocks until the droplet is in
|
|
// a state we expect, while eventually timing out.
|
|
func waitForDropletState(desiredState string, dropletId uint, client *DigitalOceanClient, c config) error {
|
|
active := make(chan bool, 1)
|
|
|
|
go func() {
|
|
attempts := 0
|
|
for {
|
|
attempts += 1
|
|
|
|
log.Printf("Checking droplet status... (attempt: %d)", attempts)
|
|
|
|
_, status, err := client.DropletStatus(dropletId)
|
|
|
|
if err != nil {
|
|
log.Println(err)
|
|
break
|
|
}
|
|
|
|
if status == desiredState {
|
|
break
|
|
}
|
|
|
|
// Wait 3 seconds in between
|
|
time.Sleep(3 * time.Second)
|
|
}
|
|
|
|
active <- true
|
|
}()
|
|
|
|
log.Printf("Waiting for up to %s for droplet to become %s", c.RawStateTimeout, desiredState)
|
|
timeout := time.After(c.StateTimeout)
|
|
|
|
ActiveWaitLoop:
|
|
for {
|
|
select {
|
|
case <-active:
|
|
// We connected. Just break the loop.
|
|
break ActiveWaitLoop
|
|
case <-timeout:
|
|
err := errors.New("Timeout while waiting to for droplet to become active")
|
|
return err
|
|
}
|
|
}
|
|
|
|
// If we got this far, there were no errors
|
|
return nil
|
|
}
|