74 lines
1.6 KiB
Go
74 lines
1.6 KiB
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)
|
|
}
|