mirror of
https://github.com/idanoo/autobrr
synced 2025-07-23 16:59:12 +00:00
0e88117702
* feat(logger): add module context * feat(logger): change errors package * feat(logger): update tests
70 lines
1.6 KiB
Go
70 lines
1.6 KiB
Go
package server
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/rs/zerolog"
|
|
|
|
"github.com/autobrr/autobrr/internal/feed"
|
|
"github.com/autobrr/autobrr/internal/indexer"
|
|
"github.com/autobrr/autobrr/internal/irc"
|
|
"github.com/autobrr/autobrr/internal/logger"
|
|
"github.com/autobrr/autobrr/internal/scheduler"
|
|
)
|
|
|
|
type Server struct {
|
|
log zerolog.Logger
|
|
Hostname string
|
|
Port int
|
|
|
|
indexerService indexer.Service
|
|
ircService irc.Service
|
|
feedService feed.Service
|
|
scheduler scheduler.Service
|
|
|
|
stopWG sync.WaitGroup
|
|
lock sync.Mutex
|
|
}
|
|
|
|
func NewServer(log logger.Logger, ircSvc irc.Service, indexerSvc indexer.Service, feedSvc feed.Service, scheduler scheduler.Service) *Server {
|
|
return &Server{
|
|
log: log.With().Str("module", "server").Logger(),
|
|
indexerService: indexerSvc,
|
|
ircService: ircSvc,
|
|
feedService: feedSvc,
|
|
scheduler: scheduler,
|
|
}
|
|
}
|
|
|
|
func (s *Server) Start() error {
|
|
s.log.Info().Msgf("Starting server. Listening on %v:%v", s.Hostname, s.Port)
|
|
|
|
// start cron scheduler
|
|
s.scheduler.Start()
|
|
|
|
// instantiate indexers
|
|
if err := s.indexerService.Start(); err != nil {
|
|
s.log.Error().Err(err).Msg("Could not start indexer service")
|
|
return err
|
|
}
|
|
|
|
// instantiate and start irc networks
|
|
s.ircService.StartHandlers()
|
|
|
|
// start torznab feeds
|
|
if err := s.feedService.Start(); err != nil {
|
|
s.log.Error().Err(err).Msg("Could not start feed service")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *Server) Shutdown() {
|
|
s.log.Info().Msg("Shutting down server")
|
|
|
|
// stop all irc handlers
|
|
s.ircService.StopHandlers()
|
|
|
|
// stop cron scheduler
|
|
s.scheduler.Stop()
|
|
}
|