154 lines
3.1 KiB
Go
154 lines
3.1 KiB
Go
/*
|
|
Copyright 2016-2017 gtalent2@gmail.com
|
|
|
|
This Source Code Form is subject to the terms of the Mozilla Public
|
|
License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
*/
|
|
package main
|
|
|
|
import (
|
|
"log"
|
|
"net"
|
|
"strconv"
|
|
)
|
|
|
|
const (
|
|
SERVERMANAGER_SPINUP = iota
|
|
SERVERMANAGER_SPINDOWN
|
|
SERVERMANAGER_STOP
|
|
)
|
|
|
|
type ServerHandler interface {
|
|
// Takes snapshot name, and returns the IP to connect to.
|
|
Spinup(name string) (string, error)
|
|
Spindown(name string) error
|
|
}
|
|
|
|
type ServerManager struct {
|
|
name string
|
|
ports []int
|
|
in chan int
|
|
done chan interface{}
|
|
usageScore int // spin down server when this reaches 0
|
|
server ServerHandler
|
|
}
|
|
|
|
func NewServerManager(name string, server ServerHandler, settings Settings) *ServerManager {
|
|
sm := new(ServerManager)
|
|
|
|
sm.name = name
|
|
sm.ports = settings.Servers[name].Ports
|
|
sm.in = make(chan int)
|
|
sm.done = make(chan interface{})
|
|
sm.usageScore = 5
|
|
sm.server = server
|
|
|
|
return sm
|
|
}
|
|
|
|
/*
|
|
Serves channel requests.
|
|
*/
|
|
func (me *ServerManager) Serve() {
|
|
// TODO: see if server is currently up, and setup port forwarding if so
|
|
|
|
// event loop
|
|
for running := true; running; {
|
|
select {
|
|
case action := <-me.in:
|
|
running = me.serveAction(action)
|
|
}
|
|
}
|
|
|
|
// notify done
|
|
me.done <- 42
|
|
}
|
|
|
|
/*
|
|
Sends the serve loop a spinup message.
|
|
*/
|
|
func (me *ServerManager) Spinup() {
|
|
me.in <- SERVERMANAGER_SPINUP
|
|
}
|
|
|
|
/*
|
|
Sends the serve loop a spindown message.
|
|
*/
|
|
func (me *ServerManager) Spindown() {
|
|
me.in <- SERVERMANAGER_SPINDOWN
|
|
}
|
|
|
|
/*
|
|
Sends the serve loop a quit message.
|
|
*/
|
|
func (me *ServerManager) Stop() {
|
|
me.in <- SERVERMANAGER_STOP
|
|
}
|
|
|
|
func (me *ServerManager) Done() {
|
|
<-me.done
|
|
}
|
|
|
|
func (me *ServerManager) addPortForwards(ip string) {
|
|
}
|
|
|
|
func (me *ServerManager) rmPortForwards() {
|
|
}
|
|
|
|
func (me *ServerManager) setupListener(port int) {
|
|
portStr := strconv.Itoa(port)
|
|
addr, err := net.ResolveTCPAddr("tcp", "0.0.0.0:"+portStr)
|
|
if err != nil {
|
|
log.Print("Could not resolve port and listen address:", err)
|
|
return
|
|
}
|
|
|
|
// listen on port
|
|
go func() {
|
|
l, err := net.ListenTCP("tcp", addr)
|
|
if err != nil {
|
|
log.Print("Could not listen for TCP connection:", err)
|
|
} else {
|
|
for {
|
|
conn, err := l.AcceptTCP()
|
|
if err != nil {
|
|
log.Print("Could not accept TCP connection:", err)
|
|
} else {
|
|
// connection accepted
|
|
|
|
// spinup machine
|
|
me.Spinup()
|
|
|
|
// close existing connection, not doing anything with it
|
|
conn.Close()
|
|
}
|
|
}
|
|
}
|
|
}()
|
|
}
|
|
|
|
func (me *ServerManager) serveAction(action int) bool {
|
|
running := true
|
|
switch action {
|
|
case SERVERMANAGER_SPINUP:
|
|
ip, err := me.server.Spinup(me.name)
|
|
if err == nil {
|
|
log.Println("ServerManager: Got IP for", me.name+":", ip)
|
|
me.addPortForwards(ip)
|
|
} else {
|
|
log.Println("ServerManager: Could not spin up "+me.name+":", err)
|
|
}
|
|
case SERVERMANAGER_SPINDOWN:
|
|
err := me.server.Spindown(me.name)
|
|
if err == nil {
|
|
me.rmPortForwards()
|
|
} else {
|
|
log.Println("ServerManager: Could not spin down "+me.name+":", err)
|
|
}
|
|
case SERVERMANAGER_STOP:
|
|
running = false
|
|
}
|
|
return running
|
|
}
|