Change wiring

This commit is contained in:
Jonathan Leibiusky @xetorthio
2017-08-10 12:13:18 -03:00
parent 801bbefc19
commit 7495f27f64
2 changed files with 26 additions and 27 deletions

24
api.go
View File

@@ -8,12 +8,34 @@ import (
"github.com/play-with-docker/play-with-docker/docker" "github.com/play-with-docker/play-with-docker/docker"
"github.com/play-with-docker/play-with-docker/event" "github.com/play-with-docker/play-with-docker/event"
"github.com/play-with-docker/play-with-docker/handlers" "github.com/play-with-docker/play-with-docker/handlers"
"github.com/play-with-docker/play-with-docker/pwd"
"github.com/play-with-docker/play-with-docker/scheduler"
"github.com/play-with-docker/play-with-docker/scheduler/task"
"github.com/play-with-docker/play-with-docker/storage" "github.com/play-with-docker/play-with-docker/storage"
) )
func main() { func main() {
config.ParseFlags() config.ParseFlags()
handlers.Bootstrap(initStorage, initEvent, initFactory)
e := initEvent()
s := initStorage()
f := initFactory(s)
core := pwd.NewPWD(f, e, s)
sch, err := scheduler.NewScheduler(s, e, core)
if err != nil {
log.Fatal("Error initializing the scheduler: ", err)
}
sch.AddTask(task.NewCheckPorts(e, f))
sch.AddTask(task.NewCheckSwarmPorts(e, f))
sch.AddTask(task.NewCheckSwarmStatus(e, f))
sch.AddTask(task.NewCollectStats(e, f))
sch.Start()
handlers.Bootstrap(core, e)
handlers.Register() handlers.Register()
} }

View File

@@ -10,12 +10,8 @@ import (
gh "github.com/gorilla/handlers" gh "github.com/gorilla/handlers"
"github.com/gorilla/mux" "github.com/gorilla/mux"
"github.com/play-with-docker/play-with-docker/config" "github.com/play-with-docker/play-with-docker/config"
"github.com/play-with-docker/play-with-docker/docker"
"github.com/play-with-docker/play-with-docker/event" "github.com/play-with-docker/play-with-docker/event"
"github.com/play-with-docker/play-with-docker/pwd" "github.com/play-with-docker/play-with-docker/pwd"
"github.com/play-with-docker/play-with-docker/scheduler"
"github.com/play-with-docker/play-with-docker/scheduler/task"
"github.com/play-with-docker/play-with-docker/storage"
"github.com/play-with-docker/play-with-docker/templates" "github.com/play-with-docker/play-with-docker/templates"
"github.com/prometheus/client_golang/prometheus/promhttp" "github.com/prometheus/client_golang/prometheus/promhttp"
"github.com/urfave/negroni" "github.com/urfave/negroni"
@@ -25,28 +21,9 @@ var core pwd.PWDApi
var e event.EventApi var e event.EventApi
var ws *socketio.Server var ws *socketio.Server
func Bootstrap(storageInit func() storage.StorageApi, eventInit func() event.EventApi, factoryInit func(storage.StorageApi) docker.FactoryApi) { func Bootstrap(c pwd.PWDApi, ev event.EventApi) {
s := storageInit() core = c
e = ev
e = eventInit()
f := factoryInit(s)
core = pwd.NewPWD(f, e, s)
sch, err := scheduler.NewScheduler(s, e, core)
if err != nil {
log.Fatal("Error initializing the scheduler: ", err)
}
sch.AddTask(task.NewCheckPorts(e, f))
sch.AddTask(task.NewCheckSwarmPorts(e, f))
sch.AddTask(task.NewCheckSwarmStatus(e, f))
sch.AddTask(task.NewCollectStats(e, f))
sch.Start()
Register()
} }
func Register() { func Register() {