func (m *Manager) init() []*shipyard.Engine {
engines := []*shipyard.Engine{}
res, err := r.Table(tblNameConfig).Run(m.session)
if err != nil {
logger.Fatalf("error getting configuration: %s", err)
}
if err := res.All(&engines); err != nil {
logger.Fatalf("error loading configuration: %s", err)
}
m.engines = engines
var engs []*citadel.Engine
for _, d := range engines {
tlsConfig := &tls.Config{}
if d.CACertificate != "" && d.SSLCertificate != "" && d.SSLKey != "" {
caCert := []byte(d.CACertificate)
sslCert := []byte(d.SSLCertificate)
sslKey := []byte(d.SSLKey)
c, err := getTLSConfig(caCert, sslCert, sslKey)
if err != nil {
logger.Errorf("error getting tls config: %s", err)
}
tlsConfig = c
}
if err := setEngineClient(d.Engine, tlsConfig); err != nil {
logger.Errorf("error setting tls config for engine: %s", err)
}
engs = append(engs, d.Engine)
logger.Infof("loaded engine id=%s addr=%s", d.Engine.ID, d.Engine.Addr)
}
clusterManager, err := cluster.New(scheduler.NewResourceManager(), engs...)
if err != nil {
logger.Fatal(err)
}
if err := clusterManager.Events(&EventHandler{Manager: m}); err != nil {
logger.Fatalf("unable to register event handler: %s", err)
}
var (
labelScheduler = &scheduler.LabelScheduler{}
uniqueScheduler = &scheduler.UniqueScheduler{}
hostScheduler = &scheduler.HostScheduler{}
multiScheduler = scheduler.NewMultiScheduler(
labelScheduler,
uniqueScheduler,
)
)
// TODO: refactor to be configurable
clusterManager.RegisterScheduler("service", labelScheduler)
clusterManager.RegisterScheduler("unique", uniqueScheduler)
clusterManager.RegisterScheduler("multi", multiScheduler)
clusterManager.RegisterScheduler("host", hostScheduler)
m.clusterManager = clusterManager
// start extension health check
go m.extensionHealthCheck()
// start engine check
go m.engineCheck()
// anonymous usage info
go m.usageReport()
return engines
}
func main() {
if err := loadConfig(); err != nil {
log.Fatal(err)
}
tlsConfig, err := getTLSConfig()
if err != nil {
log.Fatal(err)
}
for _, d := range config.Engines {
if err := setEngineClient(d, tlsConfig); err != nil {
log.Fatal(err)
}
}
clusterManager, err = cluster.New(scheduler.NewResourceManager(), config.Engines...)
if err != nil {
log.Fatal(err)
}
var (
labelScheduler = &scheduler.LabelScheduler{}
uniqueScheduler = &scheduler.UniqueScheduler{}
hostScheduler = &scheduler.HostScheduler{}
multiScheduler = scheduler.NewMultiScheduler(
labelScheduler,
uniqueScheduler,
)
)
clusterManager.RegisterScheduler("service", labelScheduler)
clusterManager.RegisterScheduler("unique", uniqueScheduler)
clusterManager.RegisterScheduler("multi", multiScheduler)
clusterManager.RegisterScheduler("host", hostScheduler)
r := mux.NewRouter()
r.HandleFunc("/containers", containers).Methods("GET")
r.HandleFunc("/run", run).Methods("POST")
r.HandleFunc("/destroy", destroy).Methods("DELETE")
r.HandleFunc("/engines", engines).Methods("GET")
log.Printf("bastion listening on %s\n", config.ListenAddr)
if err := http.ListenAndServe(config.ListenAddr, r); err != nil {
log.Fatal(err)
}
}