Core: Support seatnonbid Codes in Modules

extra/modules/greenbids-real-time-data/src/main/java/org/prebid/server/hooks/modules/greenbids/real/time/data/config/GreenbidsRealTimeDataConfiguration.java

@@ -6,16 +6,15 @@
 import com.google.cloud.storage.StorageOptions;
 import io.vertx.core.Vertx;
 import org.prebid.server.geolocation.CountryCodeMapper;
-import org.prebid.server.hooks.modules.greenbids.real.time.data.model.filter.ThrottlingThresholds;
-import org.prebid.server.hooks.modules.greenbids.real.time.data.core.ThrottlingThresholdsFactory;
-import org.prebid.server.hooks.modules.greenbids.real.time.data.core.GreenbidsInferenceDataService;
 import org.prebid.server.hooks.modules.greenbids.real.time.data.core.FilterService;
+import org.prebid.server.hooks.modules.greenbids.real.time.data.core.GreenbidsInferenceDataService;
 import org.prebid.server.hooks.modules.greenbids.real.time.data.core.ModelCache;
 import org.prebid.server.hooks.modules.greenbids.real.time.data.core.OnnxModelRunner;
 import org.prebid.server.hooks.modules.greenbids.real.time.data.core.OnnxModelRunnerFactory;
 import org.prebid.server.hooks.modules.greenbids.real.time.data.core.OnnxModelRunnerWithThresholds;
 import org.prebid.server.hooks.modules.greenbids.real.time.data.core.ThresholdCache;
-import org.prebid.server.hooks.modules.greenbids.real.time.data.core.GreenbidsInvocationService;
+import org.prebid.server.hooks.modules.greenbids.real.time.data.core.ThrottlingThresholdsFactory;
+import org.prebid.server.hooks.modules.greenbids.real.time.data.model.filter.ThrottlingThresholds;
 import org.prebid.server.hooks.modules.greenbids.real.time.data.v1.GreenbidsRealTimeDataProcessedAuctionRequestHook;
 import org.prebid.server.json.ObjectMapperProvider;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
@@ -48,16 +47,14 @@ GreenbidsInferenceDataService greenbidsInferenceDataService(DatabaseReaderFactor
     GreenbidsRealTimeDataModule greenbidsRealTimeDataModule(
             FilterService filterService,
             OnnxModelRunnerWithThresholds onnxModelRunnerWithThresholds,
-            GreenbidsInferenceDataService greenbidsInferenceDataService,
-            GreenbidsInvocationService greenbidsInvocationService) {
+            GreenbidsInferenceDataService greenbidsInferenceDataService) {
 
         return new GreenbidsRealTimeDataModule(List.of(
                 new GreenbidsRealTimeDataProcessedAuctionRequestHook(
                         ObjectMapperProvider.mapper(),
                         filterService,
                         onnxModelRunnerWithThresholds,
-                        greenbidsInferenceDataService,
-                        greenbidsInvocationService)));
+                        greenbidsInferenceDataService)));
     }
 
     @Bean
@@ -123,15 +120,7 @@ ThresholdCache thresholdCache(
     }
 
     @Bean
-    OnnxModelRunnerWithThresholds onnxModelRunnerWithThresholds(
-            ModelCache modelCache,
-            ThresholdCache thresholdCache) {
-
+    OnnxModelRunnerWithThresholds onnxModelRunnerWithThresholds(ModelCache modelCache, ThresholdCache thresholdCache) {
         return new OnnxModelRunnerWithThresholds(modelCache, thresholdCache);
     }
-
-    @Bean
-    GreenbidsInvocationService greenbidsInvocationService() {
-        return new GreenbidsInvocationService();
-    }
 }

extra/modules/greenbids-real-time-data/src/main/java/org/prebid/server/hooks/modules/greenbids/real/time/data/core/GreenbidsInvocationResultCreator.java

@@ -0,0 +1,84 @@
+package org.prebid.server.hooks.modules.greenbids.real.time.data.core;
+
+import com.fasterxml.jackson.databind.JsonNode;
+import com.iab.openrtb.request.BidRequest;
+import com.iab.openrtb.request.Imp;
+import org.apache.commons.lang3.ObjectUtils;
+import org.apache.commons.lang3.StringUtils;
+import org.prebid.server.analytics.reporter.greenbids.model.ExplorationResult;
+import org.prebid.server.analytics.reporter.greenbids.model.Ortb2ImpExtResult;
+import org.prebid.server.hooks.modules.greenbids.real.time.data.model.data.GreenbidsConfig;
+import org.prebid.server.hooks.modules.greenbids.real.time.data.model.result.AnalyticsResult;
+import org.prebid.server.hooks.modules.greenbids.real.time.data.model.result.GreenbidsInvocationResult;
+import org.prebid.server.hooks.v1.InvocationAction;
+
+import java.util.Map;
+import java.util.Optional;
+import java.util.UUID;
+import java.util.stream.Collectors;
+
+public class GreenbidsInvocationResultCreator {
+
+    private static final int RANGE_16_BIT_INTEGER_DIVISION_BASIS = 0x10000;
+    private static final double DEFAULT_EXPLORATION_RATE = 1.0;
+
+    private GreenbidsInvocationResultCreator() {
+
+    }
+
+    public static GreenbidsInvocationResult create(GreenbidsConfig greenbidsConfig,
+                                                   BidRequest bidRequest,
+                                                   Map<String, Map<String, Boolean>> impsBiddersFilterMap) {
+
+        final String greenbidsId = UUID.randomUUID().toString();
+        final boolean isExploration = isExploration(greenbidsConfig, greenbidsId);
+
+        final boolean allRejected = bidRequest.getImp().stream()
+                .noneMatch(imp -> impsBiddersFilterMap.get(imp.getId()).values().stream().anyMatch(isKept -> isKept));
+
+        final InvocationAction invocationAction = isExploration
+                ? InvocationAction.no_action
+                : allRejected
+                    ? InvocationAction.reject
+                    : InvocationAction.update;
+
+        final Map<String, Ortb2ImpExtResult> ort2ImpExtResultMap = createOrtb2ImpExtForImps(
+                bidRequest, impsBiddersFilterMap, greenbidsId, isExploration);
+        final AnalyticsResult analyticsResult = AnalyticsResult.of("success", ort2ImpExtResultMap);
+        return GreenbidsInvocationResult.of(invocationAction, analyticsResult);
+    }
+
+    private static boolean isExploration(GreenbidsConfig greenbidsConfig, String greenbidsId) {
+        final double explorationRate = ObjectUtils.defaultIfNull(greenbidsConfig.getExplorationRate(), DEFAULT_EXPLORATION_RATE);
+        final int hashInt = Integer.parseInt(greenbidsId.substring(greenbidsId.length() - 4), 16);
+        return hashInt < explorationRate * RANGE_16_BIT_INTEGER_DIVISION_BASIS;
+    }
+
+    private static Map<String, Ortb2ImpExtResult> createOrtb2ImpExtForImps(
+            BidRequest bidRequest,
+            Map<String, Map<String, Boolean>> impsBiddersFilterMap,
+            String greenbidsId,
+            boolean isExploration) {
+
+        return bidRequest.getImp().stream()
+                .collect(Collectors.toMap(
+                        Imp::getId,
+                        imp -> createOrtb2ImpExt(imp, impsBiddersFilterMap, greenbidsId, isExploration)));
+    }
+
+    private static Ortb2ImpExtResult createOrtb2ImpExt(Imp imp,
+                                                       Map<String, Map<String, Boolean>> impsBiddersFilterMap,
+                                                       String greenbidsId,
+                                                       boolean isExploration) {
+
+        final String tid = Optional.ofNullable(imp)
+                .map(Imp::getExt)
+                .map(impExt -> impExt.get("tid"))
+                .map(JsonNode::asText)
+                .orElse(StringUtils.EMPTY);
+        final Map<String, Boolean> impBiddersFilterMap = impsBiddersFilterMap.get(imp.getId());
+        final ExplorationResult explorationResult = ExplorationResult.of(
+                greenbidsId, impBiddersFilterMap, isExploration);
+        return Ortb2ImpExtResult.of(explorationResult, tid);
+    }
+}

extra/modules/greenbids-real-time-data/src/main/java/org/prebid/server/hooks/modules/greenbids/real/time/data/core/GreenbidsPayloadUpdater.java

@@ -0,0 +1,51 @@
+package org.prebid.server.hooks.modules.greenbids.real.time.data.core;
+
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import com.iab.openrtb.request.BidRequest;
+import com.iab.openrtb.request.Imp;
+
+import java.util.List;
+import java.util.Map;
+import java.util.Optional;
+
+public class GreenbidsPayloadUpdater {
+
+    private GreenbidsPayloadUpdater() {
+
+    }
+
+    public static BidRequest update(BidRequest bidRequest, Map<String, Map<String, Boolean>> impsBiddersFilterMap) {
+        return bidRequest.toBuilder()
+                .imp(updateImps(bidRequest, impsBiddersFilterMap))
+                .build();
+    }
+
+    private static List<Imp> updateImps(BidRequest bidRequest, Map<String, Map<String, Boolean>> impsBiddersFilterMap) {
+        return bidRequest.getImp().stream()
+                .filter(imp -> isImpKept(impsBiddersFilterMap.get(imp.getId())))
+                .map(imp -> updateImp(imp, impsBiddersFilterMap.get(imp.getId())))
+                .toList();
+    }
+
+    private static boolean isImpKept(Map<String, Boolean> bidderFilterMap) {
+        return bidderFilterMap.values().stream().anyMatch(isKept -> isKept);
+    }
+
+    private static Imp updateImp(Imp imp, Map<String, Boolean> bidderFilterMap) {
+        return imp.toBuilder()
+                .ext(updateImpExt(imp.getExt(), bidderFilterMap))
+                .build();
+    }
+
+    private static ObjectNode updateImpExt(ObjectNode impExt, Map<String, Boolean> bidderFilterMap) {
+        final ObjectNode updatedExt = impExt.deepCopy();
+        Optional.ofNullable((ObjectNode) updatedExt.get("prebid"))
+                .map(prebidNode -> (ObjectNode) prebidNode.get("bidder"))
+                .ifPresent(bidderNode ->
+                        bidderFilterMap.entrySet().stream()
+                                .filter(entry -> !entry.getValue())
+                                .map(Map.Entry::getKey)
+                                .forEach(bidderNode::remove));
+        return updatedExt;
+    }
+}

extra/modules/greenbids-real-time-data/src/main/java/org/prebid/server/hooks/modules/greenbids/real/time/data/model/result/GreenbidsInvocationResult.java

@@ -1,14 +1,11 @@
 package org.prebid.server.hooks.modules.greenbids.real.time.data.model.result;
 
-import com.iab.openrtb.request.BidRequest;
 import lombok.Value;
 import org.prebid.server.hooks.v1.InvocationAction;
 
 @Value(staticConstructor = "of")
 public class GreenbidsInvocationResult {
 
-    BidRequest updatedBidRequest;
-
     InvocationAction invocationAction;
 
     AnalyticsResult analyticsResult;