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

Renamed MachineManager to MachineHandler. Did same for DropletManager.

Browse Source
This commit is contained in:
Gary Talent 2016-02-27 14:39:35 -06:00
parent a5481c055e
commit 18aa0471e9
3 changed files with 15 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
dospin.go
View File

@ -31,16 +31,16 @@ func main() {
log.Fatal(err)
}
dm := NewDropletManager(settings)
dh := NewDropletHandler(settings)
ip, err := dm.SpinupMachine("minecraft")
ip, err := dh.SpinupMachine("minecraft")
if err != nil {
log.Println("Error:", err)
return
}
log.Println("IP: " + ip)
if err := dm.SpindownMachine("minecraft"); err != nil {
if err := dh.SpindownMachine("minecraft"); err != nil {
log.Println("Error:", err)
return
}

16
dropletmanager.go → droplethandler.go
View File

@ -28,13 +28,13 @@ func (t *tokenSource) Token() (*oauth2.Token, error) {
return token, nil
}
type DropletManager struct {
type DropletHandler struct {
client *godo.Client
settings Settings
}
func NewDropletManager(settings Settings) *DropletManager {
retval := new(DropletManager)
func NewDropletHandler(settings Settings) *DropletHandler {
retval := new(DropletHandler)
retval.settings = settings
// setup DO client
@ -48,7 +48,7 @@ func NewDropletManager(settings Settings) *DropletManager {
/*
Gets the Droplet if it already exists, instantiates it if it does not.
*/
func (me *DropletManager) SpinupMachine(name string) (string, error) {
func (me *DropletHandler) SpinupMachine(name string) (string, error) {
if droplet, err := me.getDroplet(name); err == nil {
return droplet.PrivateIPv4()
} else {
@ -106,7 +106,7 @@ func (me *DropletManager) SpinupMachine(name string) (string, error) {
}
}
func (me *DropletManager) SpindownMachine(name string) error {
func (me *DropletHandler) SpindownMachine(name string) error {
droplet, err := me.getDroplet(name)
if err != nil {
return err
@ -151,7 +151,7 @@ func (me *DropletManager) SpindownMachine(name string) error {
return err
}
func (me *DropletManager) getDroplet(name string) (godo.Droplet, error) {
func (me *DropletHandler) getDroplet(name string) (godo.Droplet, error) {
name = DROPLET_NS + name
page := 0
perPage := 200
@ -181,7 +181,7 @@ func (me *DropletManager) getDroplet(name string) (godo.Droplet, error) {
return droplet, errors.New("Could not find droplet: " + name)
}
func (me *DropletManager) getSnapshot(name string) (godo.Image, error) {
func (me *DropletHandler) getSnapshot(name string) (godo.Image, error) {
name = DROPLET_NS + name
page := 0
perPage := 200
@ -216,7 +216,7 @@ func (me *DropletManager) getSnapshot(name string) (godo.Image, error) {
return image, err
}
func (me *DropletManager) actionWait(actionId int) bool {
func (me *DropletHandler) actionWait(actionId int) bool {
for {
a, _, err := me.client.Actions.Get(actionId)
if err != nil {

7
servicemanager.go
View File

@ -13,9 +13,10 @@ import (
"strconv"
)
type MachineManager interface {
type ServerHandler interface {
// Takes snapshot name, and returns the IP to connect to.
SpinupMachine(name string) (string, error)
SpindownMachine(name string) error
}
type service struct {
@ -30,12 +31,12 @@ type service struct {
// Listens for clients on given ports to spin up machines for the ports
type ServiceManager struct {
machineManager MachineManager
machineManager ServerHandler
machineSvcCnt map[string]int
svcConnStatus map[int]service
}
func NewServiceHandler(mm MachineManager) *ServiceManager {
func NewServiceHandler(mm ServerHandler) *ServiceManager {
sh := new(ServiceManager)
sh.machineManager = mm
return sh