85 lines
1.9 KiB
Go
85 lines
1.9 KiB
Go
package logs
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// SLACKWriter implements beego LoggerInterface and is used to send jiaoliao webhook
|
|
type SLACKWriter struct {
|
|
WebhookURL string `json:"webhookurl"`
|
|
Level int `json:"level"`
|
|
formatter LogFormatter
|
|
Formatter string `json:"formatter"`
|
|
}
|
|
|
|
// newSLACKWriter creates jiaoliao writer.
|
|
func newSLACKWriter() Logger {
|
|
res := &SLACKWriter{Level: LevelTrace}
|
|
res.formatter = res
|
|
return res
|
|
}
|
|
|
|
func (s *SLACKWriter) Format(lm *LogMsg) string {
|
|
// text := fmt.Sprintf("{\"text\": \"%s\"}", msg)
|
|
return lm.When.Format("2006-01-02 15:04:05") + " " + lm.OldStyleFormat()
|
|
}
|
|
|
|
func (s *SLACKWriter) SetFormatter(f LogFormatter) {
|
|
s.formatter = f
|
|
}
|
|
|
|
// Init SLACKWriter with json config string
|
|
func (s *SLACKWriter) Init(config string) error {
|
|
res := json.Unmarshal([]byte(config), s)
|
|
|
|
if res == nil && len(s.Formatter) > 0 {
|
|
fmtr, ok := GetFormatter(s.Formatter)
|
|
if !ok {
|
|
return errors.New(fmt.Sprintf("the formatter with name: %s not found", s.Formatter))
|
|
}
|
|
s.formatter = fmtr
|
|
}
|
|
|
|
return res
|
|
}
|
|
|
|
// 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)
|
|
m := make(map[string]string, 1)
|
|
m["text"] = msg
|
|
|
|
body, _ := json.Marshal(m)
|
|
// resp, err := http.PostForm(s.WebhookURL, form)
|
|
resp, err := http.Post(s.WebhookURL, "application/json", bytes.NewReader(body))
|
|
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)
|
|
}
|