Andreas/watchtower
1
0
Fork 0
mirror of https://github.com/containrrr/watchtower.git synced 2025-12-16 15:10:12 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(Discord): send each log entry as Discord message

Browse source
This commit is contained in:
max 2018-03-02 23:00:47 +02:00
parent 190f8b5f56
commit 7724a47fb8
1 changed files with 19 additions and 43 deletions
Download patch file Download diff file Expand all files Collapse all files

62
notifications/discord.go
View file

@ -6,7 +6,7 @@ import (
"fmt" "fmt"
"net/http" "net/http"
log "github.com/Sirupsen/logrus" log "github.com/sirupsen/logrus"
"github.com/urfave/cli" "github.com/urfave/cli"
) )
@ -15,13 +15,14 @@ const (
) )
type discordTypeNotifier struct { type discordTypeNotifier struct {
WebhookURL string webHookURL string
entries []*log.Entry acceptedLevels []log.Level
} }
func newDiscordNotifier(c *cli.Context) typeNotifier { func newDiscordNotifier(c *cli.Context, acceptedLogLevels []log.Level) typeNotifier {
n := &discordTypeNotifier{ n := &discordTypeNotifier{
WebhookURL: c.GlobalString("notification-discord-webhook"), webHookURL: c.GlobalString("notification-discord-webhook"),
acceptedLevels: acceptedLogLevels,
} }
log.AddHook(n) log.AddHook(n)
@ -29,22 +30,18 @@ func newDiscordNotifier(c *cli.Context) typeNotifier {
return n return n
} }
func (n *discordTypeNotifier) sendEntries(entries []*log.Entry) { func (n *discordTypeNotifier) sendEntry(entry *log.Entry) {
// Do the sending in a separate goroutine so we don't block the main process.
message := ""
for _, entry := range entries {
message += entry.Time.Format("2006-01-02 15:04:05") + " (" + entry.Level.String() + "): " + entry.Message + "\r\n"
}
message := "(" + entry.Level.String() + "): " + entry.Message
go func() { go func() {
webhookBody := discordWebhookBody{Content: message} webHookBody := discordWebHookBody{Content: message}
jsonBody, err := json.Marshal(webhookBody) jsonBody, err := json.Marshal(webHookBody)
if err != nil { if err != nil {
fmt.Println("Failed to build JSON body for Discord notificattion: ", err) fmt.Println("Failed to build JSON body for Discord notificattion: ", err)
return return
} }
resp, err := http.Post(n.WebhookURL, "application/json", bytes.NewBuffer([]byte(jsonBody))) resp, err := http.Post(n.webHookURL, "application/json", bytes.NewBuffer([]byte(jsonBody)))
if err != nil { if err != nil {
fmt.Println("Failed to send Discord notificattion: ", err) fmt.Println("Failed to send Discord notificattion: ", err)
} }
@ -57,40 +54,19 @@ func (n *discordTypeNotifier) sendEntries(entries []*log.Entry) {
}() }()
} }
func (n *discordTypeNotifier) StartNotification() { func (n *discordTypeNotifier) StartNotification() {}
if n.entries == nil {
n.entries = make([]*log.Entry, 0, 10)
}
}
func (n *discordTypeNotifier) SendNotification() { func (n *discordTypeNotifier) SendNotification() {}
if n.entries != nil && len(n.entries) != 0 {
n.sendEntries(n.entries)
}
n.entries = nil
}
func (n *discordTypeNotifier) Levels() []log.Level {
// TODO: Make this configurable.
return []log.Level{
log.PanicLevel,
log.FatalLevel,
log.ErrorLevel,
log.WarnLevel,
log.InfoLevel,
}
}
func (n *discordTypeNotifier) Fire(entry *log.Entry) error { func (n *discordTypeNotifier) Fire(entry *log.Entry) error {
if n.entries != nil { n.sendEntry(entry)
n.entries = append(n.entries, entry)
} else {
// Log output generated outside a cycle is sent immediately.
n.sendEntries([]*log.Entry{entry})
}
return nil return nil
} }
type discordWebhookBody struct { func (e *discordTypeNotifier) Levels() []log.Level {
return e.acceptedLevels
}
type discordWebHookBody struct {
Content string `json:"content"` Content string `json:"content"`
} }