package misc import ( "math" "sync" "git.rosy.net.cn/jx-callback/business/jxstore/cms" "git.rosy.net.cn/jx-callback/business/jxutils" "git.rosy.net.cn/jx-callback/business/jxutils/jxcontext" "git.rosy.net.cn/jx-callback/business/jxutils/tasksch" "git.rosy.net.cn/jx-callback/business/model" "git.rosy.net.cn/jx-callback/business/model/dao" ) const ( DayNum = 30 //请求天数 LimitNum = 100 //最大数据限制 ) func GetStoreSkuSalesInfo(ctx *jxcontext.Context, storeID int) (outStoreSkuSales []*model.StoreSkuSales, err error) { db := dao.GetDB() //得到所有门店 storeList, _ := GetStoreList(ctx) storeList = GetFilterStoreListEx(storeList, nil) storeMapData := make(map[int]*cms.StoreExt) for _, value := range storeList { storeMapData[value.ID] = value } curStoreInfo := storeMapData[storeID] cityCode := curStoreInfo.CityCode //获取本市商品总销量 citySkuSalesCntMap := make(map[int]int) citySkuSalesCntList, _ := dao.GetSkuSalesCntList(db, -1, cityCode, DayNum, LimitNum, nil) citySkuIDs := []int{} for _, value := range citySkuSalesCntList { citySkuSalesCntMap[value.SkuID] = value.Count citySkuIDs = append(citySkuIDs, value.SkuID) } //获取本店商品总销量 storeSkuSalesCntMap := make(map[int]int) storeSkuSalesCntList, _ := dao.GetSkuSalesCntList(db, storeID, cityCode, DayNum, -1, citySkuIDs) for _, value := range storeSkuSalesCntList { storeSkuSalesCntMap[value.SkuID] = value.Count } //获取本店商品差评数量 storeSkuBadCommentCntMap := make(map[int]int) storeSkuBadCommentCntList, _ := dao.GetSkuBadCommentCntList(db, storeID, DayNum) for _, value := range storeSkuBadCommentCntList { storeSkuBadCommentCntMap[value.SkuID] = value.Count } //得到当前门店商品数据 storeSkuData, _ := cms.GetStoreSkus(ctx, storeID, citySkuIDs, true, "", true, map[string]interface{}{}, 0, -1) storeSkuMapData := make(map[int]*cms.StoreSkuNameExt) for _, value := range storeSkuData.SkuNames { for _, skuInfo := range value.Skus2 { storeSkuMapData[skuInfo.SkuID] = value } } //得到5KM内的所有门店 rangeStoreList := GetRangeStoreList(storeID, curStoreInfo.FloatLng, curStoreInfo.FloatLat, SaleSkuCheckRange, storeList) //得到5KM内的所有门店的商品的价格 allStoreSkus := make(map[int]map[int]int) var locker sync.RWMutex taskFunc := func(task *tasksch.ParallelTask, batchItemList []interface{}, params ...interface{}) (retVal interface{}, err error) { storeInfo := batchItemList[0].(*cms.StoreExt) storeID := storeInfo.ID jxSkuInfoData, _ := cms.GetStoreSkus(ctx, storeID, citySkuIDs, true, "", true, map[string]interface{}{}, 0, -1) jxSkuPriceMapData := make(map[int]int) for _, value := range jxSkuInfoData.SkuNames { for _, skuInfo := range value.Skus2 { jxSkuPriceMapData[skuInfo.SkuID] = skuInfo.BindPrice } } locker.Lock() defer locker.Unlock() allStoreSkus[storeID] = jxSkuPriceMapData return retVal, err } taskParallel := tasksch.NewParallelTask("得到所有门店商品", nil, ctx, taskFunc, rangeStoreList) taskParallel.Run() taskParallel.GetResult(0) //计算商品的平均价格 skusTotalPrice := make(map[int]int) skusCount := make(map[int]int) skusAveragePrice := make(map[int]int) for _, storeInfo := range rangeStoreList { storeID := storeInfo.ID storeSkuMapData := allStoreSkus[storeID] for skuID, skuPrice := range storeSkuMapData { skusTotalPrice[skuID] += skuPrice skusCount[skuID]++ } } for id, totalPrice := range skusTotalPrice { skusAveragePrice[id] = int(math.Round(float64(totalPrice) / float64(skusCount[id]))) } //输出商品销量统计结果 for _, value := range citySkuSalesCntList { skuID := value.SkuID if _, ok := storeSkuSalesCntMap[skuID]; ok { storeSkuSales := &model.StoreSkuSales{} storeSkuSales.SkuID = skuID skuInfo := storeSkuMapData[skuID] if skuInfo != nil { storeSkuSales.SkuName = skuInfo.Name storeSkuSales.SkuPrice = jxutils.IntPrice2StandardCurrencyString(int64(skuInfo.Skus2[0].BindPrice)) storeSkuSales.SkuAvgPrice = jxutils.IntPrice2StandardCurrencyString(int64(skusAveragePrice[skuID])) storeSkuSales.BadCommentCnt = storeSkuBadCommentCntMap[skuID] storeSkuSales.StoreSkuSalesCnt = storeSkuSalesCntMap[skuID] storeSkuSales.CitySkuSalesCnt = citySkuSalesCntMap[skuID] outStoreSkuSales = append(outStoreSkuSales, storeSkuSales) } } } return outStoreSkuSales, err }