Beego/pkg/logs/slack.go

package logs

import (
	"encoding/json"
	"fmt"
	"net/http"
	"net/url"

	"github.com/astaxie/beego/pkg/common"
)

// SLACKWriter implements beego LoggerInterface and is used to send jiaoliao webhook
type SLACKWriter struct {
	WebhookURL         string `json:"webhookurl"`
	Level              int    `json:"level"`
	UseCustomFormatter bool
	CustomFormatter    func(*LogMsg) string
}

// newSLACKWriter creates jiaoliao writer.
func newSLACKWriter() Logger {
	return &SLACKWriter{Level: LevelTrace}
}

func (s *SLACKWriter) Format(lm *LogMsg) string {
	return lm.Msg
}

// Init SLACKWriter with json config string
func (s *SLACKWriter) Init(jsonConfig string, opts ...common.SimpleKV) error {
	// 	if elem != nil {
	// 		s.UseCustomFormatter = true
	// 		s.CustomFormatter = elem
	// 	}
	// }

	return json.Unmarshal([]byte(jsonConfig), s)
}

// WriteMsg write message in smtp writer.
// Sends an email with subject and only this message.
func (s *SLACKWriter) WriteMsg(lm *LogMsg) error {
	if lm.Level > s.Level {
		return nil
	}
	msg := s.Format(lm)
	text := fmt.Sprintf("{\"text\": \"%s %s\"}", lm.When.Format("2006-01-02 15:04:05"), msg)

	form := url.Values{}
	form.Add("payload", text)

	resp, err := http.PostForm(s.WebhookURL, form)
	if err != nil {
		return err
	}
	defer resp.Body.Close()
	if resp.StatusCode != http.StatusOK {
		return fmt.Errorf("Post webhook failed %s %d", resp.Status, resp.StatusCode)
	}
	return nil
}

// Flush implementing method. empty.
func (s *SLACKWriter) Flush() {
}

// Destroy implementing method. empty.
func (s *SLACKWriter) Destroy() {
}

func init() {
	Register(AdapterSlack, newSLACKWriter)
}