并发问题

service/service_spu.go

@@ -9,6 +9,7 @@ import (
 	"go_sku_server/pkg/gredis"
 	"go_sku_server/pkg/vars"
 	"strings"
+	"sync"
 )
 
 var UsdRate float64                          //美金匯率
@@ -94,10 +95,11 @@ func (ss *SpuService) GetSpuList(skuArr map[string]string) map[string]string {
 // 设置查询公共变量
 func (ss *SpuService) SetInit(redisLySkuArr map[string]string, zyGoodsId, lyGoodsId []string) {
 
-	needRawGoodsIds := make([]string, 0) //批量获取 原始编码
-	needExtendFeeId := make([]string, 0) //批量获取 附加费
-	canals := make([]string, 0)          //所有渠道，目前只有专卖有
-	supplierIds := make([]string, 0)     //所有供应商查询货期的key,排除专卖
+	AllDelivery = make(map[string]map[string]string, 0) //获取供应商货期
+	needRawGoodsIds := make([]string, 0)                //批量获取 原始编码
+	needExtendFeeId := make([]string, 0)                //批量获取 附加费
+	canals := make([]string, 0)                         //所有渠道，目前只有专卖有
+	supplierIds := make([]string, 0)                    //所有供应商查询货期的key,排除专卖
 	//zhuanmaiGoodsIds := make([]string, 0) //专卖的所有goods_id
 	sourceIds := make([]string, 0) //所有来源id
 
@@ -134,160 +136,233 @@ func (ss *SpuService) SetInit(redisLySkuArr map[string]string, zyGoodsId, lyGood
 			needExtendFeeId = append(needExtendFeeId, feekey)
 		}
 	}
-	//批量获取汇率
-	AllCurrencyRate = gredis.Hmget("default_r", "erp_rate", []string{"1", "2", "3", "4", "5", "6", "7", "8", "9", "10", "11", "12", "13"})
-	UsdRate = gconv.Float64(AllCurrencyRate["2"])
-
 	//批量获取标签a
 	GoodsSourceTags = gredis.Hmget("default_r", "goods_source_tags", []string{"1", "2", "3", "4", "5", "6", "7", "8", "9", "10", "11", "12"})
 
 	//批量获取标签b
-	GoodsLabelTags = gredis.Hmget("default_r", "goods_source_tags", []string{"0", "1", "2", "3", "4", "5", "6"})
+	GoodsLabelTags = gredis.Hmget("default_r", "goods_label_tags", []string{"0", "1", "2", "3", "4", "5", "6"})
+
+	var wg sync.WaitGroup
+	wg.Add(1)
+	go func() {
+		defer wg.Done()
+		go BatchCurrency() //批量获取汇率
+	}()
+
+	wg.Add(1)
+	go func() {
+		defer wg.Done()
+		go BatchLyRawGoods(needRawGoodsIds) //批量获取原始编码
+	}()
+
+	wg.Add(1)
+	go func() {
+		defer wg.Done()
+		go BatchZyLockStock(zyGoodsId) //批量获取锁库信息
+	}()
+
+	wg.Add(1)
+	go func() {
+		defer wg.Done()
+		go BatchLyFee(needExtendFeeId) //查询附加费
+	}()
+
+	wg.Add(1)
+	go func() {
+		defer wg.Done()
+		go BatchZhuanmaiDelivery(canals) //专营货期
+	}()
 
-	//批量获取原始编码
-	if len(needRawGoodsIds) > 0 {
-		AllSkuRawMap = gredis.Hmget("default_r", "sku_raw_map", needRawGoodsIds)
+	wg.Add(1)
+	go func() {
+		defer wg.Done()
+		go BatchSeaDelivery(supplierIds) //代购货期
+	}()
+
+	wg.Add(1)
+	go func() {
+		defer wg.Done()
+		go BatchLyTags(redisLySkuArr, lyGoodsId) // 批量获取联营tags
+	}()
+
+	wg.Wait()
+
+}
+
+// 批量获取汇率
+func BatchCurrency() {
+	AllCurrencyRate = gredis.Hmget("default_r", "erp_rate", []string{"1", "2", "3", "4", "5", "6", "7", "8", "9", "10", "11", "12", "13"})
+	UsdRate = gconv.Float64(AllCurrencyRate["2"])
+}
+
+// 批量获取原始编码
+func BatchLyRawGoods(needRawGoodsIds []string) {
+	if len(needRawGoodsIds) == 0 {
+		return
 	}
-	//批量获取锁库信息
-	if len(zyGoodsId) > 0 {
-		AllSkuLockMap = gredis.Hmget("default_r", "sku_lock_stock", zyGoodsId) //所有锁库信息
+	AllSkuRawMap = gredis.Hmget("default_r", "sku_raw_map", needRawGoodsIds)
+}
+
+// 查询自营锁库信息
+func BatchZyLockStock(zyGoodsId []string) {
+	if len(zyGoodsId) == 0 {
+		return
 	}
-	//查询附加费
-	if len(needExtendFeeId) > 0 {
-		AllExtendFee = make(map[string]interface{})
-		redisFeeArr := gredis.Hmget("default_r", "supp_extend_fee", needExtendFeeId)
-		for keyOne, feeone := range redisFeeArr {
-			if feeone == "" {
-				AllExtendFee[keyOne] = false
-				continue
-			}
-			if php2go.Stripos(keyOne, "17.", 0) > -1 {
-				if strings.Contains(feeone, "cn") {
-					var extendFee ExtendFee
-					json.Unmarshal([]byte(feeone), &extendFee)
-					AllExtendFee[keyOne] = extendFee
-				} else {
-					var extendFee []map[string]interface{}
-					json.Unmarshal([]byte(feeone), &extendFee)
-					AllExtendFee[keyOne] = extendFee
-				}
-			} else {
+	AllSkuLockMap = gredis.Hmget("default_r", "sku_lock_stock", zyGoodsId) //所有锁库信息
+}
+
+// 查询附加费
+func BatchLyFee(needExtendFeeId []string) {
+	if len(needExtendFeeId) == 0 {
+		return
+	}
+	AllExtendFee = make(map[string]interface{})
+	redisFeeArr := gredis.Hmget("default_r", "supp_extend_fee", needExtendFeeId)
+	for keyOne, feeone := range redisFeeArr {
+		if feeone == "" {
+			AllExtendFee[keyOne] = false
+			continue
+		}
+		if php2go.Stripos(keyOne, "17.", 0) > -1 {
+			if strings.Contains(feeone, "cn") {
 				var extendFee ExtendFee
 				json.Unmarshal([]byte(feeone), &extendFee)
 				AllExtendFee[keyOne] = extendFee
+			} else {
+				var extendFee []map[string]interface{}
+				json.Unmarshal([]byte(feeone), &extendFee)
+				AllExtendFee[keyOne] = extendFee
 			}
+		} else {
+			var extendFee ExtendFee
+			json.Unmarshal([]byte(feeone), &extendFee)
+			AllExtendFee[keyOne] = extendFee
 		}
 	}
-	//获取供应商货期
-	AllDelivery = make(map[string]map[string]string, 0)
-	if len(canals) > 0 { //专营货期
-		supplierRatios := gredis.Hmget("default_r", "supp_ratio", canals)
-		for dkey, supplierRatio := range supplierRatios {
-			if supplierRatio == "" {
-				AllDelivery[dkey]["cn_delivery"] = ""
-				AllDelivery[dkey]["hk_delivery"] = ""
-				continue
-			}
+}
+
+// 专营货期
+func BatchZhuanmaiDelivery(canals []string) {
+	if len(canals) == 0 { //专营货期
+		return
+	}
+	supplierRatios := gredis.Hmget("default_r", "supp_ratio", canals)
+	for dkey, supplierRatio := range supplierRatios {
+		cn_delivery := ""
+		hk_delivery := ""
+		if supplierRatio != "" {
 			cnDeliveryTime := gjson.Get(supplierRatio, "cn_delivery_time").String()
 			usDeliveryTime := gjson.Get(supplierRatio, "us_delivery_time").String()
 			//为了兼容供应商修改的问题
 			if cnDeliveryTime != "周" && cnDeliveryTime != "天" {
-				AllDelivery[dkey]["cn_delivery"] = gjson.Get(supplierRatio, "cn_delivery_time").String()
+				cn_delivery = gjson.Get(supplierRatio, "cn_delivery_time").String()
 			} else {
-				AllDelivery[dkey]["cn_delivery"] = ""
+				cn_delivery = ""
 			}
 			if usDeliveryTime != "周" && usDeliveryTime != "天" {
-				AllDelivery[dkey]["hk_delivery"] = gjson.Get(supplierRatio, "us_delivery_time").String()
+				hk_delivery = gjson.Get(supplierRatio, "us_delivery_time").String()
 			} else {
-				AllDelivery[dkey]["hk_delivery"] = ""
+				hk_delivery = ""
 			}
 		}
+		AllDelivery[dkey] = map[string]string{
+			"cn_delivery": cn_delivery,
+			"hk_delivery": hk_delivery,
+		}
 	}
-	if len(supplierIds) > 0 { //代购货期
-		supplierRatios := gredis.Hmget("default_r", "SUPPLIER_REDIS_INFO_", supplierIds)
-		for dkey, sinfo := range supplierRatios {
-			cnDelivery := gjson.Get(sinfo, "cn_delivery").String()
-			hkDelivery := gjson.Get(sinfo, "hk_delivery").String()
-			if cnDelivery != "" || hkDelivery != "" {
-				AllDelivery[dkey] = map[string]string{
-					"cn_delivery": cnDelivery,
-					"hk_delivery": hkDelivery,
-				}
-			} else {
-				AllDelivery[dkey] = map[string]string{
-					"cn_delivery": "",
-					"hk_delivery": "",
-				}
+}
+
+// 代购货期
+func BatchSeaDelivery(supplierIds []string) {
+	if len(supplierIds) == 0 {
+		return
+	}
+	supplierRatios := gredis.Hmget("default_r", "SUPPLIER_REDIS_INFO_", supplierIds)
+	for dkey, sinfo := range supplierRatios {
+		cnDelivery := gjson.Get(sinfo, "cn_delivery").String()
+		hkDelivery := gjson.Get(sinfo, "hk_delivery").String()
+		if cnDelivery != "" || hkDelivery != "" {
+			AllDelivery[dkey] = map[string]string{
+				"cn_delivery": cnDelivery,
+				"hk_delivery": hkDelivery,
+			}
+		} else {
+			AllDelivery[dkey] = map[string]string{
+				"cn_delivery": "",
+				"hk_delivery": "",
 			}
 		}
 	}
+}
 
-	//批量获取联营tags-标签
-	if len(lyGoodsId) > 0 {
-		AllLyGoodsTag = make(map[string]model.GoodsTag, 0)
-		goodsTagArr := gredis.Hmget("default_r", "goods_tag", lyGoodsId)
-		var tags []int
-		for goods_id, goodsTagsStr := range goodsTagArr {
-			if goodsTagsStr == "" {
-				continue
-			}
-			var tagNames []string
-			var goodsTags model.GoodsTag
-
-			skuStr := redisLySkuArr[goods_id]
-			source := gjson.Get(skuStr, "source").Int()
-			goods_label := gjson.Get(skuStr, "goods_label").String()
-			org_id := gjson.Get(skuStr, "org_id").Int()
-
-			goodsLabelType := int(gjson.Get(goodsTagsStr, "goods_label").Int())
-			goodsLabelMap := vars.GoodsLabel
-			goodsTags.GoodsLabelName = goodsLabelMap[goodsLabelType]
-			goodsTags.GoodsLabel = goodsLabelType
-
-			//自定义标签
-			customerTag := gjson.Get(goodsTagsStr, "customer_tag").String()
-			if customerTag != "" {
-				customerTagArr := php2go.Explode(" ", php2go.Trim(customerTag))
-				for _, cname := range customerTagArr {
-					//如果是爱智的话,去掉可议价标签
-					if org_id == 3 && cname == "可议价" {
-						continue
-					}
-					tagNames = append(tagNames, cname)
+// 批量获取联营tags
+func BatchLyTags(redisLySkuArr map[string]string, lyGoodsId []string) {
+	if len(lyGoodsId) == 0 {
+		return
+	}
+
+	AllLyGoodsTag = make(map[string]model.GoodsTag, 0)
+	goodsTagArr := gredis.Hmget("default_r", "goods_tag", lyGoodsId)
+	var tags []int
+	for goods_id, goodsTagsStr := range goodsTagArr {
+		if goodsTagsStr == "" {
+			continue
+		}
+		var tagNames []string
+		var goodsTags model.GoodsTag
+
+		skuStr := redisLySkuArr[goods_id]
+		source := gjson.Get(skuStr, "source").Int()
+		goods_label := gjson.Get(skuStr, "goods_label").String()
+		org_id := gjson.Get(skuStr, "org_id").Int()
+
+		goodsLabelType := int(gjson.Get(goodsTagsStr, "goods_label").Int())
+		goodsLabelMap := vars.GoodsLabel
+		goodsTags.GoodsLabelName = goodsLabelMap[goodsLabelType]
+		goodsTags.GoodsLabel = goodsLabelType
+
+		//自定义标签
+		customerTag := gjson.Get(goodsTagsStr, "customer_tag").String()
+		if customerTag != "" {
+			customerTagArr := php2go.Explode(" ", php2go.Trim(customerTag))
+			for _, cname := range customerTagArr {
+				//如果是爱智的话,去掉可议价标签
+				if org_id == 3 && cname == "可议价" {
+					continue
 				}
+				tagNames = append(tagNames, cname)
 			}
+		}
 
-			if source == 12 {
-				if !php2go.InArray("当天发货", tagNames) {
-					tagNames = append(tagNames, "当天发货")
-				}
+		if source == 12 {
+			if !php2go.InArray("当天发货", tagNames) {
+				tagNames = append(tagNames, "当天发货")
 			}
+		}
 
-			//goods_source_tags
-			sourceTagsStr := GoodsSourceTags[gconv.String(source)]
-			if sourceTagsStr != "" {
-				systemTags := gjson.Get(sourceTagsStr, "system_tags").Array()
-				for _, tag := range systemTags {
-					tagNames = append(tagNames, tag.String())
-				}
+		//goods_source_tags
+		sourceTagsStr := GoodsSourceTags[gconv.String(source)]
+		if sourceTagsStr != "" {
+			systemTags := gjson.Get(sourceTagsStr, "system_tags").Array()
+			for _, tag := range systemTags {
+				tagNames = append(tagNames, tag.String())
 			}
+		}
 
-			//goods_label_tags
-			labelTagsStr := GoodsLabelTags[goods_label]
-			if labelTagsStr != "" {
-				labelTags := gjson.Get(labelTagsStr, "system_tags").Array()
-				for _, tag := range labelTags {
-					tagNames = append(tagNames, tag.String())
-				}
+		//goods_label_tags
+		labelTagsStr := GoodsLabelTags[goods_label]
+		if labelTagsStr != "" {
+			labelTags := gjson.Get(labelTagsStr, "system_tags").Array()
+			for _, tag := range labelTags {
+				tagNames = append(tagNames, tag.String())
 			}
-			//去重tagNames
-			tagNames = removeDuplicateString(tagNames)
-			goodsTags.GoodsTagNames = tagNames
-			goodsTags.GoodsTag = tags
-			AllLyGoodsTag[goods_id] = goodsTags
 		}
+		//去重tagNames
+		tagNames = removeDuplicateString(tagNames)
+		goodsTags.GoodsTagNames = tagNames
+		goodsTags.GoodsTag = tags
+		AllLyGoodsTag[goods_id] = goodsTags
 	}
-
 }
 
 // 获取指定的币种汇率