Skip to content

Commit

Permalink
Code clean up
Browse files Browse the repository at this point in the history
  • Loading branch information
VeronikaSolovei9 committed Oct 5, 2023
1 parent 358c3b4 commit d841a8a
Show file tree
Hide file tree
Showing 9 changed files with 31 additions and 31 deletions.
8 changes: 4 additions & 4 deletions exchange/bidder.go
Original file line number Diff line number Diff line change
Expand Up @@ -130,8 +130,8 @@ type bidderAdapterConfig struct {
}

func (bidder *bidderAdapter) requestBid(ctx context.Context, bidderRequest BidderRequest, conversions currency.Conversions, reqInfo *adapters.ExtraRequestInfo, adsCertSigner adscert.Signer, bidRequestOptions bidRequestOptions, alternateBidderCodes openrtb_ext.ExtAlternateBidderCodes, hookExecutor hookexecution.StageExecutor, ruleToAdjustments openrtb_ext.AdjustmentsByDealID) ([]*entities.PbsOrtbSeatBid, extraBidderRespInfo, []error) {
brw := openrtb_ext.RequestWrapper{BidRequest: bidderRequest.BidRequest}
reject := hookExecutor.ExecuteBidderRequestStage(&brw, string(bidderRequest.BidderName))
request := openrtb_ext.RequestWrapper{BidRequest: bidderRequest.BidRequest}
reject := hookExecutor.ExecuteBidderRequestStage(&request, string(bidderRequest.BidderName))
if reject != nil {
return nil, extraBidderRespInfo{}, []error{reject}
}
Expand All @@ -144,8 +144,8 @@ func (bidder *bidderAdapter) requestBid(ctx context.Context, bidderRequest Bidde
)

// rebuild request after modules execution
brw.RebuildRequest()
bidderRequest.BidRequest = brw.BidRequest
request.RebuildRequest()
bidderRequest.BidRequest = request.BidRequest

//check if real request exists for this bidder or it only has stored responses
dataLen := 0
Expand Down
4 changes: 2 additions & 2 deletions exchange/exchange_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -5622,12 +5622,12 @@ func (e mockUpdateBidRequestHook) HandleBidderRequestHook(_ context.Context, mct
c := hookstage.ChangeSet[hookstage.BidderRequestPayload]{}
c.AddMutation(
func(payload hookstage.BidderRequestPayload) (hookstage.BidderRequestPayload, error) {
payload.BidRequest.Site.Name = "test"
payload.Request.Site.Name = "test"
return payload, nil
}, hookstage.MutationUpdate, "bidRequest", "site.name",
).AddMutation(
func(payload hookstage.BidderRequestPayload) (hookstage.BidderRequestPayload, error) {
payload.BidRequest.Site.Domain = "test.com"
payload.Request.Site.Domain = "test.com"
return payload, nil
}, hookstage.MutationUpdate, "bidRequest", "site.domain",
)
Expand Down
4 changes: 2 additions & 2 deletions hooks/hookexecution/executor.go
Original file line number Diff line number Diff line change
Expand Up @@ -160,7 +160,7 @@ func (e *hookExecutor) ExecuteProcessedAuctionStage(request *openrtb_ext.Request

stageName := hooks.StageProcessedAuctionRequest.String()
executionCtx := e.newContext(stageName)
payload := hookstage.ProcessedAuctionRequestPayload{BidRequest: request}
payload := hookstage.ProcessedAuctionRequestPayload{Request: request}

outcome, _, contexts, reject := executeStage(executionCtx, plan, payload, handler, e.metricEngine)
outcome.Entity = entityAuctionRequest
Expand Down Expand Up @@ -194,7 +194,7 @@ func (e *hookExecutor) ExecuteBidderRequestStage(req *openrtb_ext.RequestWrapper

stageName := hooks.StageBidderRequest.String()
executionCtx := e.newContext(stageName)
payload := hookstage.BidderRequestPayload{BidRequest: req, Bidder: bidder}
payload := hookstage.BidderRequestPayload{Request: req, Bidder: bidder}
outcome, payload, contexts, reject := executeStage(executionCtx, plan, payload, handler, e.metricEngine)
outcome.Entity = entity(bidder)
outcome.Stage = stageName
Expand Down
14 changes: 7 additions & 7 deletions hooks/hookexecution/mocks_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -131,7 +131,7 @@ func (e mockTimeoutHook) HandleProcessedAuctionHook(_ context.Context, _ hooksta
time.Sleep(20 * time.Millisecond)
c := hookstage.ChangeSet[hookstage.ProcessedAuctionRequestPayload]{}
c.AddMutation(func(payload hookstage.ProcessedAuctionRequestPayload) (hookstage.ProcessedAuctionRequestPayload, error) {
payload.BidRequest.User.CustomData = "some-custom-data"
payload.Request.User.CustomData = "some-custom-data"
return payload, nil
}, hookstage.MutationUpdate, "bidRequest", "user.customData")

Expand All @@ -142,7 +142,7 @@ func (e mockTimeoutHook) HandleBidderRequestHook(_ context.Context, _ hookstage.
time.Sleep(20 * time.Millisecond)
c := hookstage.ChangeSet[hookstage.BidderRequestPayload]{}
c.AddMutation(func(payload hookstage.BidderRequestPayload) (hookstage.BidderRequestPayload, error) {
payload.BidRequest.User.CustomData = "some-custom-data"
payload.Request.User.CustomData = "some-custom-data"
return payload, nil
}, hookstage.MutationUpdate, "bidRequest", "user.customData")

Expand Down Expand Up @@ -305,8 +305,8 @@ func (e mockUpdateBidRequestHook) HandleProcessedAuctionHook(_ context.Context,
c := hookstage.ChangeSet[hookstage.ProcessedAuctionRequestPayload]{}
c.AddMutation(
func(payload hookstage.ProcessedAuctionRequestPayload) (hookstage.ProcessedAuctionRequestPayload, error) {
payload.BidRequest.User.Yob = 2000
userExt, err := payload.BidRequest.GetUserExt()
payload.Request.User.Yob = 2000
userExt, err := payload.Request.GetUserExt()
if err != nil {
return payload, err
}
Expand All @@ -318,7 +318,7 @@ func (e mockUpdateBidRequestHook) HandleProcessedAuctionHook(_ context.Context,
}, hookstage.MutationUpdate, "bidRequest", "user.yob",
).AddMutation(
func(payload hookstage.ProcessedAuctionRequestPayload) (hookstage.ProcessedAuctionRequestPayload, error) {
payload.BidRequest.User.Consent = "true"
payload.Request.User.Consent = "true"
return payload, nil
}, hookstage.MutationUpdate, "bidRequest", "user.consent",
)
Expand All @@ -330,12 +330,12 @@ func (e mockUpdateBidRequestHook) HandleBidderRequestHook(_ context.Context, _ h
c := hookstage.ChangeSet[hookstage.BidderRequestPayload]{}
c.AddMutation(
func(payload hookstage.BidderRequestPayload) (hookstage.BidderRequestPayload, error) {
payload.BidRequest.User.Yob = 2000
payload.Request.User.Yob = 2000
return payload, nil
}, hookstage.MutationUpdate, "bidRequest", "user.yob",
).AddMutation(
func(payload hookstage.BidderRequestPayload) (hookstage.BidderRequestPayload, error) {
payload.BidRequest.User.Consent = "true"
payload.Request.User.Consent = "true"
return payload, nil
}, hookstage.MutationUpdate, "bidRequest", "user.consent",
)
Expand Down
4 changes: 2 additions & 2 deletions hooks/hookstage/bidderrequest.go
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,6 @@ type BidderRequest interface {
// distilled for the particular bidder.
// Hooks are allowed to modify openrtb2.BidRequest using mutations.
type BidderRequestPayload struct {
BidRequest *openrtb_ext.RequestWrapper
Bidder string
Request *openrtb_ext.RequestWrapper
Bidder string
}
4 changes: 2 additions & 2 deletions hooks/hookstage/bidderrequest_mutations.go
Original file line number Diff line number Diff line change
Expand Up @@ -33,10 +33,10 @@ func (c ChangeSetBidderRequest[T]) BApp() ChangeSetBApp[T] {

func (c ChangeSetBidderRequest[T]) castPayload(p T) (*openrtb_ext.RequestWrapper, error) {
if payload, ok := any(p).(BidderRequestPayload); ok {
if payload.BidRequest == nil || payload.BidRequest.BidRequest == nil {
if payload.Request == nil || payload.Request.BidRequest == nil {
return nil, errors.New("payload contains a nil bid request")
}
return payload.BidRequest, nil
return payload.Request, nil
}
return nil, errors.New("failed to cast BidderRequestPayload")
}
Expand Down
2 changes: 1 addition & 1 deletion hooks/hookstage/processedauctionrequest.go
Original file line number Diff line number Diff line change
Expand Up @@ -26,5 +26,5 @@ type ProcessedAuctionRequest interface {
// ProcessedAuctionRequestPayload consists of the openrtb_ext.RequestWrapper object.
// Hooks are allowed to modify openrtb_ext.RequestWrapper using mutations.
type ProcessedAuctionRequestPayload struct {
BidRequest *openrtb_ext.RequestWrapper
Request *openrtb_ext.RequestWrapper
}
18 changes: 9 additions & 9 deletions modules/prebid/ortb2blocking/hook_bidderrequest.go
Original file line number Diff line number Diff line change
Expand Up @@ -16,11 +16,11 @@ func handleBidderRequestHook(
cfg config,
payload hookstage.BidderRequestPayload,
) (result hookstage.HookResult[hookstage.BidderRequestPayload], err error) {
if payload.BidRequest == nil || payload.BidRequest.BidRequest == nil {
if payload.Request == nil || payload.Request.BidRequest == nil {
return result, hookexecution.NewFailure("payload contains a nil bid request")
}

mediaTypes := mediaTypesFrom(payload.BidRequest.BidRequest)
mediaTypes := mediaTypesFrom(payload.Request.BidRequest)
changeSet := hookstage.ChangeSet[hookstage.BidderRequestPayload]{}
blockingAttributes := blockingAttributes{}

Expand Down Expand Up @@ -60,7 +60,7 @@ func updateBAdv(
result *hookstage.HookResult[hookstage.BidderRequestPayload],
changeSet *hookstage.ChangeSet[hookstage.BidderRequestPayload],
) (err error) {
if len(payload.BidRequest.BAdv) > 0 {
if len(payload.Request.BAdv) > 0 {
return nil
}

Expand All @@ -87,7 +87,7 @@ func updateBApp(
result *hookstage.HookResult[hookstage.BidderRequestPayload],
changeSet *hookstage.ChangeSet[hookstage.BidderRequestPayload],
) (err error) {
if len(payload.BidRequest.BApp) > 0 {
if len(payload.Request.BApp) > 0 {
return nil
}

Expand All @@ -114,7 +114,7 @@ func updateBCat(
result *hookstage.HookResult[hookstage.BidderRequestPayload],
changeSet *hookstage.ChangeSet[hookstage.BidderRequestPayload],
) (err error) {
if len(payload.BidRequest.BCat) > 0 {
if len(payload.Request.BCat) > 0 {
return nil
}

Expand Down Expand Up @@ -191,7 +191,7 @@ func updateCatTax(
attributes *blockingAttributes,
changeSet *hookstage.ChangeSet[hookstage.BidderRequestPayload],
) {
if payload.BidRequest.CatTax > 0 {
if payload.Request.CatTax > 0 {
return
}

Expand Down Expand Up @@ -226,7 +226,7 @@ func mutationForImp(
impUpdater impUpdateFunc,
) hookstage.MutationFunc[hookstage.BidderRequestPayload] {
return func(payload hookstage.BidderRequestPayload) (hookstage.BidderRequestPayload, error) {
for i, imp := range payload.BidRequest.Imp {
for i, imp := range payload.Request.Imp {
if values, ok := valuesByImp[imp.ID]; ok {
if len(values) == 0 {
continue
Expand All @@ -236,7 +236,7 @@ func mutationForImp(
imp.Banner = &openrtb2.Banner{}
}

payload.BidRequest.Imp[i] = impUpdater(imp, values)
payload.Request.Imp[i] = impUpdater(imp, values)
}
}
return payload, nil
Expand Down Expand Up @@ -310,7 +310,7 @@ func findImpressionOverrides(
overrides := map[string][]int{}
messages := []string{}

for _, imp := range payload.BidRequest.Imp {
for _, imp := range payload.Request.Imp {
// do not add override for attribute if it already exists in request
if isAttrPresent(imp) {
continue
Expand Down
4 changes: 2 additions & 2 deletions modules/prebid/ortb2blocking/module_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -568,7 +568,7 @@ func TestHandleBidderRequestHook(t *testing.T) {
for _, test := range testCases {
t.Run(test.description, func(t *testing.T) {
brw := openrtb_ext.RequestWrapper{BidRequest: test.bidRequest}
payload := hookstage.BidderRequestPayload{Bidder: test.bidder, BidRequest: &brw}
payload := hookstage.BidderRequestPayload{Bidder: test.bidder, Request: &brw}

result, err := Builder(nil, moduledeps.ModuleDeps{})
assert.NoError(t, err, "Failed to build module.")
Expand All @@ -593,7 +593,7 @@ func TestHandleBidderRequestHook(t *testing.T) {
assert.NoError(t, err)
}

assert.Equal(t, test.expectedBidRequest, payload.BidRequest.BidRequest, "Invalid BidRequest after executing BidderRequestHook.")
assert.Equal(t, test.expectedBidRequest, payload.Request.BidRequest, "Invalid BidRequest after executing BidderRequestHook.")

// reset ChangeSet not to break hookResult assertion, we validated ChangeSet separately
hookResult.ChangeSet = hookstage.ChangeSet[hookstage.BidderRequestPayload]{}
Expand Down

0 comments on commit d841a8a

Please sign in to comment.