package main

import (
	"fmt"
	"github.com/olivere/elastic/v7"
	"github.com/xuri/excelize/v2"
	"go.uber.org/zap"
	"jygit.jydev.jianyu360.cn/data_processing/common_utils/log"
	"testing"
)

func TestBiddng(t *testing.T) {
	f, err := excelize.OpenFile("./6月份未匹配到剑鱼的项目清单.xlsx")
	if err != nil {
		fmt.Println(err)
		return
	}
	defer func() {
		f.Save()
		if err := f.Close(); err != nil {
			fmt.Println(err)
		}
	}()

	rows, err := f.GetRows("Sheet1")
	if err != nil {
		fmt.Println(err)
		return
	}

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

	for i := 1; i < len(rows); i++ {
		projectName := rows[i][1]

		res, err := searchES23(client, projectName, "", 60, 20)

		if err != nil {
			fmt.Println(projectName)
			log.Info("TestBiddng", zap.Error(err))
		}
		if len(res) > 0 {
			f.SetCellValue("Sheet1", fmt.Sprintf("F%v", i+1), 1)
		}
		fmt.Printf("%s 总数是：%d\n", projectName, len(res))

	}

}

func TestSearchES23(t *testing.T) {
	client, err := elastic.NewClient(
		elastic.SetURL(GF.Es.URL),
		elastic.SetBasicAuth(GF.Es.Username, GF.Es.Password),
		elastic.SetSniff(false),
	)
	if err != nil {
		log.Fatal("创建 Elasticsearch 客户端失败", zap.Error(err))
	}

	projectName := "赤壁市综合养老中心项目"
	results, err := searchES24(client, projectName, "", 50, 50)
	log.Info("aaa", zap.Any("results", results))
}