Tidying up
This commit is contained in:
@@ -1,174 +0,0 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
"time"
|
||||
|
||||
"github.com/igodwin/notifier/api/rest"
|
||||
"github.com/igodwin/notifier/internal/config"
|
||||
"github.com/igodwin/notifier/internal/domain"
|
||||
"github.com/igodwin/notifier/internal/notifier"
|
||||
"github.com/igodwin/notifier/internal/queue"
|
||||
"github.com/igodwin/notifier/internal/service"
|
||||
)
|
||||
|
||||
func main() {
|
||||
// Load configuration
|
||||
cfg, err := config.Load("")
|
||||
if err != nil {
|
||||
log.Printf("Warning: failed to load config, using defaults: %v", err)
|
||||
cfg = getDefaultConfig()
|
||||
}
|
||||
|
||||
log.Printf("Starting Notifier REST Server on port %d", cfg.Server.RESTPort)
|
||||
|
||||
// Create context
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
// Initialize queue
|
||||
var q domain.Queue
|
||||
if cfg.Queue.Type == "local" {
|
||||
q, err = queue.NewLocalQueue(cfg.Queue.Local)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to create queue: %v", err)
|
||||
}
|
||||
log.Println("Using local queue")
|
||||
} else {
|
||||
log.Fatalf("Queue type %s not implemented yet", cfg.Queue.Type)
|
||||
}
|
||||
|
||||
// Initialize notifier factory
|
||||
factory := notifier.NewFactory()
|
||||
|
||||
// Register notifiers based on configuration
|
||||
if cfg.Notifiers.Stdout {
|
||||
stdoutNotifier := notifier.NewStdoutNotifier()
|
||||
if err := factory.RegisterNotifier(domain.TypeStdout, stdoutNotifier); err != nil {
|
||||
log.Fatalf("Failed to register stdout notifier: %v", err)
|
||||
}
|
||||
log.Println("Registered stdout notifier")
|
||||
}
|
||||
|
||||
if cfg.Notifiers.SMTP != nil {
|
||||
smtpNotifier, err := notifier.NewSMTPNotifier(cfg.Notifiers.SMTP)
|
||||
if err != nil {
|
||||
log.Printf("Warning: failed to create SMTP notifier: %v", err)
|
||||
} else {
|
||||
if err := factory.RegisterNotifier(domain.TypeEmail, smtpNotifier); err != nil {
|
||||
log.Fatalf("Failed to register SMTP notifier: %v", err)
|
||||
}
|
||||
log.Println("Registered SMTP notifier")
|
||||
}
|
||||
}
|
||||
|
||||
if cfg.Notifiers.Slack != nil {
|
||||
slackNotifier, err := notifier.NewSlackNotifier(cfg.Notifiers.Slack)
|
||||
if err != nil {
|
||||
log.Printf("Warning: failed to create Slack notifier: %v", err)
|
||||
} else {
|
||||
if err := factory.RegisterNotifier(domain.TypeSlack, slackNotifier); err != nil {
|
||||
log.Fatalf("Failed to register Slack notifier: %v", err)
|
||||
}
|
||||
log.Println("Registered Slack notifier")
|
||||
}
|
||||
}
|
||||
|
||||
if cfg.Notifiers.Ntfy != nil {
|
||||
ntfyNotifier, err := notifier.NewNtfyNotifier(cfg.Notifiers.Ntfy)
|
||||
if err != nil {
|
||||
log.Printf("Warning: failed to create Ntfy notifier: %v", err)
|
||||
} else {
|
||||
if err := factory.RegisterNotifier(domain.TypeNtfy, ntfyNotifier); err != nil {
|
||||
log.Fatalf("Failed to register Ntfy notifier: %v", err)
|
||||
}
|
||||
log.Println("Registered Ntfy notifier")
|
||||
}
|
||||
}
|
||||
|
||||
// Check if any notifiers are registered
|
||||
if len(factory.SupportedTypes()) == 0 {
|
||||
log.Fatal("No notifiers configured. Please enable at least one notifier in config.yaml")
|
||||
}
|
||||
|
||||
log.Printf("Supported notification types: %v", factory.SupportedTypes())
|
||||
|
||||
// Create notification service
|
||||
svc := service.NewNotificationService(factory, q, cfg.Queue.WorkerCount)
|
||||
|
||||
// Start workers
|
||||
if err := svc.Start(ctx); err != nil {
|
||||
log.Fatalf("Failed to start service: %v", err)
|
||||
}
|
||||
log.Printf("Started %d worker(s)", cfg.Queue.WorkerCount)
|
||||
|
||||
// Create REST router
|
||||
router := rest.NewRouter(svc)
|
||||
|
||||
// Create HTTP server
|
||||
addr := fmt.Sprintf("%s:%d", cfg.Server.Host, cfg.Server.RESTPort)
|
||||
server := &http.Server{
|
||||
Addr: addr,
|
||||
Handler: router,
|
||||
ReadTimeout: 15 * time.Second,
|
||||
WriteTimeout: 15 * time.Second,
|
||||
IdleTimeout: 60 * time.Second,
|
||||
}
|
||||
|
||||
// Start server in goroutine
|
||||
go func() {
|
||||
log.Printf("REST server listening on %s", addr)
|
||||
if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
||||
log.Fatalf("Failed to start server: %v", err)
|
||||
}
|
||||
}()
|
||||
|
||||
// Wait for interrupt signal
|
||||
sigChan := make(chan os.Signal, 1)
|
||||
signal.Notify(sigChan, os.Interrupt, syscall.SIGTERM)
|
||||
<-sigChan
|
||||
|
||||
log.Println("Shutting down server...")
|
||||
|
||||
// Graceful shutdown
|
||||
shutdownCtx, shutdownCancel := context.WithTimeout(context.Background(), 30*time.Second)
|
||||
defer shutdownCancel()
|
||||
|
||||
if err := server.Shutdown(shutdownCtx); err != nil {
|
||||
log.Printf("Error during server shutdown: %v", err)
|
||||
}
|
||||
|
||||
// Stop service
|
||||
if err := svc.Stop(); err != nil {
|
||||
log.Printf("Error stopping service: %v", err)
|
||||
}
|
||||
|
||||
log.Println("Server stopped")
|
||||
}
|
||||
|
||||
// getDefaultConfig returns a minimal default configuration
|
||||
func getDefaultConfig() *config.Config {
|
||||
return &config.Config{
|
||||
Server: config.ServerConfig{
|
||||
RESTPort: 8080,
|
||||
Host: "0.0.0.0",
|
||||
},
|
||||
Queue: domain.QueueConfig{
|
||||
Type: "local",
|
||||
WorkerCount: 5,
|
||||
RetryAttempts: 3,
|
||||
Local: &domain.LocalQueueConfig{
|
||||
BufferSize: 1000,
|
||||
},
|
||||
},
|
||||
Notifiers: config.NotifiersConfig{
|
||||
Stdout: true,
|
||||
},
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user