Added support for rejection tracking.

Author: And1sS

extra/modules/pb-rule-engine/src/main/java/org/prebid/server/hooks/modules/rule/engine/v1/PbRuleEngineProcessedAuctionRequestHook.java

@@ -5,6 +5,8 @@
 import io.vertx.core.Future;
 import org.apache.commons.lang3.StringUtils;
 import org.prebid.server.auction.model.AuctionContext;
+import org.prebid.server.auction.model.ImpRejection;
+import org.prebid.server.auction.model.Rejection;
 import org.prebid.server.hooks.execution.v1.InvocationResultImpl;
 import org.prebid.server.hooks.execution.v1.auction.AuctionRequestPayloadImpl;
 import org.prebid.server.hooks.modules.rule.engine.core.config.RuleParser;
@@ -19,8 +21,12 @@
 import org.prebid.server.hooks.v1.auction.AuctionInvocationContext;
 import org.prebid.server.hooks.v1.auction.AuctionRequestPayload;
 import org.prebid.server.hooks.v1.auction.ProcessedAuctionRequestHook;
+import org.prebid.server.proto.openrtb.ext.response.seatnonbid.SeatNonBid;
 
+import java.util.List;
+import java.util.Map;
 import java.util.Objects;
+import java.util.stream.Collectors;
 
 public class PbRuleEngineProcessedAuctionRequestHook implements ProcessedAuctionRequestHook {
 
@@ -60,6 +66,7 @@ private static Future<InvocationResult<AuctionRequestPayload>> succeeded(RuleRes
                 InvocationResultImpl.<AuctionRequestPayload>builder()
                         .status(InvocationStatus.success)
                         .action(toInvocationAction(result.getAction()))
+                        .rejections(toRejections(result.getSeatNonBid()))
                         .analyticsTags(result.getAnalyticsTags());
 
         if (result.isUpdate()) {
@@ -69,6 +76,26 @@ private static Future<InvocationResult<AuctionRequestPayload>> succeeded(RuleRes
         return Future.succeededFuture(resultBuilder.build());
     }
 
+    private static InvocationAction toInvocationAction(RuleAction ruleAction) {
+        return switch (ruleAction) {
+            case NO_ACTION -> InvocationAction.no_action;
+            case UPDATE -> InvocationAction.update;
+            case REJECT -> InvocationAction.reject;
+        };
+    }
+
+
+    private static List<Rejection> toRejections(SeatNonBid seatNonBid) {
+        return seatNonBid.getNonBid().stream()
+                .map(nonBid -> (Rejection) ImpRejection.of(nonBid.getImpId(), nonBid.getStatusCode()))
+                .toList();
+    }
+
+    private static Map<String, List<Rejection>> toRejections(List<SeatNonBid> seatNonBids) {
+        return seatNonBids.stream()
+                .collect(Collectors.toMap(SeatNonBid::getSeat, PbRuleEngineProcessedAuctionRequestHook::toRejections));
+    }
+
     private static Future<InvocationResult<AuctionRequestPayload>> failure(Throwable error) {
         return Future.succeededFuture(
                 InvocationResultImpl.<AuctionRequestPayload>builder()
@@ -78,14 +105,6 @@ private static Future<InvocationResult<AuctionRequestPayload>> failure(Throwable
                         .build());
     }
 
-    private static InvocationAction toInvocationAction(RuleAction ruleAction) {
-        return switch (ruleAction) {
-            case NO_ACTION -> InvocationAction.no_action;
-            case UPDATE -> InvocationAction.update;
-            case REJECT -> InvocationAction.reject;
-        };
-    }
-
     @Override
     public String code() {
         return CODE;

extra/modules/pb-rule-engine/src/test/java/org/prebid/server/hooks/modules/rule/engine/v1/PbRuleEngineProcessedAuctionRequestHookTest.java

@@ -9,6 +9,9 @@
 import org.mockito.Mock;
 import org.mockito.junit.jupiter.MockitoExtension;
 import org.prebid.server.auction.model.AuctionContext;
+import org.prebid.server.auction.model.BidRejectionReason;
+import org.prebid.server.auction.model.ImpRejection;
+import org.prebid.server.auction.model.Rejection;
 import org.prebid.server.exception.PreBidException;
 import org.prebid.server.hooks.modules.rule.engine.core.config.RuleParser;
 import org.prebid.server.hooks.modules.rule.engine.core.request.Granularity;
@@ -22,10 +25,14 @@
 import org.prebid.server.hooks.v1.analytics.Tags;
 import org.prebid.server.hooks.v1.auction.AuctionInvocationContext;
 import org.prebid.server.hooks.v1.auction.AuctionRequestPayload;
+import org.prebid.server.proto.openrtb.ext.response.seatnonbid.NonBid;
+import org.prebid.server.proto.openrtb.ext.response.seatnonbid.SeatNonBid;
 import org.prebid.server.settings.model.Account;
 
 import java.time.Instant;
 import java.util.Collections;
+import java.util.List;
+import java.util.Map;
 
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.mockito.ArgumentMatchers.any;
@@ -104,16 +111,23 @@ public void callShouldReturnNoActionWhenRuleActionIsNoAction() {
     @Test
     public void callShouldReturnPayloadUpdateWhenRuleActionIsUpdate() {
         // given
+        final SeatNonBid seatNonBid = SeatNonBid.of(
+                "bidder", Collections.singletonList(NonBid.of("impId", BidRejectionReason.NO_BID)));
+
         given(invocationContext.accountConfig()).willReturn(MAPPER.createObjectNode());
         given(processedAuctionRequestRule.process(
                 bidRequest,
                 RequestRuleContext.of(auctionContext, Granularity.Request.instance(), "datacenter")))
-                .willReturn(RuleResult.of(bidRequest, RuleAction.UPDATE, tags, Collections.emptyList()));
+                .willReturn(RuleResult.of(bidRequest, RuleAction.UPDATE, tags, Collections.singletonList(seatNonBid)));
 
         // when and then
+        final Map<String, List<Rejection>> rejections = Map.of(
+                "bidder", List.of(ImpRejection.of("impId", BidRejectionReason.NO_BID)));
+
         assertThat(target.call(payload, invocationContext).result()).satisfies(invocationResult -> {
             assertThat(invocationResult.status()).isEqualTo(InvocationStatus.success);
             assertThat(invocationResult.action()).isEqualTo(InvocationAction.update);
+            assertThat(invocationResult.rejections()).containsExactlyEntriesOf(rejections);
             assertThat(invocationResult.payloadUpdate()).isNotNull();
             assertThat(invocationResult.analyticsTags()).isEqualTo(tags);
         });

src/main/java/org/prebid/server/auction/model/BidRejectionReason.java

@@ -56,8 +56,6 @@ public enum BidRejectionReason {
      */
     REQUEST_BLOCKED_UNSUPPORTED_MEDIA_TYPE(202),
 
-    REQUEST_BLOCKED_OPTIMIZED(203),
-
     /**
      * This impression not sent to the bid adapter because the impression or the bidder was removed from the request.
      */