123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160 |
- package clog
- import (
- "bytes"
- "encoding/json"
- "errors"
- "fmt"
- "io/ioutil"
- "net/http"
- )
- type slackAttachment struct {
- Text string `json:"text"`
- Color string `json:"color"`
- }
- type slackPayload struct {
- Attachments []slackAttachment `json:"attachments"`
- }
- var slackColors = []string{
- "",
- "#3aa3e3",
- "warning",
- "danger",
- "#ff0200",
- }
- type SlackConfig struct {
-
- Level LEVEL
-
- BufferSize int64
-
- URL string
- }
- type slack struct {
- Adapter
- url string
- }
- func newSlack() Logger {
- return &slack{
- Adapter: Adapter{
- quitChan: make(chan struct{}),
- },
- }
- }
- func (s *slack) Level() LEVEL { return s.level }
- func (s *slack) Init(v interface{}) error {
- cfg, ok := v.(SlackConfig)
- if !ok {
- return ErrConfigObject{"SlackConfig", v}
- }
- if !isValidLevel(cfg.Level) {
- return ErrInvalidLevel{}
- }
- s.level = cfg.Level
- if len(cfg.URL) == 0 {
- return errors.New("URL cannot be empty")
- }
- s.url = cfg.URL
- s.msgChan = make(chan *Message, cfg.BufferSize)
- return nil
- }
- func (s *slack) ExchangeChans(errorChan chan<- error) chan *Message {
- s.errorChan = errorChan
- return s.msgChan
- }
- func buildSlackPayload(msg *Message) (string, error) {
- payload := slackPayload{
- Attachments: []slackAttachment{
- {
- Text: msg.Body,
- Color: slackColors[msg.Level],
- },
- },
- }
- p, err := json.Marshal(&payload)
- if err != nil {
- return "", err
- }
- return string(p), nil
- }
- func (s *slack) write(msg *Message) {
- payload, err := buildSlackPayload(msg)
- if err != nil {
- s.errorChan <- fmt.Errorf("slack: buildSlackPayload: %v", err)
- return
- }
- resp, err := http.Post(s.url, "application/json", bytes.NewReader([]byte(payload)))
- if err != nil {
- s.errorChan <- fmt.Errorf("slack: %v", err)
- return
- }
- defer resp.Body.Close()
- if resp.StatusCode/100 != 2 {
- data, _ := ioutil.ReadAll(resp.Body)
- s.errorChan <- fmt.Errorf("slack: %s", data)
- }
- }
- func (s *slack) Start() {
- LOOP:
- for {
- select {
- case msg := <-s.msgChan:
- s.write(msg)
- case <-s.quitChan:
- break LOOP
- }
- }
- for {
- if len(s.msgChan) == 0 {
- break
- }
- s.write(<-s.msgChan)
- }
- s.quitChan <- struct{}{}
- }
- func (s *slack) Destroy() {
- s.quitChan <- struct{}{}
- <-s.quitChan
- close(s.msgChan)
- close(s.quitChan)
- }
- func init() {
- Register(SLACK, newSlack)
- }
|