47 lines
1.0 KiB
Go
47 lines
1.0 KiB
Go
package es
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"git.trj.tw/golang/mtfosbot/module/config"
|
|
"github.com/olivere/elastic"
|
|
)
|
|
|
|
var client *elastic.Client
|
|
|
|
// NewClient -
|
|
func NewClient() (err error) {
|
|
conf := config.GetConf()
|
|
client, err = elastic.NewClient(
|
|
elastic.SetSniff(false),
|
|
elastic.SetURL(conf.Elasticsearch.Host),
|
|
elastic.SetBasicAuth(conf.Elasticsearch.Username, conf.Elasticsearch.Password),
|
|
)
|
|
fmt.Println("host ", conf.Elasticsearch.Host)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
return
|
|
}
|
|
|
|
// PutLog -
|
|
func PutLog(logType string, body map[string]interface{}) (err error) {
|
|
if client == nil {
|
|
return
|
|
}
|
|
if len(logType) == 0 || body == nil {
|
|
return
|
|
}
|
|
conf := config.GetConf()
|
|
ctx := context.Background()
|
|
body["timestamp"] = time.Now().UnixNano() / 1000000
|
|
body["date"] = time.Now().UTC()
|
|
nt := time.Now()
|
|
idx := fmt.Sprintf("%v-%v-%v-%v", conf.Elasticsearch.Index, nt.Year(), int(nt.Month()), nt.Day())
|
|
_, err = client.Index().Index(idx).Type(logType).BodyJson(body).Do(ctx)
|
|
return
|
|
}
|