From 82ad1ed0cc18fcf2b7f1be0b153497db0dcf5789 Mon Sep 17 00:00:00 2001
From: gazebo <gazeboxu@gmail.com>
Date: Mon, 15 Jul 2019 12:54:56 +0800
Subject: [PATCH] =?UTF-8?q?-=20=E5=AF=B9=E4=BA=8E=E4=BA=AC=E4=B8=9C?=
 =?UTF-8?q?=E5=88=B0=E5=AE=B6=E6=93=8D=E4=BD=9C=E5=8D=95=E4=B8=AA=E9=97=A8?=
 =?UTF-8?q?=E5=BA=97=E5=95=86=E5=93=81=E6=97=B6=E4=BC=98=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 business/partner/purchase/jd/store_sku2.go | 48 ++++++++++++++--------
 1 file changed, 30 insertions(+), 18 deletions(-)

diff --git a/business/partner/purchase/jd/store_sku2.go b/business/partner/purchase/jd/store_sku2.go
index af5a8acd3..5fda21747 100644
--- a/business/partner/purchase/jd/store_sku2.go
+++ b/business/partner/purchase/jd/store_sku2.go
@@ -92,29 +92,41 @@ func (p *PurchaseHandler) UpdateStoreSkusStatus(ctx *jxcontext.Context, storeID
 }
 
 func (p *PurchaseHandler) UpdateStoreSkusPrice(ctx *jxcontext.Context, storeID int, vendorStoreID string, storeSkuList []*partner.StoreSkuInfo) (err error) {
-	var skuPriceInfoList []*jdapi.SkuPriceInfo
-	for _, v := range storeSkuList {
-		skuPriceInfoList = append(skuPriceInfoList, &jdapi.SkuPriceInfo{
-			OutSkuId: utils.Int2Str(v.SkuID),
-			Price:    int(v.Price),
-		})
-	}
-	if globals.EnableJdStoreWrite {
-		_, err = api.JdAPI.UpdateVendorStationPrice("", vendorStoreID, skuPriceInfoList)
+	if len(storeSkuList) == 1 {
+		if globals.EnableJdStoreWrite {
+			_, err = api.JdAPI.UpdateStationPrice(utils.Str2Int64WithDefault(storeSkuList[0].VendorSkuID, 0), vendorStoreID, int(storeSkuList[0].Price))
+		}
+	} else {
+		var skuPriceInfoList []*jdapi.SkuPriceInfo
+		for _, v := range storeSkuList {
+			skuPriceInfoList = append(skuPriceInfoList, &jdapi.SkuPriceInfo{
+				OutSkuId: utils.Int2Str(v.SkuID),
+				Price:    int(v.Price),
+			})
+		}
+		if globals.EnableJdStoreWrite {
+			_, err = api.JdAPI.UpdateVendorStationPrice("", vendorStoreID, skuPriceInfoList)
+		}
 	}
 	return err
 }
 
 func (p *PurchaseHandler) UpdateStoreSkusStock(ctx *jxcontext.Context, storeID int, vendorStoreID string, storeSkuList []*partner.StoreSkuInfo) (err error) {
-	var skuStockList []*jdapi.SkuStock
-	for _, v := range storeSkuList {
-		skuStockList = append(skuStockList, &jdapi.SkuStock{
-			OutSkuId: utils.Int2Str(v.SkuID),
-			StockQty: v.Stock,
-		})
-	}
-	if globals.EnableJdStoreWrite {
-		_, err = api.JdAPI.BatchUpdateCurrentQtys("", vendorStoreID, skuStockList, ctx.GetUserName())
+	if len(storeSkuList) == 1 {
+		if globals.EnableJdStoreWrite {
+			err = api.JdAPI.UpdateCurrentQty(vendorStoreID, utils.Str2Int64WithDefault(storeSkuList[0].VendorSkuID, 0), storeSkuList[0].Stock)
+		}
+	} else {
+		var skuStockList []*jdapi.SkuStock
+		for _, v := range storeSkuList {
+			skuStockList = append(skuStockList, &jdapi.SkuStock{
+				OutSkuId: utils.Int2Str(v.SkuID),
+				StockQty: v.Stock,
+			})
+		}
+		if globals.EnableJdStoreWrite {
+			_, err = api.JdAPI.BatchUpdateCurrentQtys("", vendorStoreID, skuStockList, ctx.GetUserName())
+		}
 	}
 	return err
 }