first commit

pkg/common/es/es_client_test.go

@@ -0,0 +1,81 @@
+package ess
+
+import (
+	"servicebase/pkg/common/messages"
+	"context"
+	"testing"
+	"time"
+
+	"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)
+}