package main

import (
	"fmt"
	"github.com/olivere/elastic/v7"
	"log"
	"testing"
)

func TestMergeLabelData(t *testing.T) {
	labelDatas := []LabelData{
		{Name: "A", Sfield: "SF1", TotalWeight: 2.0},
		{Name: "A", Sfield: "SF1", TotalWeight: 3.0},
		{Name: "A", Sfield: "SF1", TotalWeight: 1.5},
		{Name: "B", Sfield: "SF2", TotalWeight: 1.5},
		{Name: "B", Sfield: "SF2", TotalWeight: 0.5},
		{Name: "C", Sfield: "SF3", TotalWeight: 1.5},
	}

	mergedMap := MergeLabelData(labelDatas)
	for key, values := range mergedMap {
		fmt.Printf("Sfield: %s\n", key)
		for _, value := range values {
			fmt.Printf("  Name: %s, Weight: %f\n", value.Name, value.Weight)
		}
	}

	return
}

func TestUpdateIndexByCompanyName(t *testing.T) {
	//url := "http://172.17.4.184:19908"
	url := "http://127.0.0.1:19908"
	username := "jybid"
	password := "Top2023_JEB01i@31"
	index := "ent_info" //索引名称
	// 创建 Elasticsearch 客户端
	client, err := elastic.NewClient(
		elastic.SetURL(url),
		elastic.SetBasicAuth(username, password),
		elastic.SetSniff(false),
	)
	if err != nil {
		log.Fatalf("创建 Elasticsearch 客户端失败：%s", err)
	}

	update := map[string]interface{}{
		"tag_labels": []string{"工业和信息化"},
	}

	err = UpdateIndexByID(client, index, "04e0ce5595e74803a123779a94948ac0", update)
	if err != nil {
		log.Println(err)
	}
}