get_data_controller.go

package controller

import (
	"github.com/gin-gonic/gin"
	"github.com/gomodule/redigo/redis"
	"github.com/ichunt2019/logger"
	"github.com/syyongx/php2go"
	"github.com/tidwall/gjson"
	"search_server/model"
	"search_server/pkg/common"
	"search_server/pkg/config"
	"search_server/pkg/es"
	"search_server/pkg/gredis"
	"search_server/requests"
	"search_server/service"
	"search_server/service/ly"
	"search_server/service/transformer"
	"strconv"
	"strings"
)

func GetDataIndex(ctx *gin.Context) {
	var r requests.GetDataIndexRequest
	_ = ctx.ShouldBind(&r)
	if r.Keyword != "" {
		word, _ := php2go.URLDecode(r.Keyword)
		r.Keyword = strings.ToUpper(word)
	}
	if r.KNums != 0 {
		r.Stock = r.KNums
	}
	//获取所有合并供应商列表
	supplierList := config.GetSectionValues("SEARCH_SUPPLIER_MERGE")
	supplierName := config.Get("SUPPLIER_ALL." + r.SupplierId).String()
	if supplierName == "" && r.IsSort != 2 {
		ctx.JSON(200, common.ErrorResponse(1, "供应商id有误"))
		return
	}

	common.PrintDebugHeader(ctx)
	common.PrintDebugHtml(ctx,r.SupplierId)
	common.PrintDebugHtml(ctx,r.Keyword)

	if r.SupplierId != "" && r.IsSort == 2 {
		//所有供应商列表
		//如果供应商不在合并供应商列表，则实时更新数据
		if supplierName != "" && !php2go.InArray(supplierName, supplierList) {
			redisCon := gredis.Conn("search_w")
			defer redisCon.Close()
			//获取stockFlag
			res, err := redis.Bool(redisCon.Do("SADD", "search_keyword_member", r.Keyword+"_"+r.SupplierName))
			if err != nil {
				logger.Error("%s", err.Error())
			}
			if res {
				//增加联营外链请求
                outLinkDiv(ctx,&r.SupplierId,&supplierName,&r.Keyword)
				//增加
				_, err := redis.Bool(redisCon.Do("INCR", config.Get("redis_all.SEARCH_API_TOTAL_PRE").String()+r.SupplierName))
				if err != nil {
					logger.Error("%s", err.Error())
				}
			}
			r.FirstSearchPage = 2
			quoteService := service.QuoteService{}
			result, _ := quoteService.GetGoodsBySupplier(ctx,r.GetGoodsBySupplierRequest)
			common.Output(ctx,1, "", result)
			return
		}
	}
	//查询并输出数据
	//获取除了专卖的供应商组装查询条件
	queryStr := getDataQueryStr(r, supplierList)
	esResult, err := es.CurlESMSearch(queryStr)
	if err != nil {
		return
	}
	goodsIds := service.GetGoodsIdsByEsMsearch(esResult)
	goodsList, err := service.GetGoodsInfo(ctx,goodsIds)
	dullData, err := transformer.DullDataInfo(ctx,r.Keyword, goodsList)
	result := service.GetReturnSupplierData(dullData)
	result["total"] = gjson.Get(esResult, "hits.total").Int()
	result["took"] = gjson.Get(esResult, "took").Int()
	common.Output(ctx,0, "", result)
	return
}

/*
 联营外链请求
@param supplierId 渠道id
@param supplierName 渠道名称
@param keyword 关键词
 */
func outLinkDiv(ctx *gin.Context,supplierId *string,supplierName *string,keyWord * string){

	var apiData map[string]*model.LyClearGoodsList = nil
	switch *supplierName {
		case "mouser": //调外链拿数据mouser外链请求更新修改
			apiData = ly.OutLinkMouser(ctx,keyWord)
			break;
	}

	if apiData != nil {
		//供应商详情
		supplierId,_ := strconv.ParseInt(*supplierId, 10, 64)
		lycon := ly.CommonLyService{}
		lycon.GetSkuByGoodsSn(ctx,apiData,&model.SUPPLIER_REDIS_INFO{
			SupplierId:supplierId,
			SupplierNickname: *supplierName,
		})
	}
}

//获取查询es的条件语句
func getDataQueryStr(r requests.GetDataIndexRequest, supplierList []string) string {
	getDataService := service.GetDataService{}
	query, _ := getDataService.GetQuery(r.Keyword, r.SupplierName, r.BrandId, r.GoodsNameEq, r.Stock)
	//获取专卖的组装查询条件
	zhuanmaiQuery, _ := getDataService.GetQuery(r.Keyword, "zhuanmai", "", r.GoodsNameEq, r.Stock)
	//类似物料搜索的组装查询条件
	alikeQuery, _ := getDataService.GetQuery(r.Keyword, "alike", "", r.GoodsNameEq, r.Stock)
	//类似物料供应商列表
	alikeSuppliers := config.GetSectionValues("ALIKE_SUPPLIER")
	//构建msearch搜索json
	var queryStr string
	for _, supplierName := range supplierList {
		indexQueryStr := `{"index":"` + supplierName + `","type":"goods","preference":"_primary_first"}`
		if php2go.InArray(supplierName, alikeSuppliers) {
			queryStr = queryStr + indexQueryStr + "\n" + alikeQuery + "\n"
		}
		if supplierName == "zhuanmai" {
			queryStr = queryStr + indexQueryStr + "\n" + zhuanmaiQuery + "\n"
		} else {
			queryStr = queryStr + indexQueryStr + "\n" + query + "\n"
		}
	}
	return queryStr
}