feat: update protobuf message structure for product tariffs; change Offers to Offer and adjust related conversions

This commit is contained in:
2025-08-17 10:47:52 +03:00
parent abbcc0a81a
commit 593fbc88fa
7 changed files with 76 additions and 66 deletions

View File

@@ -6,7 +6,7 @@ import (
type PbCalculateProductTariffsRequest = pb.CalculateProductTariffsRequest
type PbCalculateProductTariffsResponse = pb.CalculateProductTariffsResponse
type PbOffer = pb.CalculateProductTariffsRequest_Offers
type PbOffer = pb.CalculateProductTariffsRequest_Offer
type PbResponseOffer = pb.CalculateProductTariffsResponse_Offers
type PbParameters = pb.CalculateProductTariffsRequest_Parameters
type PbGetProductsOffer = pb.GetProductsResponse_Offer

View File

@@ -1,8 +1,9 @@
package mapping
import (
"git.denco.store/fakz9/yandex-go-client"
proto "sipro-mps/api/generated/v1/yandexmarket/products"
"git.denco.store/fakz9/yandex-go-client"
)
//go:generate go run github.com/jmattheis/goverter/cmd/goverter gen .
@@ -16,7 +17,7 @@ import (
// goverter:useZeroValueOnPointerInconsistency yes
// goverter:extend Int64ToFloat32 Int64ToInt32 Float32ToInt64 Int32ToInt64 PointerInt32ToInt64
type Converter interface {
ProtoOfferToYmOffer(details *proto.CalculateProductTariffsRequest_Offers) *ymclient.CalculateTariffsOfferDTO
ProtoOfferToYmOffer(details *proto.CalculateProductTariffsRequest_Offer) *ymclient.CalculateTariffsOfferDTO
ProtoParametersToYmParameters(details *proto.CalculateProductTariffsRequest_Parameters) *ymclient.CalculateTariffsParametersDTO
// Response converters

View File

@@ -11,16 +11,16 @@ import (
type ConverterImpl struct{}
func (c *ConverterImpl) ProtoOfferToYmOffer(source *products.CalculateProductTariffsRequest_Offers) *yandexgoclient.CalculateTariffsOfferDTO {
func (c *ConverterImpl) ProtoOfferToYmOffer(source *products.CalculateProductTariffsRequest_Offer) *yandexgoclient.CalculateTariffsOfferDTO {
var pYmclientCalculateTariffsOfferDTO *yandexgoclient.CalculateTariffsOfferDTO
if source != nil {
var ymclientCalculateTariffsOfferDTO yandexgoclient.CalculateTariffsOfferDTO
ymclientCalculateTariffsOfferDTO.CategoryId = (*source).CategoryId
ymclientCalculateTariffsOfferDTO.Price = mapping.Int64ToFloat32((*source).Price)
ymclientCalculateTariffsOfferDTO.Length = mapping.Int64ToFloat32((*source).Length)
ymclientCalculateTariffsOfferDTO.Width = mapping.Int64ToFloat32((*source).Width)
ymclientCalculateTariffsOfferDTO.Height = mapping.Int64ToFloat32((*source).Height)
ymclientCalculateTariffsOfferDTO.Weight = mapping.Int64ToFloat32((*source).Weight)
ymclientCalculateTariffsOfferDTO.Price = (*source).Price
ymclientCalculateTariffsOfferDTO.Length = (*source).Length
ymclientCalculateTariffsOfferDTO.Width = (*source).Width
ymclientCalculateTariffsOfferDTO.Height = (*source).Height
ymclientCalculateTariffsOfferDTO.Weight = (*source).Weight
pInt32 := mapping.Int64ToInt32((*source).Quantity)
ymclientCalculateTariffsOfferDTO.Quantity = &pInt32
pYmclientCalculateTariffsOfferDTO = &ymclientCalculateTariffsOfferDTO
@@ -117,11 +117,11 @@ func (c *ConverterImpl) pYmclientOfferWeightDimensionsDTOToPProductsGetProductsR
func (c *ConverterImpl) ymclientCalculateTariffsOfferDTOToPProductsCalculateProductTariffsResponse_Offer(source yandexgoclient.CalculateTariffsOfferDTO) *products.CalculateProductTariffsResponse_Offer {
var productsCalculateProductTariffsResponse_Offer products.CalculateProductTariffsResponse_Offer
productsCalculateProductTariffsResponse_Offer.CategoryId = source.CategoryId
productsCalculateProductTariffsResponse_Offer.Price = mapping.Float32ToInt64(source.Price)
productsCalculateProductTariffsResponse_Offer.Length = mapping.Float32ToInt64(source.Length)
productsCalculateProductTariffsResponse_Offer.Width = mapping.Float32ToInt64(source.Width)
productsCalculateProductTariffsResponse_Offer.Height = mapping.Float32ToInt64(source.Height)
productsCalculateProductTariffsResponse_Offer.Weight = mapping.Float32ToInt64(source.Weight)
productsCalculateProductTariffsResponse_Offer.Price = source.Price
productsCalculateProductTariffsResponse_Offer.Length = source.Length
productsCalculateProductTariffsResponse_Offer.Width = source.Width
productsCalculateProductTariffsResponse_Offer.Height = source.Height
productsCalculateProductTariffsResponse_Offer.Weight = source.Weight
if source.Quantity != nil {
productsCalculateProductTariffsResponse_Offer.Quantity = mapping.Int32ToInt64(*source.Quantity)
}

View File

@@ -184,6 +184,7 @@ func (r *apiRepository) processOfferMappings(offerMappings []ymclient.GetOfferMa
resultOffers = append(resultOffers, &pb.GetProductsResponse_Offer{})
continue
}
protoOffer.MarketCategoryId = *offerMapping.Mapping.MarketCategoryId
resultOffers = append(resultOffers, protoOffer)
}
@@ -233,15 +234,22 @@ func (r *apiRepository) setTariffsRateLimit() {
}
// processTariffChunk processes a single chunk of offers for tariff calculation
func (r *apiRepository) processTariffChunk(ctx context.Context, client *ymclient.APIClient, ymParameters *ymclient.CalculateTariffsParametersDTO, offerChunk []*pb.CalculateProductTariffsRequest_Offers, chunkIndex int) (*pb.CalculateProductTariffsResponse, error) {
func (r *apiRepository) processTariffChunk(ctx context.Context, client *ymclient.APIClient, ymParameters *ymclient.CalculateTariffsParametersDTO, offerChunk []*pb.CalculateProductTariffsRequest_Offer, chunkIndex int) (*pb.CalculateProductTariffsResponse, error) {
ymOffers := r.convertOffersToYM(offerChunk)
if len(ymOffers) == 0 {
fmt.Printf("Skipping chunk %d: no valid offers\n", chunkIndex+1)
return nil, nil
}
ymRequest := ymclient.NewCalculateTariffsRequest(*ymParameters, ymOffers)
if ymParameters.CampaignId != nil && *ymParameters.CampaignId > 0 {
ymParameters.SellingProgram = nil
ymParameters.Frequency = nil
ymParameters.Currency = nil
} else {
ymParameters.CampaignId = nil
}
ymRequest := ymclient.NewCalculateTariffsRequest(*ymParameters, ymOffers)
response, httpResp, err := client.TariffsAPI.CalculateTariffs(ctx).
CalculateTariffsRequest(*ymRequest).
Execute()
@@ -263,7 +271,7 @@ func (r *apiRepository) processTariffChunk(ctx context.Context, client *ymclient
}
// convertOffersToYM converts protobuf offers to Yandex Market format
func (r *apiRepository) convertOffersToYM(offers []*pb.CalculateProductTariffsRequest_Offers) []ymclient.CalculateTariffsOfferDTO {
func (r *apiRepository) convertOffersToYM(offers []*pb.CalculateProductTariffsRequest_Offer) []ymclient.CalculateTariffsOfferDTO {
var ymOffers []ymclient.CalculateTariffsOfferDTO
for _, offer := range offers {
ymOffer := r.converter.ProtoOfferToYmOffer(offer)