mirror of
https://github.com/ncarlier/webhookd.git
synced 2025-04-07 15:55:53 +00:00
1dab1e968d
- URL based configuration - Only prefixed output lines are notified - HTTP notifier: send a JSON with notification in the text attribute - SMTP notifier: send an email with notification text in body
79 lines
1.7 KiB
Go
79 lines
1.7 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"time"
|
|
|
|
"github.com/ncarlier/webhookd/pkg/api"
|
|
"github.com/ncarlier/webhookd/pkg/config"
|
|
"github.com/ncarlier/webhookd/pkg/logger"
|
|
"github.com/ncarlier/webhookd/pkg/notification"
|
|
"github.com/ncarlier/webhookd/pkg/worker"
|
|
)
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
|
|
if *version {
|
|
printVersion()
|
|
return
|
|
}
|
|
|
|
conf := config.Get()
|
|
|
|
level := "info"
|
|
if *conf.Debug {
|
|
level = "debug"
|
|
}
|
|
logger.Init(level)
|
|
|
|
logger.Debug.Println("Starting webhookd server...")
|
|
|
|
server := &http.Server{
|
|
Addr: *conf.ListenAddr,
|
|
Handler: api.NewRouter(config.Get()),
|
|
ErrorLog: logger.Error,
|
|
}
|
|
|
|
// Configure notification
|
|
if err := notification.Init(*conf.NotificationURI); err != nil {
|
|
logger.Error.Fatalf("Unable to create notification channel: %v\n", err)
|
|
}
|
|
|
|
// Start the dispatcher.
|
|
logger.Debug.Printf("Starting the dispatcher (%d workers)...\n", *conf.NbWorkers)
|
|
worker.StartDispatcher(*conf.NbWorkers)
|
|
|
|
done := make(chan bool)
|
|
quit := make(chan os.Signal, 1)
|
|
signal.Notify(quit, os.Interrupt)
|
|
|
|
go func() {
|
|
<-quit
|
|
logger.Debug.Println("Server is shutting down...")
|
|
api.Shutdown()
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
|
defer cancel()
|
|
|
|
server.SetKeepAlivesEnabled(false)
|
|
if err := server.Shutdown(ctx); err != nil {
|
|
logger.Error.Fatalf("Could not gracefully shutdown the server: %v\n", err)
|
|
}
|
|
close(done)
|
|
}()
|
|
|
|
logger.Info.Println("Server is ready to handle requests at", *conf.ListenAddr)
|
|
api.Start()
|
|
if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
|
logger.Error.Fatalf("Could not listen on %s: %v\n", *conf.ListenAddr, err)
|
|
}
|
|
|
|
<-done
|
|
logger.Debug.Println("Server stopped")
|
|
}
|