package GoLog

import (
	"context"
	"github.com/olivere/elastic/v7"
)

// 	+---------------------------------------------------------+
// 	+                                                         +
//  +                        EsLog	 					      +
// 	+                                                         +
// 	+---------------------------------------------------------+
// 使用es

type EsLog struct {
	log    *BufferLog
	client *elastic.Client
}

func newEsClient(url string) (*elastic.Client, error) {
	options := []elastic.ClientOptionFunc{
		elastic.SetSniff(false),
		elastic.SetURL(url),
	}
	client, err := elastic.NewClient(
		options...,
	)
	return client, err
}

func NewEsLog(url string, index string) (*EsLog, error) {
	client, err := newEsClient(url)
	if err != nil {
		return nil, err
	}
	bulkRequest := client.Bulk()
	ctx := context.Background()
	writer := func(msg IMsg) {
		r := msg.ToMap()
		indexReq := elastic.NewBulkIndexRequest().Index(index)
		indexReq = indexReq.Doc(r)
		bulkRequest = bulkRequest.Add(indexReq)
		_, _ = bulkRequest.Do(ctx)
	}
	log := NewBufferLog(50, writer)
	return &EsLog{log: log, client: client}, nil
}

func (c *EsLog) Write(msg IMsg) {
	c.log.Write(msg)
}

func (c *EsLog) Stop() {
	c.log.Stop()
	c.client.Stop()
}