func (server *Server) startProviderWorkers() error {
server.providerWorkers = worker.NewWorkerPool()
logger.Log(logger.LevelDebug, "server", "declaring providers")
for _, prov := range server.providers {
connectorType, err := config.GetString(prov.Config.Connector, "type", true)
if err != nil {
return fmt.Errorf("provider `%s' connector: %s", prov.Name, err)
} else if _, ok := connector.Connectors[connectorType]; !ok {
return fmt.Errorf("provider `%s' uses unknown connector type `%s'", prov.Name, connectorType)
}
providerWorker := worker.NewWorker()
providerWorker.RegisterEvent(eventInit, workerProviderInit)
providerWorker.RegisterEvent(eventShutdown, workerProviderShutdown)
providerWorker.RegisterEvent(eventRun, workerProviderRun)
providerWorker.RegisterEvent(eventCatalogRefresh, workerProviderRefresh)
if err := providerWorker.SendEvent(eventInit, false, prov, connectorType); err != nil {
logger.Log(logger.LevelWarning, "server", "in provider `%s', %s", prov.Name, err)
logger.Log(logger.LevelWarning, "server", "discarding provider `%s'", prov.Name)
continue
}
// Add worker into pool if initialization went fine
server.providerWorkers.Add(providerWorker)
providerWorker.SendEvent(eventRun, true, nil)
logger.Log(logger.LevelDebug, "server", "declared provider `%s'", prov.Name)
}
return nil
}
// Run starts the server serving the HTTP responses.
func (server *Server) Run() error {
server.startTime = time.Now()
// Set up server logging
if server.Config.LogFile != "" && server.Config.LogFile != "-" {
dirPath, _ := path.Split(server.Config.LogFile)
os.MkdirAll(dirPath, 0755)
serverOutput, err := os.OpenFile(server.Config.LogFile, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
if err != nil {
logger.Log(logger.LevelError, "server", "unable to open log file: %s", err)
return err
}
defer serverOutput.Close()
logger.SetOutput(serverOutput)
}
logger.SetLevel(server.logLevel)
// Load server configuration
if err := server.Config.Reload(); err != nil {
logger.Log(logger.LevelError, "server", "unable to load configuration: %s", err)
return err
}
// Handle pid file creation if set
if server.Config.PidFile != "" {
fd, err := os.OpenFile(server.Config.PidFile, os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0644)
if err != nil {
return fmt.Errorf("unable to create pid file `%s'", server.Config.PidFile)
}
defer fd.Close()
fd.Write([]byte(strconv.Itoa(os.Getpid()) + "\n"))
}
server.wg.Add(1)
// Create new catalog instance
server.Catalog = catalog.NewCatalog()
// Instanciate catalog worker
server.catalogWorker = worker.NewWorker()
server.catalogWorker.RegisterEvent(eventInit, workerCatalogInit)
server.catalogWorker.RegisterEvent(eventShutdown, workerCatalogShutdown)
server.catalogWorker.RegisterEvent(eventRun, workerCatalogRun)
if err := server.catalogWorker.SendEvent(eventInit, false, server.Catalog); err != nil {
return err
}
server.catalogWorker.SendEvent(eventRun, true, nil)
// Instanciate providers
for providerName, providerConfig := range server.Config.Providers {
server.providers[providerName] = provider.NewProvider(providerName, providerConfig, server.Catalog)
}
if err := server.startProviderWorkers(); err != nil {
return err
}
// Send initial catalog refresh event to provider workers
server.providerWorkers.Broadcast(eventCatalogRefresh, nil)
// Create library instance
server.Library = library.NewLibrary(server.Config, server.Catalog)
go server.Library.Refresh()
// Instanciate serve worker
server.serveWorker = worker.NewWorker()
server.serveWorker.RegisterEvent(eventInit, workerServeInit)
server.serveWorker.RegisterEvent(eventShutdown, workerServeShutdown)
server.serveWorker.RegisterEvent(eventRun, workerServeRun)
if err := server.serveWorker.SendEvent(eventInit, false, server); err != nil {
return err
} else if err := server.serveWorker.SendEvent(eventRun, false, nil); err != nil {
return err
}
server.wg.Wait()
return nil
}
// Run starts the server serving the HTTP responses.
func (server *Server) Run() error {
server.startTime = time.Now()
// Set server build information
server.buildInfo = &buildInfo{
Version: version,
BuildDate: buildDate,
Compiler: fmt.Sprintf("%s (%s)", runtime.Compiler, runtime.Version()),
Connectors: make([]string, 0),
}
for connector := range connector.Connectors {
server.buildInfo.Connectors = append(server.buildInfo.Connectors, connector)
}
sort.Strings(server.buildInfo.Connectors)
// Set up server logging
if server.logPath != "" && server.logPath != "-" {
dirPath, _ := path.Split(server.logPath)
os.MkdirAll(dirPath, 0755)
serverOutput, err := os.OpenFile(server.logPath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
if err != nil {
logger.Log(logger.LevelError, "server", "unable to open log file: %s", err)
return err
}
defer serverOutput.Close()
logger.SetOutput(serverOutput)
}
logger.SetLevel(server.logLevel)
// Load server configuration
if err := server.Config.Load(server.configPath); err != nil {
logger.Log(logger.LevelError, "server", "unable to load configuration: %s", err)
return err
}
// Handle pid file creation if set
if server.Config.PidFile != "" {
fd, err := os.OpenFile(server.Config.PidFile, os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0644)
if err != nil {
return fmt.Errorf("unable to create pid file `%s'", server.Config.PidFile)
}
defer fd.Close()
fd.Write([]byte(strconv.Itoa(os.Getpid()) + "\n"))
}
server.wg.Add(1)
// Generate unique server instance identifier
uuidTemp, err := uuid.NewV4()
if err != nil {
return err
}
server.ID = uuidTemp.String()
// Create new catalog instance
server.Catalog = catalog.NewCatalog()
// Instanciate catalog worker
server.catalogWorker = worker.NewWorker()
server.catalogWorker.RegisterEvent(eventInit, workerCatalogInit)
server.catalogWorker.RegisterEvent(eventShutdown, workerCatalogShutdown)
server.catalogWorker.RegisterEvent(eventRun, workerCatalogRun)
if err := server.catalogWorker.SendEvent(eventInit, false, server.Catalog); err != nil {
return err
}
server.catalogWorker.SendEvent(eventRun, true, nil)
// Instanciate providers
for providerName, providerConfig := range server.Config.Providers {
server.providers[providerName] = provider.NewProvider(providerName, providerConfig, server.Catalog)
}
if err := server.startProviderWorkers(); err != nil {
return err
}
// Send initial catalog refresh event to provider workers
server.providerWorkers.Broadcast(eventCatalogRefresh, nil)
// Create library instance
server.Library = library.NewLibrary(server.Config, server.Catalog)
go server.Library.Refresh()
// Instanciate serve worker
server.serveWorker = worker.NewWorker()
server.serveWorker.RegisterEvent(eventInit, workerServeInit)
server.serveWorker.RegisterEvent(eventShutdown, workerServeShutdown)
server.serveWorker.RegisterEvent(eventRun, workerServeRun)
if err := server.serveWorker.SendEvent(eventInit, false, server); err != nil {
return err
} else if err := server.serveWorker.SendEvent(eventRun, false, nil); err != nil {
return err
}
server.wg.Wait()
return nil
}