package es

import (
	"SciencesServer/utils"
	"context"
	"encoding/json"
	"fmt"
	"github.com/olivere/elastic/v7"
	"testing"
)

var (
	dataIndex = "test"
	//client    *elastic.Client
	host = "http://192.168.0.188:9200"
)

type Student struct {
	ID   uint64 `json:"id"`
	Name string `json:"name"`
	Age  int64  `json:"age"`
}

func _new() *elastic.Client {
	client, err := elastic.NewClient(elastic.SetSniff(false), elastic.SetURL(host))

	if err != nil {
		panic(err)
	}
	if _, _, err = client.Ping(host).Do(context.Background()); err != nil {
		panic(err)
	}
	return client
}

type ESAchievement struct {
	ID       uint64 `json:"id"`       // 成果ID
	Title    string `json:"title"`    // 成果名称
	Industry string `json:"industry"` // 行业领域
	Keyword  string `json:"keyword"`  // 关键词
}

func TestNewInstanceCreate(t *testing.T) {
	client := _new()
	//client.Delete()

	//src := Student{ID: 2, Name: "henry", Age: 28}
	src := &ESAchievement{
		ID:       6,
		Title:    "测试6",
		Industry: "测试测试",
		Keyword:  "关键词1；关键词2",
	}
	_, err := client.Index().Index("es_achievement_index").Id(fmt.Sprintf("%d", src.ID)).BodyJson(utils.AnyToJSON(src)).
		Do(context.Background())
	t.Log(err)
}

func TestNewInstanceUpdate(t *testing.T) {
	client := _new()
	//client.Delete()

	src := &Student{ID: 1, Name: "Henry", Age: 28}
	_, err := client.Update().Index("es_achievement_index").Id(fmt.Sprintf("%d", src.ID)).Doc(map[string]interface{}{
		"age": 29,
	}).Do(context.Background())
	t.Log(err)
}

func TestNewInstanceSearch(t *testing.T) {
	client := _new()
	//client.Delete()

	query := elastic.NewBoolQuery()
	//query.Must(elastic.NewMatchQuery("age", "27"))

	service := client.Search().Index("es_manage_index_2")

	result, err := service.Pretty(true).Query(query).Do(context.Background())

	t.Log(err)
	t.Log(utils.AnyToJSON(result))

	src := new(Student)

	for _, hit := range result.Hits.Hits {
		if err = json.Unmarshal(hit.Source, src); err != nil {
			t.Log(err)
			return
		}
		fmt.Println(utils.AnyToJSON(src))
	}
}

func TestNewInstanceDelete(t *testing.T) {
	client := _new()
	//client.Delete()

	_, err := client.Delete().Index("es_achievement_index").Id("5").Do(context.Background())

	t.Log(err)
}

func TestNewInstance2(t *testing.T) {
	client := _new()

	query := elastic.NewBoolQuery()
	//query.Must(elastic.NewMatchQuery("title", "一种"))
	//query.Must(elastic.NewMatchPhraseQuery("industry", "你是"))

	service := client.Search().
		//Index("es_patent_index")
		Index("es_achievement_index")

	result, err := service.From(0).Size(10).
		//Explain(true).
		AllowPartialSearchResults(true).
		TrackTotalHits(true).
		TrackScores(true).Query(query).Do(context.Background())

	if err != nil {
		panic(err)
	}
	t.Log(utils.AnyToJSON(result))
	t.Log(utils.AnyToJSON(result.Hits.TotalHits.Value))
	for _, v := range result.Hits.Hits {
		t.Log(*v.Score)
		t.Log(string(v.Source))
	}
}