From d3d514dad2f10e5f6b48a9886b5dd871de1f8ff6 Mon Sep 17 00:00:00 2001 From: kanade Date: Wed, 27 Jul 2022 15:40:22 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=B8=A0=E9=81=93=E5=8F=82?= =?UTF-8?q?=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- supply/channel/order.go | 9 +++++---- supply/channel/sku.go | 13 +++++++------ 2 files changed, 12 insertions(+), 10 deletions(-) diff --git a/supply/channel/order.go b/supply/channel/order.go index 871f15b..2450dc5 100644 --- a/supply/channel/order.go +++ b/supply/channel/order.go @@ -4,6 +4,7 @@ import ( "context" "git.oa00.com/supply-chain/service/client" "github.com/shopspring/decimal" + "github.com/smallnest/rpcx/share" ) type order struct { @@ -25,8 +26,8 @@ type ReplyOrderFreightFee struct { } // FreightFee @Title 获取运费 -func (o *order) FreightFee(ctx context.Context, args ArgsOrderFreightFee) (reply []ReplyOrderFreightFee, err error) { - err = client.GetClient(o).Call(ctx, "FreightFee", args, &reply) +func (o *order) FreightFee(ctx context.Context, channelId string, args ArgsOrderFreightFee) (reply []ReplyOrderFreightFee, err error) { + err = client.GetClient(o).Call(context.WithValue(ctx, share.ReqMetaDataKey, map[string]string{"channelId": channelId}), "FreightFee", args, &reply) return } @@ -50,7 +51,7 @@ type ReplyOrderSubmit struct { } // Submit @Title 下单 -func (o *order) Submit(ctx context.Context, args ArgsOrderSubmit) (reply []ReplyOrderSubmit, err error) { - err = client.GetClient(o).Call(ctx, "Submit", args, &reply) +func (o *order) Submit(ctx context.Context, channelId string, args ArgsOrderSubmit) (reply []ReplyOrderSubmit, err error) { + err = client.GetClient(o).Call(context.WithValue(ctx, share.ReqMetaDataKey, map[string]string{"channelId": channelId}), "Submit", args, &reply) return } diff --git a/supply/channel/sku.go b/supply/channel/sku.go index 63f4d1f..a4af4e6 100644 --- a/supply/channel/sku.go +++ b/supply/channel/sku.go @@ -5,6 +5,7 @@ import ( "git.oa00.com/supply-chain/service/client" "git.oa00.com/supply-chain/service/lib/bean" "github.com/shopspring/decimal" + "github.com/smallnest/rpcx/share" ) type sku struct { @@ -49,8 +50,8 @@ type ReplySkuList struct { } // Lists @Title 商品列表 -func (s *sku) Lists(ctx context.Context, args ArgsSkuList) (reply ReplySkuList, err error) { - err = client.GetClient(s).Call(ctx, "Lists", args, &reply) +func (s *sku) Lists(ctx context.Context, channelId string, args ArgsSkuList) (reply ReplySkuList, err error) { + err = client.GetClient(s).Call(context.WithValue(ctx, share.ReqMetaDataKey, map[string]string{"channelId": channelId}), "Lists", args, &reply) return } @@ -101,8 +102,8 @@ type SkuAttribute struct { } // Details @Title 获取sku详情 -func (s *sku) Details(ctx context.Context, args ArgsSkuDetails) (reply []SkuDetailItem, err error) { - err = client.GetClient(s).Call(ctx, "Details", args, &reply) +func (s *sku) Details(ctx context.Context, channelId string, args ArgsSkuDetails) (reply []SkuDetailItem, err error) { + err = client.GetClient(s).Call(context.WithValue(ctx, share.ReqMetaDataKey, map[string]string{"channelId": channelId}), "Details", args, &reply) return } @@ -122,7 +123,7 @@ type ReplySkuStock struct { } // Stock @Title 库存查询 -func (s *sku) Stock(ctx context.Context, args ArgsSkuStock) (reply []ReplySkuStock, err error) { - err = client.GetClient(s).Call(ctx, "Stock", args, &reply) +func (s *sku) Stock(ctx context.Context, channelId string, args ArgsSkuStock) (reply []ReplySkuStock, err error) { + err = client.GetClient(s).Call(context.WithValue(ctx, share.ReqMetaDataKey, map[string]string{"channelId": channelId}), "Stock", args, &reply) return }