package ess

import (
	"context"
	"testing"
	"time"

	"gitea.ddegame.cn/open/servicebase/pkg/common/messages"

	"github.com/anxpp/beego/logs"
	"github.com/olivere/elastic/v7"
	"github.com/spf13/viper"
)

func TestClient(t *testing.T) {
	Init()
}

func TestClientCreateUser(t *testing.T) {
	//for _, item := range models.LogUserRegisterAll() {
	//	cli := EsClient{}
	//	_, success, msg := cli.Create("test_user_register", item.Id, item)
	//	if !success {
	//		logs.Error(msg)
	//	}
	//}
}

func TestClientSearch(t *testing.T) {
}

func _testConnect() (c *elastic.Client) {
	c, e := elastic.NewSimpleClient(
		elastic.SetHealthcheck(true),
		elastic.SetHealthcheckInterval(10*time.Second),
		// elastic.SetURL("http://47.97.157.234:9200"),
		elastic.SetURL(viper.GetString("es.default.addr")),
		// elastic.SetBasicAuth(beego.AppConfig.String("es_username"), beego.AppConfig.String("es_password")),
	)
	if e != nil {
		logs.Error("NewClient_error: ", e.Error())
	}
	return c
}

func TestClientNew(t *testing.T) {
	var client EsClient
	client.Create("es_index_message", "003", map[string]interface{}{
		"id":      "003",
		"curTime": 1440570500855,
		"f1":      1234,
		"f2":      "22222",
		"f3":      "22222",
		"f4":      "22222",
		"f5":      "22222",
		"f6":      "2020-08-09 23:37:00",
	})
}

func TestSearchMessage(t *testing.T) {
	index := messages.TagIndex(string(messages.EventTagMessage))
	client := _testConnect()
	var list []elastic.Query
	list = append(list, elastic.NewWildcardQuery("eventType", "1"))
	list = append(list, elastic.NewMatchQuery("msgType", "TEXT"))
	s := client.Search().Index(index)
	for _, query := range list {
		s = s.Query(query)
	}
	res, e := s.From(0).Size(20).Pretty(true).Do(context.Background())
	if e != nil {
		return
	}
	logs.Info(res.Status)
	logs.Info(res.Hits.TotalHits)
	logs.Info(len(res.Hits.Hits))
	for i, item := range res.Hits.Hits {
		b, _ := item.Source.MarshalJSON()
		logs.Info(i, string(b))
	}
	time.Sleep(time.Second * 3)
}