muyu-apiserver/gateway/internal/logic/inventory/productbatch/updateProductBatchLogic.go

package productbatch

import (
	"context"

	"muyu-apiserver/gateway/internal/svc"
	"muyu-apiserver/gateway/internal/types"
	"muyu-apiserver/pkg/ctxdata"
	"muyu-apiserver/rpc/inventory/pb"

	"github.com/zeromicro/go-zero/core/logx"
)

type UpdateProductBatchLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewUpdateProductBatchLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UpdateProductBatchLogic {
	return &UpdateProductBatchLogic{Logger: logx.WithContext(ctx), ctx: ctx, svcCtx: svcCtx}
}

func (l *UpdateProductBatchLogic) UpdateProductBatch(req *types.UpdateProductBatchReq) (*types.IdResp, error) {
	productId := ctxdata.GetPathId(l.ctx)
	batchId, _ := l.ctx.Value("batchId").(string)
	_, err := l.svcCtx.InventoryRpc.UpdateProductBatch(l.ctx, &pb.UpdateProductBatchReq{
		ProductId: productId,
		BatchId:   batchId,
		Batch:     batchInputToPB(req.Batch),
		Status:    req.Status,
	})
	if err != nil {
		return nil, err
	}
	return &types.IdResp{Id: batchId}, nil
}