common_utils/elastic/elastic_test.go

package elastic

import (
	"encoding/json"
	"fmt"
	es "github.com/olivere/elastic/v7"
	"github.com/stretchr/testify/assert"
	"testing"
)

var esClinet Elastic

//func init() {
//	esClinet = Elastic{
//		S_esurl:  "127.0.0.1:19805",
//		I_size:   2,
//		Username: "es_all",
//		Password: "TopJkO2E_d1x",
//	}
//	fmt.Println(esClinet)
//
//	esClinet.InitElasticSize()
//
//}
func TestCount(t *testing.T) {
	esClinet = Elastic{
		S_esurl:  "http://127.0.0.1:19805",
		I_size:   2,
		Username: "es_all",
		Password: "TopJkO2E_d1x",
	}
	esClinet.InitElasticSize()

	rangeQuery := es.NewRangeQuery("id").Gte("6463a8800000000000000000").Lt("6464fa000000000000000000")
	termsQuery := es.NewTermsQuery("site", "元博网（采购与招标网）", "中国招标与采购网")
	boolQuery := es.NewBoolQuery().Must(rangeQuery, termsQuery)

	bytes, err := json.Marshal(boolQuery)
	if err != nil {
		fmt.Println("err", err.Error())
	}

	fmt.Println("qqq", string(bytes))
	count1 := esClinet.Count("bidding", boolQuery)
	fmt.Println(count1)
}

func TestBulkSave(t *testing.T) {
	esClinet = Elastic{
		S_esurl:  "http://127.0.0.1:19805",
		I_size:   2,
		Username: "es_all",
		Password: "TopJkO2E_d1x",
	}
	esClinet.InitElasticSize()

	id := "6398a381063a7b816e072adf"
	esUpdate := map[string]interface{}{
		"area":     "新疆",
		"city":     "伊犁哈萨克自治州",
		"district": "奎屯市",
	}

	updateEs := []map[string]interface{}{
		{"_id": id},
		esUpdate,
	}

	arru := make([][]map[string]interface{}, 0)
	arru = append(arru, updateEs)
	esClinet.UpdateBulk("bidding", arru...)
	fmt.Println(arru)
}

func TestElastic_ExistsIndex(t *testing.T) {
	esClinet = Elastic{
		S_esurl:  "http://127.0.0.1:19805",
		I_size:   2,
		Username: "es_all",
		Password: "TopJkO2E_d1x",
	}
	esClinet.InitElasticSize()

	res, err := esClinet.ExistsIndex("bidding_20231118")
	assert.Equal(t, nil, err)
	assert.Equal(t, true, res)
}

func TestElastic_SetAlias(t *testing.T) {
	esClinet = Elastic{
		S_esurl:  "http://127.0.0.1:19805",
		I_size:   2,
		Username: "es_all",
		Password: "TopJkO2E_d1x",
	}
	esClinet.InitElasticSize()

	err := esClinet.SetAlias("bidding_20231118", "bidding_pre")
	assert.Equal(t, nil, err)
}

func TestElastic_RemoveAlias(t *testing.T) {
	esClinet = Elastic{
		S_esurl:  "http://127.0.0.1:19805",
		I_size:   2,
		Username: "es_all",
		Password: "TopJkO2E_d1x",
	}
	esClinet.InitElasticSize()
	err := esClinet.RemoveAlias("bidding_20231118", "bidding_pre")
	assert.Equal(t, nil, err)

}