quote_service.go

package service

import (
	"encoding/json"
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/imroc/req"
	"github.com/syyongx/php2go"
	"github.com/tidwall/gjson"
	"search_server/model"
	"search_server/pkg/common"
	"search_server/pkg/es"
	"search_server/requests"
	"search_server/service/query"
	"search_server/service/transformer"
	"strconv"
)

type QuoteService struct {
}

//根据供应商获取商品信息
func (qs *QuoteService) GetGoodsBySupplier(ctx *gin.Context,r requests.GetGoodsBySupplierRequest) (result map[string]interface{}, err error) {
	//索引字典
	supplierIndex := query.GetSupplierIndex(r.SupplierId)
	//获取查询条件
	queryString, err := query.GetGoodsBySupplierIdQuery(r)
	if err != nil {
		return
	}
	esResult, err := es.CurlES(supplierIndex, queryString)
	if err != nil {
		return
	}
	goodsIds := GetGoodsIdsByEs(esResult)
	goodsList, err := GetGoodsInfo(ctx,goodsIds)
	dullData, err := transformer.DullDataInfo(ctx,r.Keyword, goodsList)
	//整合供应商数据
	result = GetReturnSupplierData(dullData)
	result["total"] = gjson.Get(esResult, "hits.total").Int()
	if gjson.Get(esResult, "hits.total").Int() == 0 {
		return nil, err
	}
	result["took"] = gjson.Get(esResult, "took").Int()
	return
}

func (qs *QuoteService) GetGoodsBySupplierName(ctx *gin.Context,r requests.GetGoodsBySupplierRequest) (result map[string]interface{}, err error) {
	supplierIndex := r.SupplierName
	queryString, err := query.GetGoodsBySupplierNameQuery(r)
	esResult, err := es.CurlES(supplierIndex, queryString)
	if err != nil {
		return
	}
	fmt.Println(esResult)
	goodsIds := GetGoodsIdsByEs(esResult)
	fmt.Println(goodsIds)
	goodsList, err := GetGoodsInfo(ctx,goodsIds)
	dullData, err := transformer.DullDataInfo(ctx,r.Keyword, goodsList)
	quoteTransformer := transformer.QuoteTransformer{}
	//还要去转一下阶梯价比例
	dullData = quoteTransformer.TransformLadderPrice(dullData)
	result = GetReturnSupplierData(dullData)
	result["total"] = gjson.Get(esResult, "hits.total").Int()
	result["took"] = gjson.Get(esResult, "took").Int()
	return
}

//获取采集那边的数据
func (qs *QuoteService) GetData(keyword, supplier string) (res interface{}, err error) {
	collectData, err := qs.GetCollectData(keyword, supplier)
	if err != nil {
		return
	}
	quoteTransformer := transformer.QuoteTransformer{}
	res = quoteTransformer.TransformCollectData(keyword, supplier, collectData)

	return
}

//去采集那边的接口获取数据
func (qs *QuoteService) GetCollectData(keyword, supplierName string) (collectData model.CollectData, err error) {
	portSet01 := []int{8084, 8084, 8084, 8084, 8085, 8086, 8086, 8086, 8086, 8087}
	portSet02 := []int{8084, 8084, 8084, 8084, 8086, 8086, 8086, 8086}
	portSet03 := []int{8086}
	portSet04 := []int{8087}
	supplierNamePortSetMap := map[string][]int{
		"vnet":       portSet03,
		"arrow":      portSet03,
		"verical":    portSet03,
		"mouser":     portSet02,
		"rs":         portSet02,
		"rochester":  portSet02,
		"master":     portSet04,
		"aipco":      portSet01,
		"alliedelec": portSet01,
		"buerklin":   portSet01,
		"digikey":    portSet01,
		"chip1stop":  portSet01,
		"element14":  portSet01,
		"tme":        portSet01,
	}
	ip := "http://39.108.51.147"
	key, exist := supplierNamePortSetMap[supplierName]
	if !exist {
		return
	}
	var index int
	if len(key) == 1 {
		index = 0
	} else {
		index = common.Rand(0, len(key)-1)
	}
	port := key[index]
	portStr := strconv.Itoa(port)
	keyword = php2go.URLEncode(keyword)
	url := ip + ":" + portStr + "/elec/" + supplierName + "?kw=" + keyword
	resp, err := req.Get(url)
	if err != nil {
		return
	}
	err = json.Unmarshal(resp.Bytes(), &collectData)
	if err != nil {
		return
	}
	return
}