mirror of
https://github.com/astaxie/beego.git
synced 2024-12-23 11:40:49 +00:00
124 lines
2.5 KiB
Go
124 lines
2.5 KiB
Go
package es
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"net/url"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/elastic/go-elasticsearch/v6"
|
|
"github.com/elastic/go-elasticsearch/v6/esapi"
|
|
|
|
"github.com/astaxie/beego/pkg/infrastructure/logs"
|
|
)
|
|
|
|
// NewES returns a LoggerInterface
|
|
func NewES() logs.Logger {
|
|
cw := &esLogger{
|
|
Level: logs.LevelDebug,
|
|
}
|
|
return cw
|
|
}
|
|
|
|
// esLogger will log msg into ES
|
|
// before you using this implementation,
|
|
// please import this package
|
|
// usually means that you can import this package in your main package
|
|
// for example, anonymous:
|
|
// import _ "github.com/astaxie/beego/logs/es"
|
|
type esLogger struct {
|
|
*elasticsearch.Client
|
|
DSN string `json:"dsn"`
|
|
Level int `json:"level"`
|
|
formatter logs.LogFormatter
|
|
Formatter string `json:"formatter"`
|
|
}
|
|
|
|
func (el *esLogger) Format(lm *logs.LogMsg) string {
|
|
|
|
msg := lm.OldStyleFormat()
|
|
idx := LogDocument{
|
|
Timestamp: lm.When.Format(time.RFC3339),
|
|
Msg: msg,
|
|
}
|
|
body, err := json.Marshal(idx)
|
|
if err != nil {
|
|
return msg
|
|
}
|
|
return string(body)
|
|
}
|
|
|
|
func (el *esLogger) SetFormatter(f logs.LogFormatter) {
|
|
el.formatter = f
|
|
}
|
|
|
|
// {"dsn":"http://localhost:9200/","level":1}
|
|
func (el *esLogger) Init(config string) error {
|
|
|
|
err := json.Unmarshal([]byte(config), el)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if el.DSN == "" {
|
|
return errors.New("empty dsn")
|
|
} else if u, err := url.Parse(el.DSN); err != nil {
|
|
return err
|
|
} else if u.Path == "" {
|
|
return errors.New("missing prefix")
|
|
} else {
|
|
conn, err := elasticsearch.NewClient(elasticsearch.Config{
|
|
Addresses: []string{el.DSN},
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
el.Client = conn
|
|
}
|
|
if len(el.Formatter) > 0 {
|
|
fmtr, ok := logs.GetFormatter(el.Formatter)
|
|
if !ok {
|
|
return errors.New(fmt.Sprintf("the formatter with name: %s not found", el.Formatter))
|
|
}
|
|
el.formatter = fmtr
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// WriteMsg writes the msg and level into es
|
|
func (el *esLogger) WriteMsg(lm *logs.LogMsg) error {
|
|
if lm.Level > el.Level {
|
|
return nil
|
|
}
|
|
|
|
msg := el.formatter.Format(lm)
|
|
|
|
req := esapi.IndexRequest{
|
|
Index: fmt.Sprintf("%04d.%02d.%02d", lm.When.Year(), lm.When.Month(), lm.When.Day()),
|
|
DocumentType: "logs",
|
|
Body: strings.NewReader(msg),
|
|
}
|
|
_, err := req.Do(context.Background(), el.Client)
|
|
return err
|
|
}
|
|
|
|
// Destroy is a empty method
|
|
func (el *esLogger) Destroy() {
|
|
}
|
|
|
|
// Flush is a empty method
|
|
func (el *esLogger) Flush() {
|
|
|
|
}
|
|
|
|
type LogDocument struct {
|
|
Timestamp string `json:"timestamp"`
|
|
Msg string `json:"msg"`
|
|
}
|
|
|
|
func init() {
|
|
logs.Register(logs.AdapterEs, NewES)
|
|
}
|