96 lines
2.1 KiB
Go
96 lines
2.1 KiB
Go
package logs
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
|
|
"github.com/astaxie/beego/pkg/common"
|
|
)
|
|
|
|
// JLWriter implements beego LoggerInterface and is used to send jiaoliao webhook
|
|
type JLWriter struct {
|
|
AuthorName string `json:"authorname"`
|
|
Title string `json:"title"`
|
|
WebhookURL string `json:"webhookurl"`
|
|
RedirectURL string `json:"redirecturl,omitempty"`
|
|
ImageURL string `json:"imageurl,omitempty"`
|
|
Level int `json:"level"`
|
|
customFormatter func(*LogMsg) string
|
|
}
|
|
|
|
// newJLWriter creates jiaoliao writer.
|
|
func newJLWriter() Logger {
|
|
return &JLWriter{Level: LevelTrace}
|
|
}
|
|
|
|
// Init JLWriter with json config string
|
|
func (s *JLWriter) Init(jsonConfig string, opts ...common.SimpleKV) error {
|
|
for _, elem := range opts {
|
|
if elem.Key == "formatter" {
|
|
formatter, err := GetFormatter(elem)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
s.customFormatter = formatter
|
|
}
|
|
}
|
|
|
|
return json.Unmarshal([]byte(jsonConfig), s)
|
|
}
|
|
|
|
func (s *JLWriter) Format(lm *LogMsg) string {
|
|
return lm.Msg
|
|
}
|
|
|
|
// WriteMsg writes message in smtp writer.
|
|
// Sends an email with subject and only this message.
|
|
func (s *JLWriter) WriteMsg(lm *LogMsg) error {
|
|
if lm.Level > s.Level {
|
|
return nil
|
|
}
|
|
|
|
text := ""
|
|
|
|
if s.customFormatter != nil {
|
|
text = fmt.Sprintf("%s %s", lm.When.Format("2006-01-02 15:04:05"), s.customFormatter(lm))
|
|
} else {
|
|
text = fmt.Sprintf("%s %s", lm.When.Format("2006-01-02 15:04:05"), s.Format(lm))
|
|
|
|
}
|
|
|
|
form := url.Values{}
|
|
form.Add("authorName", s.AuthorName)
|
|
form.Add("title", s.Title)
|
|
form.Add("text", text)
|
|
if s.RedirectURL != "" {
|
|
form.Add("redirectUrl", s.RedirectURL)
|
|
}
|
|
if s.ImageURL != "" {
|
|
form.Add("imageUrl", s.ImageURL)
|
|
}
|
|
|
|
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 *JLWriter) Flush() {
|
|
}
|
|
|
|
// Destroy implementing method. empty.
|
|
func (s *JLWriter) Destroy() {
|
|
}
|
|
|
|
func init() {
|
|
Register(AdapterJianLiao, newJLWriter)
|
|
}
|