From 9ddea52e0d88255d21686285e00b4e9014be6c57 Mon Sep 17 00:00:00 2001
From: kanade <kanade@kumo.work>
Date: Fri, 19 Aug 2022 16:09:28 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E5=BA=93=E5=AD=98=E6=8E=A5?=
 =?UTF-8?q?=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/v2/controller/app/jcook/jcook.go | 20 +++++++-------------
 internal/v2/lib/supply/sku.go             | 19 +++++++++++++++++++
 2 files changed, 26 insertions(+), 13 deletions(-)

diff --git a/internal/v2/controller/app/jcook/jcook.go b/internal/v2/controller/app/jcook/jcook.go
index 3a58e79..8985315 100644
--- a/internal/v2/controller/app/jcook/jcook.go
+++ b/internal/v2/controller/app/jcook/jcook.go
@@ -5,6 +5,7 @@ import (
 	"recook/internal/back"
 	jcook2 "recook/internal/v2/lib/jcook"
 	"recook/internal/v2/lib/shama"
+	"recook/internal/v2/lib/supply"
 	"recook/internal/v2/logic/app/jcook"
 	"recook/internal/v2/model/recook/goods"
 	"recook/tools"
@@ -93,18 +94,11 @@ func (a *Proxy) Stock(c *gin.Context) {
 	}
 	if gs.ThirdPartyType == 5 {
 		id, _ := strconv.Atoi(gs.ThirdPartySkuId)
-		req := shama.SkuStockReq{
-			Address: p.Address,
-			SkuList: []shama.SkuQuantity{
-				{
-					SkuID:    uint(id),
-					Quantity: p.Quantity,
-				},
-			},
-		}
-		client := shama.GetClient()
-		var resp []shama.SkuStockResp
-		if err = client.Exec(req, &resp); err != nil {
+		stock, err := supply.Api.Sku.Stock(p.Address, []supply.SkuStockItem{{
+			SkuId:    uint(id),
+			Quantity: p.Quantity,
+		}})
+		if err != nil {
 			log.Println(err)
 			back.Suc(c, "", gin.H{
 				"stock_state": 0,
@@ -112,7 +106,7 @@ func (a *Proxy) Stock(c *gin.Context) {
 			return
 		}
 		back.Suc(c, "", gin.H{
-			"stock_state": resp[0].StockState,
+			"stock_state": stock[0].State,
 		})
 	}
 }
diff --git a/internal/v2/lib/supply/sku.go b/internal/v2/lib/supply/sku.go
index b2ab2c2..4d559a9 100644
--- a/internal/v2/lib/supply/sku.go
+++ b/internal/v2/lib/supply/sku.go
@@ -5,6 +5,7 @@ const (
 	skuDetails = "/sku/details" // 商品详情
 	skuPrices  = "/sku/prices"  // 商品价格
 	skuGroups  = "/sku/groups"  // 商品分组信息
+	skuStock   = "/sku/stock"   // 商品库存
 )
 
 type sku struct {
@@ -137,3 +138,21 @@ func (s *sku) Groups(skuIds []uint) (result []SkuGroup, err error) {
 	}, &result)
 	return
 }
+
+type SkuStockItem struct {
+	SkuId    uint `json:"skuId"`
+	Quantity uint `json:"quantity"`
+}
+type SkuStock struct {
+	SkuId uint `json:"skuId"` // skuId
+	State uint `json:"state"` // 库存状态
+}
+
+// Stock @Title 获取商品库存
+func (s *sku) Stock(address string, skus []SkuStockItem) (result []SkuStock, err error) {
+	err = exec(skuStock, map[string]interface{}{
+		"address": address,
+		"skus":    skus,
+	}, &result)
+	return
+}